English

Establishing a Scale for Kullback-Leibler Divergence in Language Models Across Various Settings

Computation and Language 2026-04-21 v3

Abstract

Log-likelihood vectors define a common space for comparing language models as probability distributions, enabling unified comparisons across heterogeneous settings. We extend this framework to training checkpoints and intermediate layers, and establish a consistent scale for KL divergence across pretraining, model size, random seeds, quantization, fine-tuning, and layers. Analysis of Pythia pretraining trajectories further shows that changes in log-likelihood space, as measured by the scaling behavior of KL divergence, are much smaller than in weight space, resulting in subdiffusive learning trajectories and early stabilization of language-model behavior despite weight drift.

Keywords

Cite

@article{arxiv.2505.15353,
  title  = {Establishing a Scale for Kullback-Leibler Divergence in Language Models Across Various Settings},
  author = {Ryo Kishino and Yusuke Takase and Momose Oyama and Hiroaki Yamagiwa and Hidetoshi Shimodaira},
  journal= {arXiv preprint arXiv:2505.15353},
  year   = {2026}
}

Comments

ACL 2026 Findings

R2 v1 2026-07-01T02:28:05.349Z