Update README.md

README.md

@@ -69,9 +69,9 @@ It builds on the work done in [SciRepEval: A Multi-Format Benchmark for Scientif
 - **Shared by :** Allen AI
 - **Model type:** bert-base-uncased + adapters
 - **License:** Apache 2.0
-- **Finetuned from model [optional]:** [allenai/scibert](https://huggingface.co/allenai/scibert_scivocab_uncased).
+- **Finetuned from model:** [allenai/scibert](https://huggingface.co/allenai/scibert_scivocab_uncased).
 
-## Model Sources [optional]
+## Model Sources
 
 <!-- Provide the basic links for the model. -->
 
@@ -85,13 +85,14 @@ It builds on the work done in [SciRepEval: A Multi-Format Benchmark for Scientif
 
 ## Direct Use
 
-|Model|Type|Name and HF link|
+|Model|Name and HF link|Description|
 |--|--|--|
-|Base|Transformer|[allenai/specter2](https://huggingface.co/allenai/specter2)|
-|Classification|Adapter|[allenai/specter2_classification](https://huggingface.co/allenai/specter2_classification)|
-|Regression|Adapter|[allenai/specter2_regression](https://huggingface.co/allenai/specter2_regression)|
-|Retrieval|Adapter|[allenai/specter2_proximity](https://huggingface.co/allenai/specter2_proximity)|
-|Adhoc Query|Adapter|[allenai/specter2_adhoc_query](https://huggingface.co/allenai/specter2_adhoc_query)|
+|Retrieval*|[allenai/specter2_proximity](https://huggingface.co/allenai/specter2_proximity)|Encode papers as queries and candidates eg. Link Prediction, Nearest Neighbor Search|
+|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)|
+|Classification|[allenai/specter2_classification](https://huggingface.co/allenai/specter2_classification)|Encode papers to feed into linear classifiers as features|
+|Regression|[allenai/specter2_regression](https://huggingface.co/allenai/specter2_regression)|Encode papers to feed into linear regressors as features|
+
+*Retrieval model should suffice for downstream task types not mentioned above
 
 ```python
 from transformers import AutoTokenizer, AutoModel
@@ -118,7 +119,7 @@ output = model(**inputs)
 embeddings = output.last_hidden_state[:, 0, :]
 ```
 
-## Downstream Use [optional]
+## Downstream Use
 
 <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
 
@@ -167,7 +168,6 @@ We also evaluate and establish a new SoTA on [MDCR](https://github.com/zoranmedi
 |[SPECTER](https://huggingface.co/allenai/specter)|54.7|57.4|68.0|(30.6, 25.5)|
 |[SciNCL](https://huggingface.co/malteos/scincl)|55.6|57.8|69.0|(32.6, 27.3)|
 |[SciRepEval-Adapters](https://huggingface.co/models?search=scirepeval)|61.9|59.0|70.9|(35.3, 29.6)|
-|[SPECTER 2.0-base](https://huggingface.co/allenai/specter2)|56.3|58.0|69.2|(38.0, 32.4)|
 |[SPECTER 2.0-Adapters](https://huggingface.co/models?search=allenai/specter-2)|**62.3**|**59.2**|**71.2**|**(38.4, 33.0)**|
 
 Please cite the following works if you end up using SPECTER 2.0: