Calibration, Entropy Rates, and Memory in Language Models

Gap in Standard Metrics

Prior work on long-term properties of language models focused on architectural improvements. This paper identifies a more fundamental issue: standard metrics like perplexity and cross-entropy loss do not actually reflect long-term properties of generated text.

Entropy Rate Calibration

The authors propose using entropy rate for calibration. The key result (Theorem 4.4): calibration improves both:

  • Conventional training outcomes (lower perplexity)
  • Long-term dependency modeling

Experimental validation (Figure 2) shows that entropy-rate calibration simultaneously improves perplexity, demonstrating that the two objectives are not in conflict — better long-term modeling and better short-term modeling go together.

Memory via Mutual Information

The paper formalizes memory in language models using mutual information between a generated token and earlier context tokens at varying distances.

Empirical finding: during generation, language models rely heavily on recent context and progressively less on distant tokens — even with sufficient context window, the effective memory horizon is shorter than the architectural maximum.

Significance

The work bridges information theory and language modeling: it provides a principled metric for “long-term behavior” beyond perplexity, and a quantitative tool for analyzing how much of the context window LMs actually use.




    Enjoy Reading This Article?

    Here are some more articles you might like to read next:

  • The Expressive Power of Transformers with Chain of Thought
  • Landscape of Thoughts — Visualizing Where LLM Reasoning Actually Goes
  • Magellan — Guided MCTS for Escaping the Gravity Wells of LLM Creativity
  • PriorZero — Injecting LLM Priors into MuZero-Style World Models at the MCTS Root
  • SuperThoughts — Reasoning Tokens in Superposition