English

L$^2$M: Mutual Information Scaling Law for Long-Context Language Modeling

Computation and Language 2025-10-27 v2 Artificial Intelligence Information Theory Machine Learning math.IT Data Analysis, Statistics and Probability

Abstract

We present a universal theoretical framework for understanding long-context language modeling based on a bipartite mutual information scaling law that we rigorously verify in natural language. We demonstrate that bipartite mutual information captures multi-token interactions distinct from and scaling independently of conventional two-point mutual information, and show that this provides a more complete characterization of the dependencies needed for accurately modeling long sequences. Leveraging this scaling law, we formulate the Long-context Language Modeling (L2^2M) condition, which lower bounds the necessary scaling of a model's history state -- the latent variables responsible for storing past information -- for effective long-context modeling. We validate the framework and its predictions on transformer and state-space models. Our work provides a principled foundation to understand long-context modeling and to design more efficient architectures with stronger long-context capabilities, with potential applications beyond natural language.

Keywords

Cite

@article{arxiv.2503.04725,
  title  = {L$^2$M: Mutual Information Scaling Law for Long-Context Language Modeling},
  author = {Zhuo Chen and Oriol Mayné i Comas and Zhuotao Jin and Di Luo and Marin Soljačić},
  journal= {arXiv preprint arXiv:2503.04725},
  year   = {2025}
}

Comments

34 pages, 13 figures, 2 tables

R2 v1 2026-06-28T22:09:39.878Z