CoolFace
Modelpublic

hmbyt5-preliminary/byt5-small-multilingual-4g-2e

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes12downloads
Model Card

hmByT5 - Preliminary Language Models

Preliminary Historic Multilingual and Monolingual ByT5 Models. Following languages are currently covered:

  • —English (British Library Corpus - Books)
  • —German (Europeana Newspaper)
  • —French (Europeana Newspaper)
  • —Finnish (Europeana Newspaper)
  • —Swedish (Europeana Newspaper)
  • —Dutch (Delpher Corpus)

More details can be found in our GitHub repository.

In this experiment we sample 4B bytes (~4GB of text) from each corpora (and upsample Swedish and Finnish) and train for another epoch (2 epochs in total).

Pretraining

We use the official JAX/FLAX example in Hugging Face Transformers to pretrain a ByT5 model on a single v3-8 TPU. Details about the training can be found here.

Evaluation on Downstream Tasks (NER)

We evaluated the hmByT5 model on downstream tasks:

ModelEnglish AjMCGerman AjMCFrench AjMCFinnish NewsEyeSwedish NewsEyeDutch ICDARFrench ICDARAvg.
`hmbyt5-preliminary/byt5-small-multilingual-4g-2e`83.86 ± 0.6187.54 ± 0.1984.29 ± 0.41

Acknowledgements

Research supported with Cloud TPUs from Google's TPU Research Cloud (TRC). Many Thanks for providing access to the TPUs ❤️