mlsanityData to 14 Oct 2026

A Geometric Perspective on Next-Token Prediction in Large Language Models: Three Emerging Phases

arXiv 2605.09011 ↗First posted 9 May 2026cs.LG cs.AIcited by 3

Not on the map. Why?

Abstract (arXiv)

We investigate the geometry of predictive information across the layers of large language models (LLMs). We repurpose representation lenses-learned affine maps trained to predict the next token from intermediate residual streams-as geometric diagnostic tools. Rather than asking what the model predicts at each layer, we ask where predictive information resides and how it evolves across depth. We define at each layer a predictive readout subspace as the dominant k-dimensional singular subspace of such a map on the d-dimensional residual stream (where k is a resolution parameter), and track its trajectory on the Grassmann manifold as a similarity profile across layers. The profile is well described by unimodal distributions exhibiting a rise, near-plateau, and descent; varying k from 1% to 50% of d traces a Pareto frontier between visibility and energy retention, yet the same structure emerges at all scales. Across eight models from two families (Qwen2.5 and OLMo2, 1B-32B), we identify three geometric phases. Updates are approximately orthogonal to the residual stream throughout; what distinguishes the phases is their effect on the effective rank, which expands, stabilizes, and concentrates. In the first, Seeding Multiplexing, feed-forward memories and attention layers seed a candidate set in superposition in family-specific proportions, with the final token rising as leading candidate from 20% to 35% of positions across this phase. In the second, Hoisting Overriding, updates override existing subspaces to concentrate the candidate distribution without expanding the rank. In the third, Focal Convergence, high-energy low-rank updates write the winner into a form aligned with the unembedding direction. Phases 1 and 3 grow slowly with model depth, while Phase 2 expands linearly. The additional capacity of deeper LLMs is largely absorbed by candidate disambiguation.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 15 of its 21 references that are in the map. First posted on arXiv 9 May 2026.

Builds on

15 of 21 references.

References · 15

  1. 2025Qwen3 Technical Reportcited by 8,820
  2. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  3. 2017Decoupled Weight Decay Regularizationcited by 1,831
  4. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  5. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  6. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
  7. 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 671
  8. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
  9. 20242 OLMo 2 Furiouscited by 340
  10. 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 319
  11. 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 235
  12. 2023Jump to Conclusions: Short-Cutting Transformers With Linear Transformationscited by 90
3 more
  1. 2024Transformers represent belief state geometry in their residual streamcited by 77
  2. 2026When Models Manipulate Manifolds: The Geometry of a Counting Taskcited by 46
  3. 2025A multiscale analysis of mean-field transformers in the moderate interaction regimecited by 25

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Bridging the Dimensional Chasm: Uncover Layer-wise Dimensional Reduction in Transformers through Token Correlation
  2. 2025Tracing the Representation Geometry of Language Models from Pretraining to Post-training
  3. 2026On the Geometric Structure of Layer Updates in Deep Language Models
  4. 2024Lines of Thought in Large Language Models
  5. 2025Large Language Models Encode Semantics and Alignment in Linearly Separable Representations
  6. 2026Learning Representations through Token Prediction: Geometry, Approximation, and Downstream Guarantees
  7. 2025Unraveling Token Prediction Refinement and Identifying Essential Layers in Language Models
  8. 2026Learning to Read Out: Unembedding Dynamics in Language Model Pretraining