mlsanityData to 14 Oct 2026

Every Component Is a Lookup: One Linear Graph for Interaction, Composition and Attribution

arXiv 2605.23393 ↗First posted 22 May 2026cs.LG cs.AIcited by 1

On the map: Mechanistic interpretability → Circuits and mechanistic analysis

Abstract (arXiv)

Interpretability methods for transformers are typically built around separate questions: which components interact, how information routes to the output, and which input tokens contribute. Because these methods rely on different assumptions, their answers are difficult to relate. We argue that two architecturally motivated assumptions suffice to address all three questions: attention and MLPs share a key-value form, φ(S),U, in which φ(S) selects over values U, and components read from an additive residual stream, the sum of component outputs. Holding these selections at their forward-pass values turns the model into a computational graph, of which component interactions, composition paths, and token attribution are different readouts. We develop Unpack, a backward attribution procedure over this graph, and validate each readout against the corresponding established test: interaction scores predict ablation effects across models from 160M to 6.9B parameters, recovered routes reproduce established circuits down to the key, query, or value branch the circuit specifies, and token attribution passes the same faithfulness test as dedicated attribution methods. The results suggest that these two assumptions suffice for the interpretability questions above. On a task with a known circuit, we find that contribution and causal effect can differ, and that the difference has a recognisable signature: components that matter for the task change their contribution when the task is removed from the input, while components that act like a bias term do not. Code is available at https://github.com/Fun-Cry/unpacklm.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Circuits and mechanistic analysis, one of 13 directions in Mechanistic interpretability. Builds on 13 of its 17 references that are in the map. First posted on arXiv 22 May 2026.

Builds on

13 of 17 references.

Closest ancestors · 1

Same direction, newest first

  1. 2023Towards Automated Circuit Discovery for Mechanistic Interpretability28 Apr

Other references · 12

  1. 2017Attention Is All You Needcited by 60,361
  2. 2017Axiomatic Attribution for Deep Networkscited by 3,384
  3. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
  4. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  5. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  6. 2020Quantifying Attention Flow in Transformerscited by 775
  7. 2020Transformer Interpretability Beyond Attention Visualizationcited by 627
  8. 2023How does GPT-2 compute greater-than?: Interpreting mathematical abilities in a pre-trained language modelcited by 202
  9. 2024Have Faith in Faithfulness: Going Beyond Circuit Overlap When Finding Model Mechanismscited by 149
  10. 2024AttnLRP: Attention-Aware Layer-Wise Relevance Propagation for Transformerscited by 120
  11. 2022XAI for Transformers: Better Explanations through Conservative Propagationcited by 100
  12. 2024Information Flow Routes: Automatically Interpreting Language Models at Scalecited by 95

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2025DePass: Unified Feature Attributing by Simple Decomposed Forward Pass
  2. 2026Hidden Heroes and Gradient Bloats: Layer-Wise Redundancy Inverts Attribution in Transformers
  3. 2025Beyond Components: Singular Vector-Based Interpretability of Transformer Circuits
  4. 2026Representation as a Bottleneck for Mechanistic Interpretability: The Manifestation Unit Protocol
  5. 2025From Concepts to Components: Concept-Agnostic Attention Module Discovery in Transformers
  6. 2026How Do Linear Probes Emerge? A Circuit-Tracing Framework with Concept-Targeted Attribution
  7. 2026The Metagame of Interpretability and Meta-Attributions
  8. 2026MechaTerp-TRACE: A Novel Approach for Component Ablation Analysis in Language Models