Measuring the Effect of Training Data on Deep Learning Predictions via Randomized Experiments
Not on the map. Why?
Abstract (arXiv)
We develop a new, principled algorithm for estimating the contribution of training data points to the behavior of a deep learning model, such as a specific prediction it makes. Our algorithm estimates the AME, a quantity that measures the expected (average) marginal effect of adding a data point to a subset of the training data, sampled from a given distribution. When subsets are sampled from the uniform distribution, the AME reduces to the well-known Shapley value. Our approach is inspired by causal inference and randomized experiments: we sample different subsets of the training data to train multiple submodels, and evaluate each submodel's behavior. We then use a LASSO regression to jointly estimate the AME of each data point, based on the subset compositions. Under sparsity assumptions (k ll N datapoints have large AME), our estimator requires only O(klog N) randomized submodel trainings, improving upon the best prior Shapley value estimators.
Key facts. Cited by 50 papers in the map (data through 7 Oct 2026). Builds on 39 of its 92 references that are in the map. First posted on arXiv 20 Jun 2022.
Builds on
39 of 92 references.
References · 39
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2017Targeted Backdoor Attacks on Deep Learning Systems Using Data Poisoningcited by 1,012
- 2016CVXPY: A Python-Embedded Modeling Language for Convex Optimizationcited by 633
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
- 2018Poison Frogs! Targeted Clean-Label Poisoning Attacks on Neural Networkscited by 500
- 2020What Neural Networks Memorize and Why: Discovering the Long Tail via Influence Estimationcited by 492
- 2018Spectral Signatures in Backdoor Attackscited by 457
27 more
- 2019STRIP: A Defence Against Trojan Attacks on Deep Neural Networkscited by 437
- 2018Detecting Backdoor Attacks on Deep Neural Networks by Activation Clusteringcited by 429
- 2019Towards Efficient Data Valuation Based on the Shapley Valuecited by 279
- 2020Influence Functions in Deep Learning Are Fragilecited by 247
- 2018Manipulating Machine Learning: Poisoning Attacks and Countermeasures for Regression Learningcited by 219
- 2018Representer Point Selection for Explaining Deep Neural Networkscited by 187
- 2022Datamodels: Predicting Predictions from Training Datacited by 186
- 2017A Rewriting System for Convex Optimization Problemscited by 178
- 2021Beta Shapley: a Unified and Noise-reduced Data Valuation Framework for Machine Learningcited by 162
- 2021Counterfactual Memorization in Neural Language Modelscited by 155
- 2020Improving KernelSHAP: Practical Shapley Value Estimation via Linear Regressioncited by 153
- 2019On the Accuracy of Influence Functions for Measuring Group Effectscited by 146
- 2018Stronger Data Poisoning Attacks Break Data Sanitization Defensescited by 125
- 2021SPECTRE: Defending Against Backdoor Attacks Using Robust Statisticscited by 123
- 2021FastSHAP: Real-Time Shapley Value Estimationcited by 123
- 2019Demon in the Variant: Statistical Analysis of DNNs for Robust Backdoor Contamination Detectioncited by 121
- 2016Panning for Gold: Model-X Knockoffs for High-dimensional Controlled Variable Selectioncited by 120
- 2019Asymmetric Shapley values: incorporating causal knowledge into model-agnostic explainabilitycited by 114
- 2021Sampling Permutations for Shapley Value Estimationcited by 101
- 2020A Distributional Framework for Data Valuationcited by 79
- 2020Efficient nonparametric statistical inference on population feature importance using Shapley valuescited by 61
- 2013Bounding the Estimation Error of Sampling-based Shapley Value Approximationcited by 56
- 2019Model Agnostic Defence against Backdoor Attacks in Machine Learningcited by 52
- 2017BoostClean: Automated Error Detection and Repair for Machine Learningcited by 23
- 2020Complaint-driven Training Data Debugging for Query 2.0cited by 16
- 2021Joint Shapley values: a measure of joint feature importancecited by 15
- 2016Debugging Machine Learning Taskscited by 9
Built on by
Citing papers per year: 2022: 2; 2023: 13; 2024: 22; 2025: 8; 2026: 5.
50 papers.
Sort:
By area:
- 2023TRAK: Attributing Model Behavior at Scalecited by 273
- 2022Training Data Influence Analysis and Estimation: A Surveycited by 134
- 2024Data Shapley in One Training Runcited by 76
- 2023OpenDataVal: a Unified Benchmark for Data Valuationcited by 55
- 2023Intriguing Properties of Data Attribution on Diffusion Modelscited by 47
- 2023The Journey, Not the Destination: How Data Guides Diffusion Modelscited by 42
- 2023Data-OOB: Out-of-bag Estimate as a Simple and Efficient Data Valuecited by 38
- 2024Rethinking Data Shapley for Data Selection Tasks: Misleads and Meritscited by 29
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
- 2025DataMIL: Selecting Data for Robot Imitation Learning with Datamodelscited by 25
- 2024Diffusion Attribution Score: Evaluating Training Data Influence in Diffusion Modelscited by 21
- 2023On the Variance of Neural Network Training with respect to Test Sets and Distributionscited by 20
38 more
- 2024Causal Estimation of Memorisation Profilescited by 20
- 2023Threshold KNN-Shapley: A Linear-Time and Privacy-Friendly Approach to Data Valuationcited by 19
- 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Valuescited by 19
- 2024Efficient Data Shapley for Weighted Nearest Neighbor Algorithmscited by 18
- 2024An Economic Solution to Copyright Challenges of Generative AIcited by 18
- 2023Data Valuation and Detections in Federated Learningcited by 17
- 2023Rethinking Backdoor Attackscited by 14
- 2024Data Attribution for Diffusion Models: Timestep-induced Bias in Influence Estimationcited by 13
- 2023A Note on "Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithms"cited by 11
- 2023Selectivity Drives Productivity: Efficient Dataset Pruning for Enhanced Transfer Learningcited by 11
- 2022XInsight: eXplainable Data Analysis Through The Lens of Causalitycited by 9
- 2024Towards Explainable Artificial Intelligence (XAI): A Data Mining Perspectivecited by 9
- 2024One Sample Fits All: Approximating All Probabilistic Values Simultaneously and Efficientlycited by 9
- 20232D-Shapley: A Framework for Fragmented Data Valuationcited by 7
- 2025Fast Data Attribution for Text-to-Image Modelscited by 7
- 2023Data Optimization in Deep Learning: A Surveycited by 6
- 2024Sparse, Efficient and Explainable Data Attribution with DualXDAcited by 6
- 2024Task-Oriented GNNs Training on Large Knowledge Graphs for Accurate and Efficient Modelingcited by 5
- 2024Improve Knowledge Distillation via Label Revision and Data Selectioncited by 5
- 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selectioncited by 5
- 2024Final-Model-Only Data Attribution with a Unifying View of Gradient-Based Methodscited by 4
- 2024CHG Shapley: Efficient Data Valuation and Selection towards Trustworthy Machine Learningcited by 3
- 2026In-Run Data Shapley for Adam Optimizercited by 3
- 2024Uncertainty-aware Evaluation of Auxiliary Anomalies with the Expected Anomaly Posteriorcited by 2
- 2024Adversarial Attacks on Data Attributioncited by 2
- 2024Is Data Valuation Learnable and Interpretable?cited by 1
- 2024LossVal: Efficient Data Valuation for Neural Networkscited by 1
- 2025Towards Unified Attribution in Explainable AI, Data-Centric AI, and Mechanistic Interpretabilitycited by 1
- 2025Unifying and Optimizing Data Values for Selection via Sequential Decision-Makingcited by 1
- 2026Local Shapley: Model-Induced Locality and Optimal Reuse in Data Valuationcited by 1
- 2026Adalina: Adaptive Linear Approximation for the Shapley Value and Beyondcited by 1
- 2024Neural Dynamic Data Valuation via Stochastic State-Adjoint Trajectoriescited by 0
- 2024Data Valuation by Fusing Global and Local Statistical Informationcited by 0
- 2025Data Overvaluation Attack and Truthful Data Valuation in Federated Learningcited by 0
- 2025A Causal Framework for Aligning Image Quality Metrics and Deep Neural Network Robustnesscited by 0
- 2025Onto-Epistemological Analysis of AI Explanationscited by 0
- 2026Feature Attribution in Directed Acyclic Graphs Using Edge Interventioncited by 0
- 2026Localizing Global Discrepancies: Marginal Contributions and Contextual Anomaly Detectioncited by 0
- 2026Localizing Global Discrepancies: Marginal Contributions and Contextual Anomaly Detection28 Aug 2026
- 2026Feature Attribution in Directed Acyclic Graphs Using Edge Intervention13 Jun 2026
- 2026Adalina: Adaptive Linear Approximation for the Shapley Value and Beyond9 Apr 2026
- 2026Local Shapley: Model-Induced Locality and Optimal Reuse in Data Valuation4 Mar 2026
- 2026In-Run Data Shapley for Adam Optimizer30 Jan 2026
- 2025Fast Data Attribution for Text-to-Image Models13 Nov 2025
- 2025Onto-Epistemological Analysis of AI Explanations3 Oct 2025
- 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Values13 Jun 2025
- 2025DataMIL: Selecting Data for Robot Imitation Learning with Datamodels14 May 2025
- 2025A Causal Framework for Aligning Image Quality Metrics and Deep Neural Network Robustness4 Mar 2025
- 2025Unifying and Optimizing Data Values for Selection via Sequential Decision-Making6 Feb 2025
- 2025Data Overvaluation Attack and Truthful Data Valuation in Federated Learning1 Feb 2025
38 more
- 2025Towards Unified Attribution in Explainable AI, Data-Centric AI, and Mechanistic Interpretability31 Jan 2025
- 2024LossVal: Efficient Data Valuation for Neural Networks5 Dec 2024
- 2024Final-Model-Only Data Attribution with a Unifying View of Gradient-Based Methods5 Dec 2024
- 2024One Sample Fits All: Approximating All Probabilistic Values Simultaneously and Efficiently31 Oct 2024
- 2024Diffusion Attribution Score: Evaluating Training Data Influence in Diffusion Models24 Oct 2024
- 2024Adversarial Attacks on Data Attribution9 Sep 2024
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models4 Aug 2024
- 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selection24 Jun 2024
- 2024CHG Shapley: Efficient Data Valuation and Selection towards Trustworthy Machine Learning17 Jun 2024
- 2024Data Shapley in One Training Run16 Jun 2024
- 2024Causal Estimation of Memorisation Profiles6 Jun 2024
- 2024Is Data Valuation Learnable and Interpretable?3 Jun 2024
- 2024Data Valuation by Fusing Global and Local Statistical Information23 May 2024
- 2024Uncertainty-aware Evaluation of Auxiliary Anomalies with the Expected Anomaly Posterior22 May 2024
- 2024Rethinking Data Shapley for Data Selection Tasks: Misleads and Merits6 May 2024
- 2024Neural Dynamic Data Valuation via Stochastic State-Adjoint Trajectories30 Apr 2024
- 2024An Economic Solution to Copyright Challenges of Generative AI22 Apr 2024
- 2024Improve Knowledge Distillation via Label Revision and Data Selection3 Apr 2024
- 2024Task-Oriented GNNs Training on Large Knowledge Graphs for Accurate and Efficient Modeling9 Mar 2024
- 2024Sparse, Efficient and Explainable Data Attribution with DualXDA19 Feb 2024
- 2024Efficient Data Shapley for Weighted Nearest Neighbor Algorithms20 Jan 2024
- 2024Data Attribution for Diffusion Models: Timestep-induced Bias in Influence Estimation17 Jan 2024
- 2024Towards Explainable Artificial Intelligence (XAI): A Data Mining Perspective9 Jan 2024
- 2023The Journey, Not the Destination: How Data Guides Diffusion Models11 Dec 2023
- 2023Data Valuation and Detections in Federated Learning9 Nov 2023
- 2023Intriguing Properties of Data Attribution on Diffusion Models1 Nov 2023
- 2023Data Optimization in Deep Learning: A Survey25 Oct 2023
- 2023Selectivity Drives Productivity: Efficient Dataset Pruning for Enhanced Transfer Learning13 Oct 2023
- 2023Threshold KNN-Shapley: A Linear-Time and Privacy-Friendly Approach to Data Valuation30 Aug 2023
- 2023Rethinking Backdoor Attacks19 Jul 2023
- 2023OpenDataVal: a Unified Benchmark for Data Valuation18 Jun 2023
- 20232D-Shapley: A Framework for Fragmented Data Valuation18 Jun 2023
- 2023Data-OOB: Out-of-bag Estimate as a Simple and Efficient Data Value16 Apr 2023
- 2023A Note on "Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithms"9 Apr 2023
- 2023On the Variance of Neural Network Training with respect to Test Sets and Distributions4 Apr 2023
- 2023TRAK: Attributing Model Behavior at Scale24 Mar 2023
- 2022Training Data Influence Analysis and Estimation: A Survey9 Dec 2022
- 2022XInsight: eXplainable Data Analysis Through The Lens of Causality26 Jul 2022
- 2023Intriguing Properties of Data Attribution on Diffusion Modelscited by 47
- 2023The Journey, Not the Destination: How Data Guides Diffusion Modelscited by 42
- 2024Diffusion Attribution Score: Evaluating Training Data Influence in Diffusion Modelscited by 21
- 2024Data Attribution for Diffusion Models: Timestep-induced Bias in Influence Estimationcited by 13
- 2025Fast Data Attribution for Text-to-Image Modelscited by 7
- 2025Fast Data Attribution for Text-to-Image Models13 Nov 2025
- 2024Diffusion Attribution Score: Evaluating Training Data Influence in Diffusion Models24 Oct 2024
- 2024Data Attribution for Diffusion Models: Timestep-induced Bias in Influence Estimation17 Jan 2024
- 2023The Journey, Not the Destination: How Data Guides Diffusion Models11 Dec 2023
- 2023Intriguing Properties of Data Attribution on Diffusion Models1 Nov 2023
- 2023Rethinking Backdoor Attackscited by 14
- 2023Rethinking Backdoor Attacks19 Jul 2023
- 2023TRAK: Attributing Model Behavior at Scalecited by 273
- 2022Training Data Influence Analysis and Estimation: A Surveycited by 134
- 2024Data Shapley in One Training Runcited by 76
- 2023OpenDataVal: a Unified Benchmark for Data Valuationcited by 55
- 2023Data-OOB: Out-of-bag Estimate as a Simple and Efficient Data Valuecited by 38
- 2024Rethinking Data Shapley for Data Selection Tasks: Misleads and Meritscited by 29
- 2023On the Variance of Neural Network Training with respect to Test Sets and Distributionscited by 20
- 2024Causal Estimation of Memorisation Profilescited by 20
- 2023Threshold KNN-Shapley: A Linear-Time and Privacy-Friendly Approach to Data Valuationcited by 19
- 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Valuescited by 19
- 2024Efficient Data Shapley for Weighted Nearest Neighbor Algorithmscited by 18
- 2024An Economic Solution to Copyright Challenges of Generative AIcited by 18
12 most cited of 42 All 42 (JSON)
- 2026Localizing Global Discrepancies: Marginal Contributions and Contextual Anomaly Detection28 Aug 2026
- 2026Feature Attribution in Directed Acyclic Graphs Using Edge Intervention13 Jun 2026
- 2026Adalina: Adaptive Linear Approximation for the Shapley Value and Beyond9 Apr 2026
- 2026Local Shapley: Model-Induced Locality and Optimal Reuse in Data Valuation4 Mar 2026
- 2026In-Run Data Shapley for Adam Optimizer30 Jan 2026
- 2025Onto-Epistemological Analysis of AI Explanations3 Oct 2025
- 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Values13 Jun 2025
- 2025A Causal Framework for Aligning Image Quality Metrics and Deep Neural Network Robustness4 Mar 2025
- 2025Unifying and Optimizing Data Values for Selection via Sequential Decision-Making6 Feb 2025
- 2025Data Overvaluation Attack and Truthful Data Valuation in Federated Learning1 Feb 2025
- 2025Towards Unified Attribution in Explainable AI, Data-Centric AI, and Mechanistic Interpretability31 Jan 2025
- 2024LossVal: Efficient Data Valuation for Neural Networks5 Dec 2024
12 newest of 42 All 42 (JSON)