|
--- |
|
language: en |
|
license: mit |
|
tags: |
|
- flair |
|
- token-classification |
|
- sequence-tagger-model |
|
base_model: hmbyt5-preliminary/byt5-small-historic-multilingual-span20-flax |
|
inference: false |
|
widget: |
|
- text: On Wednesday , a public dinner was given by the Conservative Burgesses of |
|
Leads , to the Conservative members of the Leeds Town Council , in the Music Hall |
|
, Albion-street , which was very numerously attended . |
|
--- |
|
|
|
# Fine-tuned Flair Model on TopRes19th English NER Dataset (HIPE-2022) |
|
|
|
This Flair model was fine-tuned on the |
|
[TopRes19th English](https://github.com/hipe-eval/HIPE-2022-data/blob/main/documentation/README-topres19th.md) |
|
NER Dataset using hmByT5 as backbone LM. |
|
|
|
The TopRes19th dataset consists of NE-annotated historical English newspaper articles from 19C. |
|
|
|
The following NEs were annotated: `BUILDING`, `LOC` and `STREET`. |
|
|
|
# ⚠️ Inference Widget ⚠️ |
|
|
|
Fine-Tuning ByT5 models in Flair is currently done by implementing an own [`ByT5Embedding`][0] class. |
|
|
|
This class needs to be present when running the model with Flair. |
|
|
|
Thus, the inference widget is not working with hmByT5 at the moment on the Model Hub and is currently disabled. |
|
|
|
This should be fixed in future, when ByT5 fine-tuning is supported in Flair directly. |
|
|
|
[0]: https://github.com/stefan-it/hmBench/blob/main/byt5_embeddings.py |
|
|
|
# Results |
|
|
|
We performed a hyper-parameter search over the following parameters with 5 different seeds per configuration: |
|
|
|
* Batch Sizes: `[8, 4]` |
|
* Learning Rates: `[0.00015, 0.00016]` |
|
|
|
And report micro F1-score on development set: |
|
|
|
| Configuration | Run 1 | Run 2 | Run 3 | Run 4 | Run 5 | Avg. | |
|
|-------------------|--------------|--------------|--------------|--------------|--------------|--------------| |
|
| bs4-e10-lr0.00015 | [0.7992][1] | [0.8226][2] | [0.8205][3] | [0.8364][4] | [0.809][5] | 81.75 ± 1.26 | |
|
| bs8-e10-lr0.00015 | [0.8095][6] | [0.83][7] | [0.8024][8] | [0.8112][9] | [0.8189][10] | 81.44 ± 0.94 | |
|
| bs8-e10-lr0.00016 | [0.8144][11] | [0.8209][12] | [0.8065][13] | [0.8056][14] | [0.82][15] | 81.35 ± 0.65 | |
|
| bs4-e10-lr0.00016 | [0.8056][16] | [0.8105][17] | [0.809][18] | [0.808][19] | [0.8056][20] | 80.77 ± 0.19 | |
|
|
|
[1]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-1 |
|
[2]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-2 |
|
[3]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-3 |
|
[4]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-4 |
|
[5]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-5 |
|
[6]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-1 |
|
[7]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-2 |
|
[8]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-3 |
|
[9]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-4 |
|
[10]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00015-poolingfirst-layers-1-crfFalse-5 |
|
[11]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-1 |
|
[12]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-2 |
|
[13]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-3 |
|
[14]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-4 |
|
[15]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs8-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-5 |
|
[16]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-1 |
|
[17]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-2 |
|
[18]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-3 |
|
[19]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-4 |
|
[20]: https://hf.co/stefan-it/hmbench-topres19th-en-hmbyt5-bs4-wsFalse-e10-lr0.00016-poolingfirst-layers-1-crfFalse-5 |
|
|
|
The [training log](training.log) and TensorBoard logs (only for hmByT5 and hmTEAMS based models) are also uploaded to the model hub. |
|
|
|
More information about fine-tuning can be found [here](https://github.com/stefan-it/hmBench). |
|
|
|
# Acknowledgements |
|
|
|
We thank [Luisa März](https://github.com/LuisaMaerz), [Katharina Schmid](https://github.com/schmika) and |
|
[Erion Çano](https://github.com/erionc) for their fruitful discussions about Historic Language Models. |
|
|
|
Research supported with Cloud TPUs from Google's [TPU Research Cloud](https://sites.research.google/trc/about/) (TRC). |
|
Many Thanks for providing access to the TPUs ❤️ |
|
|