Search papers, labs, and topics across Lattice.
This study introduces a token-level framework that dissects the scaling laws of language model loss into localized learning events for individual contextualized tokens. By analyzing over one hundred pre-training runs on large-scale datasets, the authors demonstrate that token learning occurs in concentrated transitions, which significantly influences the overall scaling-law shape. The findings reveal that optimizing the training distribution based on token learnability can enhance validation loss reduction by 11%, providing empirical evidence that scaling laws are primarily driven by token-level learning dynamics.
Token-level learning dynamics, not just model size, dictate scaling laws in language models, revealing actionable insights for training optimization.
Language model loss follows remarkably regular scaling laws over model and data size, yet it remains unclear why the aggregate loss should exhibit a power-law form. Existing explanations often attribute this regularity to a heavy-tailed spectrum of pattern difficulty in natural language, but this view has not been directly validated at token-level granularity in large-scale real-data training. We present a token-level framework that decomposes scaling laws into localized learning events of individual contextualized tokens. By fitting token loss trajectories with sigmoids, we show that token learning is concentrated in localized transitions, giving rise to a learning-time spectrum that dominates the scaling-law shape. Across more than one hundred pre-training runs on large and diverse real-language corpora with modern LLM architectures, scaling up to 6B parameters and 300B training tokens, the measured learning-time spectrum quantitatively reconstructs the validation loss derivative along the training-step $T$, data-scale $D$, and model-scale $M$ axes. We further show that the same signal is actionable: by reshaping the training distribution according to when tokens become learnable, we alter the optimization trajectory and achieve 11\% faster validation-loss reduction. These results provide direct empirical evidence that scaling laws are governed primarily by the distribution of token-level learning times, and that this distribution can be used not only to explain scaling behavior but also to improve training performance.
Tsinghua AI