DT4H_XLM-R_stl_es-it-ro_disease
Model Description
This
multilingual clinical Named Entity Recognition (NER) model is designed to identify
disease mentions in biomedical and clinical text. It is based on
xlm-roberta-base and fine-tuned on translated variants of the clinical NER datasets
DisTEMIST and
CardioCCC, which consist of clinical case reports with manually annotated disease mentions, following a single-task learning (STL) approach and using the BIO tagging scheme for sequence labeling.
- Architecture: Single-task learning (STL)
- Training setup: Multilingual, Monolabel (DISEASE)
- Supported languages:
- Spanish (
es)
- Italian (
it)
- Romanian (
ro)
- Base model:
xlm-roberta-base
- Task: Token classification (NER)
- Label scheme: BIO
Training Data
The model is trained on multilingual clinical NER data combining DisTEMIST and CardioCCC across the supported languages.
The data is part of the
MultiClinNER subtask of the
MultiClinAI shared task, an initiative as part of the DataTools4Heart (DT4H) project, which provides translated and annotation-projected clinical corpora.
Training and test splits correspond to the MultiClinNER task at the 11th SMM4H-HeaRD Workshop (ACL 2026).
How to use
You can load the model using Hugging Face Transformers:
1from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
2
3model_name = "judithrosell/DT4H_XLM-R_stl_es-it-ro_disease"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForTokenClassification.from_pretrained(model_name)
7
8ner_pipeline = pipeline("token-classification", model=model, tokenizer=tokenizer)
9
10text = "El paciente presenta insuficiencia cardíaca."
11predictions = ner_pipeline(text)
12print(predictions)
⚠ Note: We recommend pre-tokenizing the input text into words, as this matches the training setup. Providing raw text directly may lead to slightly degraded performance.
The model outputs token-level predictions. For evaluation or submission, these predictions should be converted into character-level spans.
To facilitate this process, we provide an inference script in a
GitHub repository that:
- Loads the model
- Processes .txt files from an input directory
- Extracts disease mentions
- Exports predictions as a TSV file in the format required for the MultiClinAI evaluation library:
1filename label start_span end_span text
2MultiClinNER-es-test-disease-00019 DISEASE 158 169 miocarditis
Limitations and bias
At the time of submission, no formal bias or fairness evaluation has been conducted. We intend to conduct research in these areas in the future, and if completed, this model card will be updated.
Evaluation
Evaluation was conducted using strict (exact match) and character-level metrics on the MultiClinNER test set.
| Language | Strict P | Strict R | Strict F1 | Char P | Char R | Char F1 |
|---|
| es | 0.6525 | 0.6806 | 0.6663 | 0.7786 | 0.8087 | 0.7933 |
| it | 0.6852 | 0.6329 | 0.6580 | 0.8025 | 0.7410 | 0.7705 |
| ro | 0.7014 | 0.6954 | 0.6984 | 0.8115 | 0.8061 | 0.8088 |
| Average | | | 0.6742 | | | 0.7909 |
Additional information
Authors
NLP4BIA team at the Barcelona Supercomputing Center (
nlp4bia@bsc.es).
Contact information
judith.rosell [at] bsc.es
Funding
This model is part of the DataTools4Heart project, funded by the European Union’s Horizon Europe Framework Under Grant Agreement No. 101057849.