English

Exploiting Vocabulary Frequency Imbalance in Language Model Pre-training

Computation and Language 2025-12-01 v3 Machine Learning

Abstract

Large language models are trained with tokenizers, and the resulting token distribution is highly imbalanced: a few words dominate the stream while most occur rarely. Recent practice favors ever-larger vocabularies, but it is unclear where the benefit comes from. To this end, we perform a controlled study that scales the vocabulary of the language model from 24K to 196K while holding data, computation, and optimization unchanged. We begin by quantifying the complexity of tokenized text -- formalized via Kolmogorov complexity -- and show that larger vocabularies reduce this complexity. Above 24K, every common word is already tokenized as a single token, so enlarging vocabulary only deepens the relative token-frequency imbalance. Word-level loss decomposition shows that larger vocabularies reduce cross-entropy loss almost exclusively by lowering uncertainty on the 2,500 most frequent words, even though loss on the rare tail rises. The same frequent words cover roughly 75% of tokens in downstream benchmarks, so this training advantage transfers intact. We further show that enlarging model parameters with a fixed vocabulary yields the same frequent-word benefit. Our results recast "bigger vocabularies help" as "lowering complexity of tokenized text helps," offering a simple, principled knob for tokenizer-model co-design and clarifying the loss dynamics that govern language model scaling in pre-training.

Keywords

Cite

@article{arxiv.2508.15390,
  title  = {Exploiting Vocabulary Frequency Imbalance in Language Model Pre-training},
  author = {Woojin Chung and Jeonghoon Kim},
  journal= {arXiv preprint arXiv:2508.15390},
  year   = {2025}
}

Comments

NeurIPS 2025

R2 v1 2026-07-01T04:59:45.211Z