MAGIC: Near-Optimal Data Attribution for Deep Learning
Not on the map. Why?
Abstract (arXiv)
The goal of predictive data attribution is to estimate how adding or removing a given set of training datapoints will affect model predictions. In convex settings, this goal is straightforward (i.e., via the infinitesimal jackknife). In large-scale (non-convex) settings, however, existing methods are far less successful -- current methods' estimates often only weakly correlate with ground truth. In this work, we present a new data attribution method (MAGIC) that combines classical methods and recent advances in metadifferentiation to (nearly) optimally estimate the effect of adding or removing training data on model predictions.
Key facts. Cited by 22 papers in the map (data through 14 Oct 2026). Builds on 34 of its 46 references that are in the map. First posted on arXiv 23 Apr 2025.
Builds on
34 of 46 references.
References · 34
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
- 2017Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)cited by 1,319
- 2024Gemma: Open Models Based on Gemini Research and Technologycited by 916
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2023The Flan Collection: Designing Data and Methods for Effective Instruction Tuningcited by 739
- 2023OpenAssistant Conversations -- Democratizing Large Language Model Alignmentcited by 721
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 532
22 more
- 2023Scaling Data-Constrained Language Modelscited by 403
- 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 360
- 2023Studying Large Language Model Generalization with Influence Functionscited by 326
- 2023TRAK: Attributing Model Behavior at Scalecited by 283
- 2020Influence Functions in Deep Learning Are Fragilecited by 252
- 2017Forward and Reverse Gradient-Based Hyperparameter Optimizationcited by 234
- 2022Datamodels: Predicting Predictions from Training Datacited by 189
- 2019On the Accuracy of Influence Functions for Measuring Group Effectscited by 147
- 2018Fast Approximate Natural Gradient Descent in a Kronecker-factored Eigenbasiscited by 146
- 2022If Influence Functions are the Answer, Then What is the Question?cited by 144
- 2022Training Data Influence Analysis and Estimation: A Surveycited by 135
- 2021Randomness In Neural Network Training: Characterizing The Impact of Toolingcited by 59
- 2023Simfluence: Modeling the Influence of Individual Training Examples by Simulating Training Runscited by 53
- 2024Training Data Attribution via Approximate Unrolled Differentiationcited by 34
- 2022ModelDiff: A Framework for Comparing Learning Algorithmscited by 30
- 2022Understanding Influence Functions and Datamodels via Harmonic Analysiscited by 23
- 2025Optimizing ML Training with Metagradient Descentcited by 23
- 2023On the Variance of Neural Network Training with respect to Test Sets and Distributionscited by 22
- 2024Attribute-to-Delete: Machine Unlearning via Datamodel Matchingcited by 18
- 2018A scalable estimate of the extra-sample prediction error via approximate leave-one-outcited by 11
- 2024Generalized Group Data Attributioncited by 6
- 2024RandALO: Out-of-sample risk estimation in no time flatcited by 1
Built on by
Citing papers per year: 2025: 5; 2026: 17.
22 papers.
Sort:
By area:
- 2025CUPID: Curating Data your Robot Loves with Influence Functionscited by 23
- 2026Subliminal Effects in Your Data: A General Mechanism via Log-Linearitycited by 19
- 2025Better Training Data Attribution via Better Inverse Hessian-Vector Productscited by 10
- 2025Train Once, Answer All: Many Pretraining Experiments for the Cost of Onecited by 6
- 2026Infusion: Shaping Model Behavior by Editing Training Data via Influence Functionscited by 5
- 2026Efficient Estimation of Kernel Surrogate Models for Task Attributioncited by 4
- 2026How Faithful Is Trajectory-Based Data Attribution? Error Sources, Remedies, and Practical Guidelinescited by 4
- 2025Distributional Training Data Attribution: What do Influence Functions Sample?cited by 3
- 2025Better Hessians Matter: Studying the Impact of Curvature Approximations in Influence Functionscited by 3
- 2026In-Run Data Shapley for Adam Optimizercited by 3
- 2026Position: Don't Just "Fix it in Post": A Science of AI Must Study Training Dynamicscited by 3
- 2026Bergson: An Open Source Library for Data Attributioncited by 3
10 more
- 2026How to sketch a learning algorithmcited by 2
- 2026Prototype Language Modelscited by 2
- 2026Challenges in Enabling Private Data Valuationcited by 1
- 2026Curation Leaks: Membership Inference Attacks against Data Curation for Machine Learningcited by 0
- 2026Small edits, large models: How Wikipedia advocacy shapes LLM valuescited by 0
- 2026Assert, don't describe: Linguistic features that shift LLM reasoning about animal welfarecited by 0
- 2026Data Attribution via Sketched Metadifferentiationcited by 0
- 2026Scalable Attribution and Control of Model Behavior During Trainingcited by 0
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scalecited by 0
- 2026Framework for an AI-Ready ARPES Community Databasecited by 0
- 2026Framework for an AI-Ready ARPES Community Database11 Oct 2026
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale30 Sep 2026
- 2026Scalable Attribution and Control of Model Behavior During Training27 Sep 2026
- 2026Data Attribution via Sketched Metadifferentiation14 Sep 2026
- 2026Prototype Language Models1 Jul 2026
- 2026Bergson: An Open Source Library for Data Attribution10 Jun 2026
- 2026Position: Don't Just "Fix it in Post": A Science of AI Must Study Training Dynamics3 Jun 2026
- 2026How Faithful Is Trajectory-Based Data Attribution? Error Sources, Remedies, and Practical Guidelines12 May 2026
- 2026Assert, don't describe: Linguistic features that shift LLM reasoning about animal welfare30 Apr 2026
- 2026Small edits, large models: How Wikipedia advocacy shapes LLM values30 Apr 2026
- 2026How to sketch a learning algorithm8 Apr 2026
- 2026Curation Leaks: Membership Inference Attacks against Data Curation for Machine Learning28 Feb 2026
10 more
- 2026Challenges in Enabling Private Data Valuation27 Feb 2026
- 2026Infusion: Shaping Model Behavior by Editing Training Data via Influence Functions10 Feb 2026
- 2026Subliminal Effects in Your Data: A General Mechanism via Log-Linearity4 Feb 2026
- 2026Efficient Estimation of Kernel Surrogate Models for Task Attribution3 Feb 2026
- 2026In-Run Data Shapley for Adam Optimizer30 Jan 2026
- 2025Better Hessians Matter: Studying the Impact of Curvature Approximations in Influence Functions27 Sep 2025
- 2025Train Once, Answer All: Many Pretraining Experiments for the Cost of One27 Sep 2025
- 2025Better Training Data Attribution via Better Inverse Hessian-Vector Products19 Jul 2025
- 2025CUPID: Curating Data your Robot Loves with Influence Functions23 Jun 2025
- 2025Distributional Training Data Attribution: What do Influence Functions Sample?15 Jun 2025
- 2026Subliminal Effects in Your Data: A General Mechanism via Log-Linearitycited by 19
- 2025Better Training Data Attribution via Better Inverse Hessian-Vector Productscited by 10
- 2025Train Once, Answer All: Many Pretraining Experiments for the Cost of Onecited by 6
- 2026Infusion: Shaping Model Behavior by Editing Training Data via Influence Functionscited by 5
- 2026How Faithful Is Trajectory-Based Data Attribution? Error Sources, Remedies, and Practical Guidelinescited by 4
- 2025Distributional Training Data Attribution: What do Influence Functions Sample?cited by 3
- 2025Better Hessians Matter: Studying the Impact of Curvature Approximations in Influence Functionscited by 3
- 2026In-Run Data Shapley for Adam Optimizercited by 3
- 2026Position: Don't Just "Fix it in Post": A Science of AI Must Study Training Dynamicscited by 3
- 2026Bergson: An Open Source Library for Data Attributioncited by 3
- 2026How to sketch a learning algorithmcited by 2
- 2026Prototype Language Modelscited by 2
12 most cited of 17 All 17 (JSON)
- 2026Framework for an AI-Ready ARPES Community Database11 Oct 2026
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale30 Sep 2026
- 2026Scalable Attribution and Control of Model Behavior During Training27 Sep 2026
- 2026Prototype Language Models1 Jul 2026
- 2026Bergson: An Open Source Library for Data Attribution10 Jun 2026
- 2026Position: Don't Just "Fix it in Post": A Science of AI Must Study Training Dynamics3 Jun 2026
- 2026How Faithful Is Trajectory-Based Data Attribution? Error Sources, Remedies, and Practical Guidelines12 May 2026
- 2026Small edits, large models: How Wikipedia advocacy shapes LLM values30 Apr 2026
- 2026How to sketch a learning algorithm8 Apr 2026
- 2026Challenges in Enabling Private Data Valuation27 Feb 2026
- 2026Infusion: Shaping Model Behavior by Editing Training Data via Influence Functions10 Feb 2026
- 2026Subliminal Effects in Your Data: A General Mechanism via Log-Linearity4 Feb 2026
12 newest of 17 All 17 (JSON)