|
--- |
|
license: apache-2.0 |
|
--- |
|
This is the finetuned model presented in MIReAD: a simple method for learning high-quality representations from |
|
scientific documents (ACL 2023). |
|
|
|
We trained MIReAD on >500,000 PubMed and arXiv abstracts across over 2,000 journal classes. MIReAD was initialized with SciBERT weights and finetuned to predict journal class based on the abstract and title of the paper. MIReAD uses SciBERT's tokenizer. |
|
We show that MIREAD produces representations that can be used for similar papers retrieval, topic categorization and literature search. |
|
|
|
Overall, with MIReAD you can: |
|
* extract semantically meaningful representation using paper's abstact |
|
* predict journal class based on paper's abstract |
|
|
|
To load the MIReAD model: |
|
```python |
|
from transformers import BertForSequenceClassification, AutoTokenizer |
|
|
|
mpath = 'arazd/miread' |
|
model_hub = BertForSequenceClassification.from_pretrained(mpath) |
|
tokenizer = AutoTokenizer.from_pretrained(mpath) |
|
``` |
