On the Geometric Structure of Layer Updates in Deep Language Models
Not on the map. Why?
Abstract (arXiv)
We study the geometric structure of layer updates in deep language models. Rather than analyzing what information is encoded in intermediate representations, we ask how representations change from one layer to the next. We show that layerwise updates admit a decomposition into a dominant tokenwise component and a residual that is not captured by restricted tokenwise function classes. Across multiple architectures, including Transformers and state-space models, we find that the full layer update is almost perfectly aligned with the tokenwise component, while the residual exhibits substantially weaker alignment, larger angular deviation, and significantly lower projection onto the dominant tokenwise subspace. This indicates that the residual is not merely a small correction, but a geometrically distinct component of the transformation. This geometric separation has functional consequences: approximation error under the restricted tokenwise model is strongly associated with output perturbation, with Spearman correlations often exceeding 0.7 and reaching up to 0.95 in larger models. Together, these results suggest that most layerwise updates behave like structured reparameterizations along a dominant direction, while functionally significant computation is concentrated in a geometrically distinct residual component. Our framework provides a simple, architecture-agnostic method for probing the geometric and functional structure of layer updates in modern language models.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 6 of its 9 references that are in the map. First posted on arXiv 2 Apr 2026.
Builds on
6 of 9 references.
References · 6
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2016The Mythos of Model Interpretabilitycited by 716
- 2020Probing the Probing Paradigm: Does Probing Accuracy Entail Task Relevance?cited by 144
- 2021DirectProbe: Studying Representations without Classifierscited by 25
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Large Language Models Encode Semantics and Alignment in Linearly Separable Representations
- 2025Layer by Layer: Uncovering Hidden Representations in Language Models
- 2026Topology-Aware Layer Pruning for Large Vision-Language Models
- 2025How Language Directions Align with Token Geometry in Multilingual LLMs
- 2025HELM: Hyperbolic Large Language Models via Mixture-of-Curvature Experts
- 2026Understanding Pruning Regimes in Vision-Language Models Through Domain-Aware Layer Selection
- 2025Multi-Scale Manifold Alignment for Interpreting Large Language Models: A Unified Information-Geometric Framework
- 2026Training-Driven Representational Geometry Modularization Predicts Brain Alignment in Language Models