Model Card for impresso-project/ner-stacked-bert-multilingual-v1.1.0
The Impresso NER model is a multilingual named entity recognition model trained for historical document processing. It is based on a stacked Transformer architecture and is designed to identify fine-grained and coarse-grained entity types in digitized historical texts, including names, titles, and locations.
This model was trained only on columns: NE-COARSE-LIT (pers, loc, and org) and NE-FINE-COMP (title, name and function).
Model Details
Model Description
- Developed by: EPFL from the Impresso team. The project is an interdisciplinary project focused on historical media analysis across languages, time, and modalities. Funded by the Swiss National Science Foundation (CRSII5_173719, CRSII5_213585) and the Luxembourg National Research Fund (grant No. 17498891).
- Model type: Stacked BERT-based token classification for named entity recognition
- Languages: French, German, English (with support for multilingual historical texts)
- License: AGPL v3+
- Finetuned from:
dbmdz/bert-medium-historic-multilingual-cased
Model Architecture
The model architecture consists of the following components:
- A pre-trained BERT encoder (multilingual historic BERT) as the base.
- One or two Transformer encoder layers stacked on top of the BERT encoder.
- A Conditional Random Field (CRF) decoder layer to model label dependencies.
- Learned absolute positional embeddings for improved handling of noisy inputs.
These additional Transformer layers help in mitigating the effects of OCR noise, spelling variation, and non-standard linguistic usage found in historical documents. The entire stack is fine-tuned end-to-end for token classification.
Training and Evaluation Results
The model was trained on the HIPE-2020 dataset (HIPE-2022 data release v2.1), combining French and German subsets for training.
The development set corresponds to the German portion (dev-de), and the test set corresponds to the French portion (test-fr).
Training Hyperparameters
- Training regime: Mixed precision (fp16)
- Epochs: 5
- Max sequence length: 512
- Base model:
dbmdz/bert-medium-historic-multilingual-cased - Stacked Transformer layers: 2
Results
All results below were obtained using the held-out French test set (HIPE-2020).
| Evaluation | Label | P | R | F1 |
|---|---|---|---|---|
| NE-COARSE-LIT-micro-fuzzy-TIME-ALL-LED-ALL | ALL | 0.826 | 0.796 | 0.811 |
| ↑ | LOC | 0.875 | 0.869 | 0.872 |
| ↑ | ORG | 0.596 | 0.500 | 0.544 |
| ↑ | PERS | 0.819 | 0.928 | 0.870 |
| NE-COARSE-LIT-micro-strict-TIME-ALL-LED-ALL | ALL | 0.720 | 0.693 | 0.706 |
| ↑ | LOC | 0.811 | 0.806 | 0.808 |
| ↑ | ORG | 0.505 | 0.423 | 0.460 |
| ↑ | PERS | 0.643 | 0.729 | 0.683 |
| NE-COARSE-LIT-macro_doc-fuzzy-TIME-ALL-LED-ALL | ALL | 0.852 | 0.800 | 0.816 |
| ↑ | LOC | 0.870 | 0.909 | 0.877 |
| ↑ | ORG | 0.538 | 0.494 | 0.547 |
| ↑ | PERS | 0.802 | 0.905 | 0.870 |
| NE-COARSE-LIT-macro_doc-strict-TIME-ALL-LED-ALL | ALL | 0.749 | 0.705 | 0.720 |
| ↑ | LOC | 0.816 | 0.854 | 0.823 |
| ↑ | ORG | 0.363 | 0.301 | 0.357 |
| ↑ | PERS | 0.651 | 0.727 | 0.703 |
Entity Types Supported
The model supports both coarse-grained and fine-grained entity types defined in the HIPE-2020/2022 guidelines. The output format of the model includes structured predictions with contextual and semantic details. Each prediction is a dictionary with the following fields:
{
'type': 'pers' | 'org' | 'loc',
'confidence_ner': float, # Confidence score
'surface': str, # Surface form in text
'lOffset': int, # Start character offset
'rOffset': int, # End character offset
'name': str, # Optional: full name (for persons)
'title': str, # Optional: title (for persons)
'function': str # Optional: function (if detected)
}
Coarse-Grained Entity Types:
- pers: Person entities (individuals, collectives, authors)
- org: Organizations (administrative, enterprise, press agencies)
- loc: Locations (towns, regions, countries, physical, facilities)
If present in the text, surrounding an entity, model returns person-specific attributes such as:
name: canonical full nametitle: honorific or title (e.g., "king", "chancellor")function: role or function in context (if available)
Model Sources
- Repository: https://huggingface.co/impresso-project/ner-stacked-bert-multilingual
- Paper: CoNLL 2020
- Demo: Impresso project
Uses
Direct Use
The model is intended to be used directly with the Hugging Face pipeline for token-classification, specifically with generic-ner tasks on historical texts.
Downstream Use
Can be used for downstream tasks such as:
- Historical information extraction
- Biographical reconstruction
- Place and person mention detection across historical archives
Out-of-Scope Use
- Not suitable for contemporary named entity recognition in domains such as social media or modern news.
- Not optimized for OCR-free modern corpora.
Bias, Risks, and Limitations
Due to training on historical documents, the model may reflect historical biases and inaccuracies. It may underperform on contemporary or non-European languages.
Recommendations
- Users should be cautious of historical and typographical biases.
- Consider post-processing to filter false positives from OCR noise.
How to Get Started with the Model
from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
MODEL_NAME = "impresso-project/ner-stacked-bert-multilingual-v1.1.0"
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
ner_pipeline = pipeline("generic-ner", model=MODEL_NAME, tokenizer=tokenizer, trust_remote_code=True, device='cpu')
sentence = "En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe, le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité. À la cour du roi Philippe VI, les murs du Louvre étaient animés par les rapports sombres venus de Paris et des villes environnantes. La peste ne montrait aucun signe de répit, et le chancelier Guillaume de Nogaret, le conseiller le plus fidèle du roi, portait le lourd fardeau de gérer la survie du royaume."
entities = ner_pipeline(sentence)
print(entities)
Example Output
[
{'type': 'time', 'confidence_ner': 85.0, 'surface': "an 1348", 'lOffset': 0, 'rOffset': 12},
{'type': 'loc', 'confidence_ner': 90.75, 'surface': "Europe", 'lOffset': 69, 'rOffset': 75},
{'type': 'loc', 'confidence_ner': 75.45, 'surface': "Royaume de France", 'lOffset': 80, 'rOffset': 97},
{'type': 'pers', 'confidence_ner': 85.27, 'surface': "roi Philippe VI", 'lOffset': 181, 'rOffset': 196, 'title': "roi", 'name': "roi Philippe VI"},
{'type': 'loc', 'confidence_ner': 30.59, 'surface': "Louvre", 'lOffset': 210, 'rOffset': 216},
{'type': 'loc', 'confidence_ner': 94.46, 'surface': "Paris", 'lOffset': 266, 'rOffset': 271},
{'type': 'pers', 'confidence_ner': 96.1, 'surface': "chancelier Guillaume de Nogaret", 'lOffset': 350, 'rOffset': 381, 'title': "chancelier", 'name': "Guillaume de Nogaret"},
{'type': 'loc', 'confidence_ner': 49.35, 'surface': "Royaume", 'lOffset': 80, 'rOffset': 87},
{'type': 'loc', 'confidence_ner': 24.18, 'surface': "France", 'lOffset': 91, 'rOffset': 97}
]
Citation
BibTeX:
@inproceedings{boros2020alleviating,
title={Alleviating digitization errors in named entity recognition for historical documents},
author={Boros, Emanuela and Hamdi, Ahmed and Pontes, Elvys Linhares and Cabrera-Diego, Luis-Adri{\'a}n and Moreno, Jose G and Sidere, Nicolas and Doucet, Antoine},
booktitle={Proceedings of the 24th conference on computational natural language learning},
pages={431--441},
year={2020}
}
Contact
- Website: https://impresso-project.ch
- Downloads last month
- 552