English

Representational Curvature Modulates Behavioral Uncertainty in Large Language Models

Artificial Intelligence 2026-04-28 v1 Computation and Language Machine Learning

Abstract

In autoregressive large language models (LLMs), temporal straightening offers an account of how the next-token prediction objective shapes representations. Models learn to progressively straighten the representational trajectory of input sequences across layers, potentially facilitating next-token prediction via linear extrapolation. However, a direct link between this trajectory and token-level behavior has been missing. We provide such a link by relating contextual curvature-a geometric measure of how sharply the representational trajectory bends over recent context-to next-token entropy. Across two models (GPT-2 XL and Pythia-2.8B), contextual curvature is correlated with entropy, and this relationship emerges during training. Perturbation experiments reveal selective dependence: manipulating curvature through trajectory-aligned interventions reliably modulates entropy, while geometrically misaligned perturbations have no effect. Finally, regularizing representations to be straighter during training modestly reduces token-level entropy without degrading validation loss. These results identify trajectory curvature as a task-aligned representational feature that influences behavioral uncertainty in LLMs.

Keywords

Cite

@article{arxiv.2604.23985,
  title  = {Representational Curvature Modulates Behavioral Uncertainty in Large Language Models},
  author = {Jack King and Evelina Fedorenko and Eghbal A. Hosseini},
  journal= {arXiv preprint arXiv:2604.23985},
  year   = {2026}
}
R2 v1 2026-07-01T12:36:16.229Z