Language Adaptation on a Tight Academic Compute Budget: Tokenizer Swapping Works and Pure bfloat16 Is Enough
Not on the map. Why?
Abstract (arXiv)
We investigate continued pretraining of LLMs for language adaptation on a tight academic budget: a setting in which only a few GPUs can be used in parallel, for a heavily constrained duration. We focus on adapting Mistral-7B to German or Arabic and evaluate several techniques to improve efficiency and effectiveness in this setting. Our German models adapted on this tight compute budget underperform compared to the base Mistral-7B, while our Arabic models outperform several baselines, showing that for sufficiently well-represented languages, continued pretraining for specialization is not always helpful. Our main findings focus on training precision and tokenizer swapping. Our results show that pure bfloat16 training is a viable alternative to mixed-precision training, while being much faster when only using a few GPUs. Swapping the tokenizer for a specialized one yields more efficient tokenization and is competitive with the original tokenizer, which already contains some German tokens, but did not significantly increase performance for German. Code and model weights are available at on GitHub.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 36 of its 56 references that are in the map. First posted on arXiv 28 Aug 2024.
Builds on
36 of 56 references.
References · 36
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2019Unsupervised Cross-lingual Representation Learning at Scalecited by 3,935
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,975
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2018SentencePiece: A simple and language independent subword tokenizer and detokenizer for Neural Text Processingcited by 1,989
24 more
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2017Mixed Precision Trainingcited by 1,156
- 2021Scaling Vision Transformerscited by 999
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 861
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 463
- 2023Efficient and Effective Text Encoding for Chinese LLaMA and Alpacacited by 305
- 2019PAWS-X: A Cross-lingual Adversarial Dataset for Paraphrase Identificationcited by 264
- 2022Towards a Cleaner Document-Oriented Multilingual Crawled Corpuscited by 143
- 2024Simple and Scalable Strategies to Continually Pre-train Large Language Modelscited by 127
- 2024Continual Pre-Training for Cross-Lingual LLM Adaptation: Enhancing Japanese Language Capabilitiescited by 111
- 2024LLaMA Beyond English: An Empirical Study on Language Capability Transfercited by 95
- 2021WECHSEL: Effective initialization of subword embeddings for cross-lingual transfer of monolingual language modelscited by 89
- 2024ArabicMMLU: Assessing Massive Multitask Language Understanding in Arabiccited by 88
- 2020As Good as New. How to Successfully Recycle English GPT-2 to Make Models for Other Languagescited by 54
- 2024Zero-Shot Tokenizer Transfercited by 31
- 2023Efficiently Adapting Pretrained Language Models To New Languagescited by 28
- 2023Efficient Language Model Training through Cross-Lingual and Progressive Transfer Learningcited by 26
- 2024Teaching Llama a New Language Through Cross-Lingual Knowledge Transfercited by 23
- 2023Embedding structure matters: Comparing methods to adapt multilingual vocabularies to new languagescited by 16
- 2023Efficient Parallelization Layouts for Large-Scale Distributed Model Trainingcited by 8
- 2024Efficient Language Adaptive Pre-training: Extending State-of-the-Art Large Language Models for Polishcited by 4
- 2023Memory Efficient Mixed-Precision Optimizerscited by 1
Built on by
Citing papers per year: 2024: 1; 2025: 1.
2 papers.