Geometric and Behavioral Stratification in Transformer Residual Streams
Not on the map. Why?
Abstract (arXiv)
Trained transformer models develop privileged bases: coordinate axes whose statistics differ from the rest of the residual stream. But what kind of direction does such a basis select? We investigate the prediction direction, the unembedding direction of the token a model currently predicts, and find that it functions as a content-defined privileged anchor. Measured with respect to this anchor, residual-stream variation is geometrically and behaviorally stratified by proximity to the prediction. The stratification holds in all eighteen models tested (dense and mixture-of-experts, 7B-120B, base and instruction-tuned). A narrow, scale-invariant prediction interface concentrates readout-relevant structure, while the vast prediction-distal complement expands with model scale. Because the prediction direction sits nearly orthogonal to the principal variance axes, variance-based analyses recover this organization only partly, and the shortfall grows with prompt heterogeneity. Anchoring reveals a steep geometric gradient: prediction-proximal regions are highly structured and cluster related prompts, while the complement is flatter and anti-discriminates among prompt groups. The interface is a narrow slice but functionally decisive. Disrupting the variance directions closest to the prediction causes immediate divergence and frequent task-frame shifts; disrupting the next level down delays divergence and preserves framing. The complement is weakly readout-aligned per direction yet causally and temporally load-bearing, and behavior is driven by direction rather than magnitude. These results establish the prediction direction as a privileged anchor distinct from previously described coordinate axes, and give a geometric account of how high-dimensional computation coexists with linear readout.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). First posted on arXiv 12 Aug 2026.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026The Geometry of Inference in Transformer Residual Streams29 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026Where Pretraining writes and Alignment reads: the asymmetry of Transformer weight space
- 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?
- 2026LEAP: Layer-wise Exit-Aware Pretraining for Efficient Transformer Inference
- 2026The Residual Stream Is All You Need: On the Redundancy of the KV Cache in Transformer Inference
- 2026SiameseNorm: Breaking the Barrier to Reconciling Pre/Post-Norm
- 2026Dropout Robustness and Cognitive Profiling of Transformer Models via Stochastic Inference
- 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
- 2024ResiDual Transformer Alignment with Spectral Decomposition