Model: specter2-review-suggestion
The model snsf-data/specter2-review-suggestion is based on the allenai/specter2_base model and fine-tuned for a binary classification task.
In particular, the model is fine-tuned to classify if a sentence from SNSF grant peer review report is addressing the following aspect:
Does the sentence suggest how to improve the proposal?
The model was fine-tuned based on a training set of 2'500 sentences from the SNSF grant peer review reports, which were manually annotated by multiple human annotators via majority rule.
The fine-tuning was performed locally without access to the internet to prevent any potential data leakage or network interference.
The following setup was used for the fine-tuning:
- Loss function: cross-entropy loss
- Optimizer: AdamW
- Weight decay: 0.01
- Learning rate: 2e-5
- Epochs: 3
- Batch size: 10
- GPU: NVIDIA RTX A2000
The model was then evaluated based on a validation set of 500 sentences, which were also manually annotated by multiple human annotators via majority rule.
The resulting macro-average F1 score: 0.82 was achieved on the validation set. The share of the outcome label amounts to 4.5%.
The fine-tuning codes are open-sourced on GitHub:
https://github.com/snsf-data/ml-peer-review-analysis .
Due to data privacy laws no data used for the fine-tuning can be publicly shared.
For a detailed description of data protection please refer to the data management plan underlying this work:
https://doi.org/10.46446/DMP-peer-review-assessment-ML.
The annotation codebook is available online:
https://doi.org/10.46446/Codebook-peer-review-assessment-ML.
For more details, see the the following preprint:
A Supervised Machine Learning Approach for Assessing Grant Peer Review Reports
How to Get Started with the Model
The model can be used to classify sentences from grant peer review reports for addressing suggestions to improve the proposal.
Use the code below to get started with the model.
1# import transformers library
2import transformers
3
4# load tokenizer from specter2_base - the base model
5tokenizer = transformers.AutoTokenizer.from_pretrained("allenai/specter2_base")
6
7# load the SNSF fine-tuned model for classification of suggestions in review texts
8model = transformers.AutoModelForSequenceClassification.from_pretrained("snsf-data/specter2-review-suggestion")
9
10# setup the classification pipeline
11classification_pipeline = transformers.TextClassificationPipeline(
12 model=model,
13 tokenizer=tokenizer,
14 return_all_scores=True
15)
16
17# prediction for an example review sentence addressing suggestion
18classification_pipeline("If the project is accepted, then I would suggest that the project pays specific attention to increase practical relevance.")
19
20# prediction for an example review sentence not addressing suggestion
21classification_pipeline("There are currently several activities on an international level that have identified the issue and activities are underway.")
Model Limitations
- Human Assessment Required: This model should not be used for automatic classification of grant peer review reports without human oversight.
- Limited Training Data: The model was fine-tuned on a limited sample of 2,500 annotated sentences. Therefore, its classification accuracy should be critically evaluated before deployment.
- Specific Training Data: The training data consists of a random sample of SNSF grant peer review reports. As such, the model's external validity to other datasets may be limited.
Citation
BibTeX:
1@article{okasa2024supervised,
2 title={A Supervised Machine Learning Approach for Assessing Grant Peer Review Reports},
3 author={Okasa, Gabriel and de Le{\'o}n, Alberto and Strinzel, Michaela and Jorstad, Anne and Milzow, Katrin and Egger, Matthias and M{\"u}ller, Stefan},
4 journal={arXiv preprint arXiv:2411.16662},
5 year={2024}
6}
APA:
Okasa, G., de León, A., Strinzel, M., Jorstad, A., Milzow, K., Egger, M., & Müller, S. (2024). A Supervised Machine Learning Approach for Assessing Grant Peer Review Reports. arXiv preprint arXiv:2411.16662.
Model Card Authors
Model Card Contact
