Equivalent Linear Mappings of Large Language Models
Not on the map. Why?
Abstract (arXiv)
Despite significant progress in transformer interpretability, an understanding of the computational mechanisms of large language models (LLMs) remains a fundamental challenge. Many approaches interpret a network's hidden representations but remain agnostic about how those representations are generated. We address this by mapping LLM inference for a given input sequence to an equivalent and interpretable linear system which reconstructs the predicted output embedding with relative error below 10-13 at double floating-point precision, requiring no additional model training. We exploit a property of transformers wherein every operation (gated activations, attention, and normalization) can be expressed as A(x) · x, where A(x) represents an input-dependent linear transform and x preserves the linear pathway. To expose this linear structure, we strategically detach components of the gradient computation with respect to an input sequence, freezing the A(x) terms at their values computed during inference, such that the Jacobian yields an equivalent linear mapping. This detached Jacobian of the model reconstructs the output with one linear operator per input token, which is shown for Qwen 3, Gemma 3 and Llama 3, up to Qwen 3 14B. These linear representations demonstrate that LLMs operate in extremely low-dimensional subspaces where the singular vectors can be decoded to interpretable semantic concepts. The computation for each intermediate output also has a linear equivalent, and we examine how the linear representations of individual layers and their attention and multilayer perceptron modules build predictions, and use these as steering operators to insert semantic concepts into unrelated text. Despite their global nonlinearity, LLMs can be interpreted through equivalent linear representations that reveal low-dimensional semantic structures in the next-token prediction process.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 16 of its 20 references that are in the map. First posted on arXiv 30 May 2025.
Builds on
16 of 20 references.
References · 16
- 2017Attention Is All You Needcited by 59,932
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2020GLU Variants Improve Transformercited by 1,724
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
- 2023In-context Vectors: Making In Context Learning More Effective and Controllable Through Latent Space Steeringcited by 240
- 2019Understanding and Improving Layer Normalizationcited by 213
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representationscited by 179
- 2025Open Problems in Mechanistic Interpretabilitycited by 173
- 2019Robust and interpretable blind image denoising via bias-free convolutional neural networkscited by 87
4 more
- 2023White-Box Transformers via Sparse Rate Reductioncited by 77
- 2022Interpreting Neural Networks through the Polytope Lenscited by 33
- 2024Geometric Dynamics of Signal Propagation Predict Trainability of Transformerscited by 24
- 2021Fast Jacobian-Vector Product for Deep Networkscited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.