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
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
Collections including this paper 0
No Collection including this paper