mlsanityData to 14 Oct 2026

L2M: Mutual Information Scaling Law for Long-Context Language Modeling

arXiv 2503.04725 ↗First posted 6 Mar 2025cs.CL cs.AI cs.ITcited by 9

Not on the map. Why?

Abstract (arXiv)

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 (L2M) 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.

Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 63 of its 114 references that are in the map. First posted on arXiv 6 Mar 2025.

Builds on

63 of 114 references.

References · 63

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  5. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  6. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  7. 2020Scaling Laws for Neural Language Modelscited by 6,586
  8. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  9. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  10. 2024Qwen2.5 Technical Reportcited by 4,942
  11. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  12. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
38 more
  1. 2020Longformer: The Long-Document Transformercited by 3,413
  2. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  3. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  4. 2023Sparks of Artificial General Intelligence: Early experiments with GPT-4cited by 2,592
  5. 2020Learning to summarize from human feedbackcited by 2,588
  6. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
  7. 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
  8. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  9. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
  10. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  11. 2020Big Bird: Transformers for Longer Sequencescited by 1,684
  12. 2019Generating Long Sequences with Sparse Transformerscited by 1,546
  13. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
  14. 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
  15. 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
  16. 2016InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Netscited by 860
  17. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  18. 2015Deep Learning and the Information Bottleneck Principlecited by 802
  19. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
  20. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
  21. 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 713
  22. 2019On Variational Bounds of Mutual Informationcited by 559
  23. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 519
  24. 2021Explaining Neural Scaling Lawscited by 363
  25. 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
  26. 2023CodeT5+: Open Code Large Language Models for Code Understanding and Generationcited by 333
  27. 2024xLSTM: Extended Long Short-Term Memorycited by 327
  28. 2019On Mutual Information Maximization for Representation Learningcited by 316
  29. 2023ToRA: A Tool-Integrated Reasoning Agent for Mathematical Problem Solvingcited by 282
  30. 2023LongNet: Scaling Transformers to 1,000,000,000 Tokenscited by 204
  31. 2023OpenAGI: When LLM Meets Domain Expertscited by 197
  32. 2019Adaptive Attention Span in Transformerscited by 149
  33. 2024A Dynamical Model of Neural Scaling Lawscited by 111
  34. 2019Quantum Natural Gradientcited by 106
  35. 2021Long-Short Transformer: Efficient Transformers for Language and Visioncited by 102
  36. 2024On the Origins of Linear Representations in Large Language Modelscited by 79
  37. 2020The Information Bottleneck Problem and Its Applications in Machine Learningcited by 66
  38. 2024Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Modelscited by 57

50 most cited of 63 All 63 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 5; 2026: 4.

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Rethinking the Understanding Ability across LLMs through Mutual Information
  2. 2025On the Fundamental Limits of LLMs at Scale
  3. 2026The Impossibility Triangle of Long-Context Modeling
  4. 2025Beyond Exponential Decay: Rethinking Error Accumulation in Large Language Models
  5. 2026The Root Theorem of Context Engineering
  6. 2024Michelangelo: Long Context Evaluations Beyond Haystacks via Latent Structure Queries
  7. 2024Beyond the Limits: A Survey of Techniques to Extend the Context Length in Large Language Models
  8. 2024Just read twice: closing the recall gap for recurrent language models