In-Place Tokenizer Expansion for Pre-trained LLMs
Summary
Tokenizer expansion is an in-place method designed to upgrade the tokenizers of pre-trained Large Language Models, addressing inefficiencies for languages not initially prioritized. Fixed tokenizers often result in high token fragmentation for later-added languages, increasing latency, compute, and energy consumption. This recipe, suitable for model producers, continues existing BPE merges on a multilingual corpus, ensuring most source tokens remain unchanged and new tokens have exact decompositions. The process involves copying existing embedding rows and initializing new ones as the mean of their source sub-token embeddings. A two-stage adaptation, comprising embedding-only training followed by full-model continued pre-training, restores source-checkpoint quality. Applied to LFM2-8B-A1B, an 8B-parameter Mixture-of-Experts model, it produced LFM2.5-8B-A1B with a 128K tokenizer. This resulted in \$2.4\times$ fewer tokens for Hindi, \$2.6\times$ for Vietnamese, and up to \$4.0\times$ for Thai, yielding an estimated \$2.2$-\$3.7\times$ per-character decode speedup. The model weights and expanded tokenizer are released.
Key takeaway
For Machine Learning Engineers tasked with expanding language support for deployed LLMs, this in-place tokenizer expansion offers a critical pathway. You can significantly reduce token fragmentation and improve decode speeds for new languages, such as achieving \$2.2$-\$3.7\times$ speedups for Hindi and Vietnamese, without compromising original model quality. Consider implementing this two-stage adaptation to efficiently upgrade your model's multilingual capabilities and enhance user experience.
Key insights
In-place tokenizer expansion efficiently upgrades LLMs for new languages, reducing token fragmentation and improving decode speed without quality loss.
Principles
- Tokenizer fixed at pre-training reflects initial priorities.
- Fragmented tokens raise latency, compute, and energy.
- Embedding/LM-head matrices are material for compact models.
Method
Continue BPE merges on multilingual corpus; copy existing embedding rows; initialize new rows as mean of sub-token embeddings. Adapt via embedding-only training, then full-model continued pre-training.
In practice
- Upgrade pre-trained LLMs for new language support.
- Reduce per-character decode time for underrepresented languages.
- Expand vocabulary without full model retraining.
Topics
- Tokenizer Expansion
- Large Language Models
- Multilingual NLP
- BPE Merges
- Model Adaptation
- LFM2-8B-A1B
Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.