L2M: Mutual Information Scaling Law for Long-Context Language Modeling
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
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
38 more
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2023Sparks of Artificial General Intelligence: Early experiments with GPT-4cited by 2,592
- 2020Learning to summarize from human feedbackcited by 2,588
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
- 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
- 2016InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Netscited by 860
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
- 2015Deep Learning and the Information Bottleneck Principlecited by 802
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 713
- 2019On Variational Bounds of Mutual Informationcited by 559
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 519
- 2021Explaining Neural Scaling Lawscited by 363
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
- 2023CodeT5+: Open Code Large Language Models for Code Understanding and Generationcited by 333
- 2024xLSTM: Extended Long Short-Term Memorycited by 327
- 2019On Mutual Information Maximization for Representation Learningcited by 316
- 2023ToRA: A Tool-Integrated Reasoning Agent for Mathematical Problem Solvingcited by 282
- 2023LongNet: Scaling Transformers to 1,000,000,000 Tokenscited by 204
- 2023OpenAGI: When LLM Meets Domain Expertscited by 197
- 2019Adaptive Attention Span in Transformerscited by 149
- 2024A Dynamical Model of Neural Scaling Lawscited by 111
- 2019Quantum Natural Gradientcited by 106
- 2021Long-Short Transformer: Efficient Transformers for Language and Visioncited by 102
- 2024On the Origins of Linear Representations in Large Language Modelscited by 79
- 2020The Information Bottleneck Problem and Its Applications in Machine Learningcited by 66
- 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
Citing papers per year: 2025: 5; 2026: 4.
9 papers.
- 2025Demystifying Reasoning Dynamics with Mutual Information: Thinking Tokens are Information Peaks in LLM Reasoningcited by 51
- 2025Tensorization is a powerful but underexplored tool for compression and interpretability of neural networkscited by 4
- 2025Correlation Dimension of Auto-Regressive Large Language Modelscited by 2
- 2025Advancing AI Challenges for the United States Department of the Air Forcecited by 1
- 2025Semantic-aware Graph-guided Behavior Sequences Generation with Large Language Models for Smart Homescited by 0
- 2026Semantic Rate Distortion and Posterior Design: Compute Constraints, Multimodality, and Strategic Inferencecited by 0
- 2026Trading Complexity for Expressivity Through Structured Generalized Linear Token Mixingcited by 0
- 2026Do Quantum Models Scale Like LLMs?cited by 0
- 2026The Cost of Long Memory: State, Context, and Stability Complexity in Sequence Modelscited by 0
- 2026The Cost of Long Memory: State, Context, and Stability Complexity in Sequence Models24 Sep 2026
- 2026Do Quantum Models Scale Like LLMs?17 Sep 2026
- 2026Trading Complexity for Expressivity Through Structured Generalized Linear Token Mixing29 May 2026
- 2026Semantic Rate Distortion and Posterior Design: Compute Constraints, Multimodality, and Strategic Inference3 Feb 2026
- 2025Advancing AI Challenges for the United States Department of the Air Force31 Oct 2025
- 2025Correlation Dimension of Auto-Regressive Large Language Models24 Oct 2025
- 2025Semantic-aware Graph-guided Behavior Sequences Generation with Large Language Models for Smart Homes5 Aug 2025
- 2025Demystifying Reasoning Dynamics with Mutual Information: Thinking Tokens are Information Peaks in LLM Reasoning3 Jun 2025
- 2025Tensorization is a powerful but underexplored tool for compression and interpretability of neural networks26 May 2025
Similar, not cited
Close in wording, no citation either way
- 2025Rethinking the Understanding Ability across LLMs through Mutual Information
- 2025On the Fundamental Limits of LLMs at Scale
- 2026The Impossibility Triangle of Long-Context Modeling
- 2025Beyond Exponential Decay: Rethinking Error Accumulation in Large Language Models
- 2026The Root Theorem of Context Engineering
- 2024Michelangelo: Long Context Evaluations Beyond Haystacks via Latent Structure Queries
- 2024Beyond the Limits: A Survey of Techniques to Extend the Context Length in Large Language Models
- 2024Just read twice: closing the recall gap for recurrent language models