File size: 1,761 Bytes
adc2246 24248ed 49edbcb 24248ed 0a970fa 24248ed 0a970fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
---
license: apache-2.0
language:
- en
metrics:
- precision
- recall
- f1
base_model:
- microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext
pipeline_tag: token-classification
library_name: transformers
---
# Fine-tuned NER Model for DiMB-RE
## Model Description
This is a fine-tuned **Named Entity Recognition (NER)** model based on the [BiomedNLP-BiomedBERT-base-uncased](https://huggingface.co/microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext) model, specifically designed for **span prediction task** to extract entity and trigger mentions for diet, human metabolism and microbiome field. The model has been trained on the DiMB-RE dataset and is optimized to identify spans for 15 different entity types, as well as 13 different trigger types.
<!-- ### Key Features:
- **Language**: English
- **Task**: Token classification for Named Entity Recognition (NER)
- **Base Model**: BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext
- **Domains**: Biomedical, Clinical, Scientific -->
## Performance
The model has been evaluated on the DiMB-RE using the following metrics:
- **NER** - P: 0.777, R: 0.745, F1: 0.760
- **NER Relaxed** - P: 0.852, R: 0.788, F1: 0.819
- **TRG** - P: 0.691, R: 0.631, F1: 0.660
- **TRG Relaxed** - P: 0.742, R: 0.678, F1: 0.708
## Citation
If you use this model, please cite like below:
```bibtex
@misc{hong2024dimbreminingscientificliterature,
title={DiMB-RE: Mining the Scientific Literature for Diet-Microbiome Associations},
author={Gibong Hong and Veronica Hindle and Nadine M. Veasley and Hannah D. Holscher and Halil Kilicoglu},
year={2024},
eprint={2409.19581},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2409.19581},
}
``` |