mlsanityData to 14 Oct 2026

Pre-trained Models Perform the Best When Token Distributions Follow Zipf's Law

arXiv 2507.22543 ↗First posted 30 Jul 2025cs.LG cs.CLcited by 3

Not on the map. Why?

Abstract (arXiv)

Tokenization is a fundamental step in natural language processing (NLP) and other sequence modeling domains, where the choice of vocabulary size significantly impacts model performance. Despite its importance, selecting an optimal vocabulary size remains underexplored, typically relying on heuristics or dataset-specific choices. In this work, we propose a principled method for determining the vocabulary size by analyzing token frequency distributions through Zipf's law. We show that downstream task performance correlates with how closely token distributions follow power-law behavior, and that aligning with Zipfian scaling improves both model efficiency and effectiveness. Extensive experiments across NLP, genomics, and chemistry demonstrate that models consistently achieve peak performance when the token distribution closely adheres to Zipf's law, establishing Zipfian alignment as a robust and generalizable criterion for vocabulary size selection.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 14 of its 34 references that are in the map. First posted on arXiv 30 Jul 2025.

Builds on

14 of 34 references.

References · 14

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  4. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  5. 2022Robust Speech Recognition via Large-Scale Weak Supervisioncited by 4,057
  6. 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,918
  7. 2018SentencePiece: A simple and language independent subword tokenizer and detokenizer for Neural Text Processingcited by 1,989
  8. 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,043
  9. 2020Multilingual Denoising Pre-training for Neural Machine Translationcited by 1,035
  10. 2018Molecular Transformer - A Model for Uncertainty-Calibrated Chemical Reaction Predictioncited by 229
  11. 2019BPE-Dropout: Simple and Effective Subword Regularizationcited by 195
  12. 2023Tokenizer Choice For LLM Training: Negligible or Crucial?cited by 108
2 more
  1. 2024Unpacking Tokenization: Evaluating Text Compression and its Correlation with Model Performancecited by 61
  2. 2022Scaling Laws Under the Microscope: Predicting Transformer Performance from Small Scale Experimentscited by 23

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.