mlsanityData to 7 Oct 2026

Mechanistic Data Attribution: Tracing the Training Origins of Interpretable LLM Units

arXiv 2601.21996 ↗First posted 29 Jan 2026cs.CL cs.AI cs.LGcited by 10

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

  1. 2023GPT-4 Technical Reportcited by 18,132
  2. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
  3. 2016Pointer Sentinel Mixture Modelscited by 2,992
  4. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  5. 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
  6. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  7. 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,103
  8. 2022In-context Learning and Induction Headscited by 930
  9. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  10. 2021Knowledge Neurons in Pretrained Transformerscited by 590
  11. 2023Studying Large Language Model Generalization with Influence Functionscited by 311
  12. 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 303
14 more
  1. 2018Fast Approximate Natural Gradient Descent in a Kronecker-factored Eigenbasiscited by 143
  2. 2023Sudden Drops in the Loss: Syntax Acquisition, Phase Transitions, and Simplicity Bias in MLMscited by 121
  3. 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 86
  4. 2024On the Role of Attention Heads in Large Language Model Safetycited by 70
  5. 2024LLM Circuit Analyses Are Consistent Across Training and Scalecited by 60
  6. 2024Towards Understanding Safety Alignment: A Mechanistic Perspective from Safety Neuronscited by 47
  7. 2024Induction Heads as an Essential Mechanism for Pattern Matching in In-context Learningcited by 46
  8. 2024Understanding Factual Recall in Transformers via Associative Memoriescited by 45
  9. 2024A Implies B: Circuit Analysis in LLMs for Propositional Logical Reasoningcited by 15
  10. 2025Structural Inference: Interpreting Small Language Models with Susceptibilitiescited by 11
  11. 2025Evolution of Concepts in Language Model Pre-Trainingcited by 8
  12. 2025Which Data Attributes Stimulate Math and Code Reasoning? An Investigation via Influence Functionscited by 7
  13. 2025From Shortcut to Induction Head: How Data Diversity Shapes Algorithm Selection in Transformerscited by 6
  14. 2025Influence Dynamics and Stagewise Data Attributioncited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 10.

10 papers.