Disentangling Representation Evolution in Transformers through Directional Decomposition

Shwai He, Haichao Zhang, Shen Yan


Abstract
Transformer layers evolve representations through learned additive transformations that combine direction-preserving and direction-changing effects. We study this geometry by decomposing learned updates into parallel and perpendicular components. Across pretrained models, we find substantial learned parallel components, showing that direction-preserving behavior is not confined to the residual identity path. To test whether these components are redundant or behaviorally important, we apply the decomposition to residual-space updates and to value aggregation inside attention. Inference-time edits reveal a space-dependent asymmetry: value-space parallel edits preserve behavior more effectively than residual-space or perpendicular counterparts, especially when the direct self message is fixed and only the non-self aggregate is scaled. The decomposition also provides a compact, component-resolved view of compression error. During from-scratch pretraining, using parallel suppression as part of the forward computation lowers validation-loss trajectories and improves downstream averages in our evaluated runs, with the value-space variant strongest. Overall, this geometry connects update direction to editing robustness, compression diagnostics, and training-time intervention. Code is available in the https://github.com/Shwai-He/Transformer-Geometry.
Anthology ID:
2026.findings-emnlp.1397
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
29759–29772
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1397/
DOI:
Bibkey:
Cite (ACL):
Shwai He, Haichao Zhang, and Shen Yan. 2026. Disentangling Representation Evolution in Transformers through Directional Decomposition. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 29759–29772, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Disentangling Representation Evolution in Transformers through Directional Decomposition (He et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1397.pdf