tdx_latn_5mb / README.md
goldfish-models's picture
Upload README.md with huggingface_hub
d79aec0 verified
metadata
license: apache-2.0
language:
  - mlg
  - tdx
datasets:
  - allenai/MADLAD-400
library_name: transformers
pipeline_tag: text-generation
tags:
  - goldfish
  - arxiv:2408.10441

tdx_latn_5mb

Goldfish is a suite of monolingual language models trained for 350 languages. This model is the Tandroy-Mahafaly Malagasy (Latin script) model trained on 5MB of data, after accounting for an estimated byte premium of 1.00; content-matched text in Tandroy-Mahafaly Malagasy takes on average 1.00x as many UTF-8 bytes to encode as English. The Goldfish models are trained primarily for comparability across languages and for low-resource languages; Goldfish performance for high-resource languages is not designed to be comparable with modern large language models (LLMs).

Note: tdx_latn is an individual language code. Macrolanguage code mlg_latn (Malagasy) is included in Goldfish. Consider using that model depending on your use case.

All training and hyperparameter details are in our paper, Goldfish: Monolingual Language Models for 350 Languages (Chang et al., 2024).

Training code and sample usage: https://github.com/tylerachang/goldfish

Sample usage also in this Google Colab: link

Model details:

To access all Goldfish model details programmatically, see https://github.com/tylerachang/goldfish/blob/main/model_details.json. All models are trained with a [CLS] (same as [BOS]) token prepended, and a [SEP] (same as [EOS]) token separating sequences. For best results, make sure that [CLS] is prepended to your input sequence (see sample usage linked above)! Details for this model specifically:

  • Architecture: gpt2
  • Parameters: 34892800
  • Maximum sequence length: 512 tokens
  • Training text data (raw): 5.01MB
  • Training text data (byte premium scaled): 5.005MB
  • Training tokens: 1250304 (x10 epochs)
  • Vocabulary size: 41557
  • Compute cost: 945324703088640.0 FLOPs or ~0.1 NVIDIA A6000 GPU hours

Training datasets (percentages prior to deduplication):

Citation

If you use this model, please cite:

@article{chang-etal-2024-goldfish,
  title={Goldfish: Monolingual Language Models for 350 Languages},
  author={Chang, Tyler A. and Arnett, Catherine and Tu, Zhuowen and Bergen, Benjamin K.},
  journal={Preprint},
  year={2024},
  url={https://www.arxiv.org/abs/2408.10441},
}