Mechanistic Data Attribution: Tracing the Training Origins of Interpretable LLM Units
Not on the map. Why?
Abstract (arXiv)
While Mechanistic Interpretability has identified interpretable circuits in LLMs, their causal origins in training data remain elusive. We introduce Mechanistic Data Attribution (MDA), a scalable framework that employs Influence Functions to trace interpretable units back to specific training samples. Through extensive experiments on the Pythia family, we causally validate that targeted intervention--removing or augmenting a small fraction of high-influence samples--significantly modulates the emergence of interpretable heads, whereas random interventions show no effect. Our analysis reveals that repetitive structural data (e.g., LaTeX, XML) acts as a mechanistic catalyst. Furthermore, we observe that interventions targeting induction head formation induce a concurrent change in the model's in-context learning (ICL) capability. This provides direct causal evidence for the long-standing hypothesis regarding the functional link between induction heads and ICL. Finally, we propose a mechanistic data augmentation pipeline that consistently accelerates circuit convergence across model scales, providing a principled methodology for steering the developmental trajectories of LLMs.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 26 of its 36 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
26 of 36 references.
References · 26
- 2023GPT-4 Technical Reportcited by 18,132
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,103
- 2022In-context Learning and Induction Headscited by 930
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2021Knowledge Neurons in Pretrained Transformerscited by 590
- 2023Studying Large Language Model Generalization with Influence Functionscited by 311
- 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 303
14 more
- 2018Fast Approximate Natural Gradient Descent in a Kronecker-factored Eigenbasiscited by 143
- 2023Sudden Drops in the Loss: Syntax Acquisition, Phase Transitions, and Simplicity Bias in MLMscited by 121
- 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 86
- 2024On the Role of Attention Heads in Large Language Model Safetycited by 70
- 2024LLM Circuit Analyses Are Consistent Across Training and Scalecited by 60
- 2024Towards Understanding Safety Alignment: A Mechanistic Perspective from Safety Neuronscited by 47
- 2024Induction Heads as an Essential Mechanism for Pattern Matching in In-context Learningcited by 46
- 2024Understanding Factual Recall in Transformers via Associative Memoriescited by 45
- 2024A Implies B: Circuit Analysis in LLMs for Propositional Logical Reasoningcited by 15
- 2025Structural Inference: Interpreting Small Language Models with Susceptibilitiescited by 11
- 2025Evolution of Concepts in Language Model Pre-Trainingcited by 8
- 2025Which Data Attributes Stimulate Math and Code Reasoning? An Investigation via Influence Functionscited by 7
- 2025From Shortcut to Induction Head: How Data Diversity Shapes Algorithm Selection in Transformerscited by 6
- 2025Influence Dynamics and Stagewise Data Attributioncited by 5
Built on by
Citing papers per year: 2026: 10.
10 papers.
- 2026From Data to Behavior: Predicting Unintended Model Behaviors Before Trainingcited by 6
- 2026Symbolic Mechanistic Data Attribution: Tracing Training Influence to Learned Behavioral Policiescited by 1
- 2026Mechanistic Circuit Identification for Controllable Data Generationcited by 1
- 2026GIF: Locally Sound Geometric Information Flow Control for LLMscited by 0
- 2026Tracing LLM Behavior to the Training Data with Empirical Next-Token Distributionscited by 0
- 2026Tool Specifications Matter: Uncovering and Mitigating Safety Risks in AI Agentscited by 0
- 2026Preference Reasoning under Indeterminacy in Large Language Modelscited by 0
- 2026From Reweighting to Rewriting: Unlocking the Intervention Effects of Influential Samples in Training Data Attributioncited by 0
- 2026What Should Data Teach? Moving Bottlenecks Across Circuit, Store, and Usecited by 0
- 2026Commutator Memory: Sparse, Path-Local Reading and Steering in Language Modelscited by 0
- 2026Commutator Memory: Sparse, Path-Local Reading and Steering in Language Models28 Sep 2026
- 2026What Should Data Teach? Moving Bottlenecks Across Circuit, Store, and Use26 Sep 2026
- 2026From Reweighting to Rewriting: Unlocking the Intervention Effects of Influential Samples in Training Data Attribution2 Sep 2026
- 2026Mechanistic Circuit Identification for Controllable Data Generation25 Aug 2026
- 2026Preference Reasoning under Indeterminacy in Large Language Models19 Aug 2026
- 2026Tool Specifications Matter: Uncovering and Mitigating Safety Risks in AI Agents31 Jul 2026
- 2026Tracing LLM Behavior to the Training Data with Empirical Next-Token Distributions15 Jul 2026
- 2026Symbolic Mechanistic Data Attribution: Tracing Training Influence to Learned Behavioral Policies28 Jun 2026
- 2026GIF: Locally Sound Geometric Information Flow Control for LLMs22 Jun 2026
- 2026From Data to Behavior: Predicting Unintended Model Behaviors Before Training4 Feb 2026