A Geometric Perspective on Next-Token Prediction in Large Language Models: Three Emerging Phases
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
- 2025Qwen3 Technical Reportcited by 8,820
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2017Decoupled Weight Decay Regularizationcited by 1,831
- 2016Understanding intermediate layers using linear classifier probescited by 1,353
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
- 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 671
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
- 20242 OLMo 2 Furiouscited by 340
- 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 319
- 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 235
- 2023Jump to Conclusions: Short-Cutting Transformers With Linear Transformationscited by 90
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Bridging the Dimensional Chasm: Uncover Layer-wise Dimensional Reduction in Transformers through Token Correlation
- 2025Tracing the Representation Geometry of Language Models from Pretraining to Post-training
- 2026On the Geometric Structure of Layer Updates in Deep Language Models
- 2024Lines of Thought in Large Language Models
- 2025Large Language Models Encode Semantics and Alignment in Linearly Separable Representations
- 2026Learning Representations through Token Prediction: Geometry, Approximation, and Downstream Guarantees
- 2025Unraveling Token Prediction Refinement and Identifying Essential Layers in Language Models
- 2026Learning to Read Out: Unembedding Dynamics in Language Model Pretraining