Update README.md
Browse files
README.md
CHANGED
|
@@ -41,21 +41,38 @@ adapter_name = model.load_adapter("allenai/specter2", source="hf", set_active=Tr
|
|
| 41 |
|
| 42 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 43 |
|
| 44 |
-
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
Given the combination of title and abstract of a scientific paper or a short texual query, the model can be used to generate effective embeddings to be used in downstream applications.
|
| 47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
# Model Details
|
| 49 |
|
| 50 |
## Model Description
|
| 51 |
|
| 52 |
SPECTER 2.0 has been trained on over 6M triplets of scientific paper citations, which are available [here](https://huggingface.co/datasets/allenai/scirepeval/viewer/cite_prediction_new/evaluation).
|
| 53 |
-
Post that it is trained on all the [SciRepEval](https://huggingface.co/datasets/allenai/scirepeval) training tasks
|
| 54 |
|
| 55 |
Task Formats trained on:
|
| 56 |
- Classification
|
| 57 |
- Regression
|
| 58 |
-
- Proximity
|
| 59 |
- Adhoc Search
|
| 60 |
|
| 61 |
This is a retrieval specific adapter. For tasks where given a paper query, other relevant papers have to be retrieved from a corpus, use this adapter to generate the embeddings.
|
|
@@ -87,12 +104,13 @@ It builds on the work done in [SciRepEval: A Multi-Format Benchmark for Scientif
|
|
| 87 |
|
| 88 |
|Model|Name and HF link|Description|
|
| 89 |
|--|--|--|
|
| 90 |
-
|
|
| 91 |
-
|Adhoc Query|[allenai/specter2_adhoc_query](https://huggingface.co/allenai/specter2_adhoc_query)|Encode short raw text queries for search tasks. (Candidate papers can be encoded with proximity)|
|
| 92 |
|Classification|[allenai/specter2_classification](https://huggingface.co/allenai/specter2_classification)|Encode papers to feed into linear classifiers as features|
|
| 93 |
|Regression|[allenai/specter2_regression](https://huggingface.co/allenai/specter2_regression)|Encode papers to feed into linear regressors as features|
|
| 94 |
|
| 95 |
-
*
|
|
|
|
| 96 |
|
| 97 |
```python
|
| 98 |
from transformers import AutoTokenizer, AutoModel
|
|
|
|
| 41 |
|
| 42 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 43 |
|
| 44 |
+
**Aug 2023 Update:**
|
| 45 |
+
1. The SPECTER 2.0 Base and proximity adapter models have been renamed in Hugging Face based upon usage patterns as follows:
|
| 46 |
+
|
| 47 |
+
|Old Name|New Name|
|
| 48 |
+
|--|--|
|
| 49 |
+
|allenai/specter2|[allenai/specter2_base](https://huggingface.co/allenai/specter2_base)|
|
| 50 |
+
|allenai/specter2_proximity|[allenai/specter2](https://huggingface.co/allenai/specter2)|
|
| 51 |
+
|
| 52 |
+
2. We have a parallel version (termed [aug2023refresh](https://huggingface.co/allenai/specter2_aug2023refresh)) where the base transformer encoder version is pre-trained on a collection of newer papers (published after 2018).
|
| 53 |
+
However, for benchmarking purposes, please continue using the current version.
|
| 54 |
+
|
| 55 |
+
|
| 56 |
+
|
| 57 |
+
SPECTER 2.0 is the successor to [SPECTER](https://huggingface.co/allenai/specter) and is capable of generating task specific embeddings for scientific tasks when paired with [adapters](https://huggingface.co/models?search=allenai/specter-2_).
|
| 58 |
+
This is the base model to be used along with the adapters.
|
| 59 |
Given the combination of title and abstract of a scientific paper or a short texual query, the model can be used to generate effective embeddings to be used in downstream applications.
|
| 60 |
|
| 61 |
+
**Note:For general embedding purposes, please use [allenai/specter2](https://huggingface.co/allenai/specter2).**
|
| 62 |
+
|
| 63 |
+
**To get the best performance on a downstream task type please load the associated adapter with the base model as in the example below.**
|
| 64 |
+
|
| 65 |
# Model Details
|
| 66 |
|
| 67 |
## Model Description
|
| 68 |
|
| 69 |
SPECTER 2.0 has been trained on over 6M triplets of scientific paper citations, which are available [here](https://huggingface.co/datasets/allenai/scirepeval/viewer/cite_prediction_new/evaluation).
|
| 70 |
+
Post that it is trained with additionally attached task format specific adapter modules on all the [SciRepEval](https://huggingface.co/datasets/allenai/scirepeval) training tasks.
|
| 71 |
|
| 72 |
Task Formats trained on:
|
| 73 |
- Classification
|
| 74 |
- Regression
|
| 75 |
+
- Proximity (Retrieval)
|
| 76 |
- Adhoc Search
|
| 77 |
|
| 78 |
This is a retrieval specific adapter. For tasks where given a paper query, other relevant papers have to be retrieved from a corpus, use this adapter to generate the embeddings.
|
|
|
|
| 104 |
|
| 105 |
|Model|Name and HF link|Description|
|
| 106 |
|--|--|--|
|
| 107 |
+
|Proximity*|[allenai/specter2](https://huggingface.co/allenai/specter2)|Encode papers as queries and candidates eg. Link Prediction, Nearest Neighbor Search|
|
| 108 |
+
|Adhoc Query|[allenai/specter2_adhoc_query](https://huggingface.co/allenai/specter2_adhoc_query)|Encode short raw text queries for search tasks. (Candidate papers can be encoded with the proximity adapter)|
|
| 109 |
|Classification|[allenai/specter2_classification](https://huggingface.co/allenai/specter2_classification)|Encode papers to feed into linear classifiers as features|
|
| 110 |
|Regression|[allenai/specter2_regression](https://huggingface.co/allenai/specter2_regression)|Encode papers to feed into linear regressors as features|
|
| 111 |
|
| 112 |
+
*Proximity model should suffice for downstream task types not mentioned above
|
| 113 |
+
|
| 114 |
|
| 115 |
```python
|
| 116 |
from transformers import AutoTokenizer, AutoModel
|