Disentangling Representation Evolution in Transformers through Directional Decomposition
Abstract
Transformer representations evolve through learned additive transformations that either preserve their current direction or redirect it. We study this evolution as a functional geometry, decomposing learned updates into parallel and perpendicular components. Across pretrained models, we find substantial parallel components beyond the residual identity path. We then apply the decomposition in two spaces: to attention and MLP updates relative to the hidden state, and to attention value aggregation relative to the current token's value. Targeted edits reveal a strongly space-dependent asymmetry: exclude-self value-space parallel manipulation is markedly more robust than residual-space and perpendicular counterparts, preserving the direct self message while scaling only the non-self aggregate. The same decomposition gives a component-resolved description of compression-induced update error: perpendicular error separates compression methods more clearly than parallel error. Extensive experiments further demonstrate that full-aggregate parallel suppression during from-scratch pretraining lowers validation-loss trajectories and improves downstream averages, with the value-space variant strongest. Together, these results connect representation geometry to editing robustness, compression diagnosis, and training-time intervention. Code is available in the https://github.com/Shwai-He/Transformer-Geometry{project repository}.
Community
โ Core Question:
Are direction-preserving updates in Transformers merely redundant rescaling? We find that the answer depends strongly on where the geometry is defined.
๐ What We Did & Found:
We decompose learned transformations into components parallel and perpendicular to the current representation, across both residual space and attention value space.
โข Parallel components are widespread beyond the residual identity path, yet their functional role is highly space-dependent.
โข In value space, preserving the direct self-message while suppressing the non-self parallel component is markedly more robust than residual-space or perpendicular interventions.
๐ Practical Applications:
- Model Compression ๐: Perpendicular error more clearly distinguishes pruning- and quantization-induced distortions.
- From-Scratch Pretraining โก: Full-aggregate parallel suppression lowers validation-loss trajectories and improves downstream task averages, with the value-space variant being the strongest.
๐ Interactive Project Page & Studio: https://shwai-he.github.io/Transformer-Geometry/
๐ arXiv Paper: https://arxiv.org/abs/2609.15975
๐ป Code & Artifacts: https://github.com/Shwai-He/Transformer-Geometry
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Dual Attention Residuals (2026)
- CausalGate: Causal Importance Distillation for Transformer Module Pruning (2026)
- Gated Recurrent Transformers: Expressive Depth through Recurrent Modulation (2026)
- Reduced Matrix Multiplication: Input-Adaptive Matrix-Product Reduction for LLM Inference (2026)
- Feed-Forward Steering in Transformer Residual Dynamics (2026)
- CircuitSteer: Geometrically Aligned Multi-Layer Steering via Sparse Autoencoder Circuits (2026)
- Sparse Weight Decomposition for Efficient Circuit Extraction (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper