mlsanityData to 7 Oct 2026

Geometric and Behavioral Stratification in Transformer Residual Streams

arXiv 2608.12447 ↗First posted 12 Aug 2026cs.LG cs.CLcited by 1

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

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Where Pretraining writes and Alignment reads: the asymmetry of Transformer weight space
  2. 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?
  3. 2026LEAP: Layer-wise Exit-Aware Pretraining for Efficient Transformer Inference
  4. 2026The Residual Stream Is All You Need: On the Redundancy of the KV Cache in Transformer Inference
  5. 2026SiameseNorm: Breaking the Barrier to Reconciling Pre/Post-Norm
  6. 2026Dropout Robustness and Cognitive Profiling of Transformer Models via Stochastic Inference
  7. 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
  8. 2024ResiDual Transformer Alignment with Spectral Decomposition