Data Shapley in One Training Run
Not on the map. Why?
Abstract (arXiv)
Data Shapley provides a principled framework for attributing data's contribution within machine learning contexts. However, existing approaches require re-training models on different data subsets, which is computationally intensive, foreclosing their application to large-scale models. Furthermore, they produce the same attribution score for any models produced by running the learning algorithm, meaning they cannot perform targeted attribution towards a specific model obtained from a single run of the algorithm. This paper introduces In-Run Data Shapley, which addresses these limitations by offering scalable data attribution for a target model of interest. In its most efficient implementation, our technique incurs negligible additional runtime compared to standard model training. This dramatic efficiency improvement makes it possible to perform data attribution for the foundation model pretraining stage for the first time. We present several case studies that offer fresh insights into pretraining data's contribution and discuss their implications for copyright in generative AI and pretraining data curation.
Key facts. Cited by 76 papers in the map (data through 7 Oct 2026). Builds on 53 of its 71 references that are in the map. First posted on arXiv 16 Jun 2024.
Builds on
53 of 71 references.
References · 53
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 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
- 2020Gradient Surgery for Multi-Task Learningcited by 1,210
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
- 2019Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Boundscited by 602
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2020What Neural Networks Memorize and Why: Discovering the Long Tail via Influence Estimationcited by 492
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 405
38 more
- 2023Studying Large Language Model Generalization with Influence Functionscited by 311
- 2019Towards Efficient Data Valuation Based on the Shapley Valuecited by 279
- 2023TRAK: Attributing Model Behavior at Scalecited by 273
- 2020Influence Functions in Deep Learning Are Fragilecited by 247
- 2022Datamodels: Predicting Predictions from Training Datacited by 186
- 2021Beta Shapley: a Unified and Noise-reduced Data Valuation Framework for Machine Learningcited by 162
- 2019Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithmscited by 147
- 2022If Influence Functions are the Answer, Then What is the Question?cited by 142
- 2020A Principled Approach to Data Valuation for Federated Learningcited by 127
- 2022Data Banzhaf: A Robust Data Valuation Framework for Machine Learningcited by 126
- 2023Foundation Models and Fair Usecited by 112
- 2021Sampling Permutations for Shapley Value Estimationcited by 101
- 2024What is Your Data Worth to GPT? LLM-Scale Data Valuation with Influence Functionscited by 88
- 2020A Distributional Framework for Data Valuationcited by 79
- 2023LAVA: Data Valuation without Pre-Specified Learning Algorithmscited by 69
- 2022Differentially Private Optimization on Large Model at Small Costcited by 65
- 2023OpenDataVal: a Unified Benchmark for Data Valuationcited by 55
- 2022Measuring the Effect of Training Data on Deep Learning Predictions via Randomized Experimentscited by 50
- 2022Scalable and Efficient Training of Large Convolutional Neural Networks with Differential Privacycited by 47
- 2019Emergent properties of the local geometry of neural loss landscapescited by 46
- 2020Scaling up Differentially Private Deep Learning with Fast Per-Example Gradient Clippingcited by 46
- 2021Revisiting Methods for Finding Influential Examplescited by 39
- 2020Efficient computation and analysis of distributional Shapley valuescited by 38
- 2023Data-OOB: Out-of-bag Estimate as a Simple and Efficient Data Valuecited by 38
- 2023Data Valuation Without Training of a Modelcited by 35
- 2020A Multilinear Sampling Algorithm to Estimate Shapley Valuescited by 33
- 2024Rethinking Data Shapley for Data Selection Tasks: Misleads and Meritscited by 29
- 2022Understanding Influence Functions and Datamodels via Harmonic Analysiscited by 23
- 2024Stochastic Amortization: A Unified Approach to Accelerate Feature and Data Attributioncited by 21
- 2021Incentivizing Collaboration in Machine Learning via Synthetic Data Rewardscited by 20
- 2022Fundamentals of Task-Agnostic Data Valuationcited by 19
- 2023Threshold KNN-Shapley: A Linear-Time and Privacy-Friendly Approach to Data Valuationcited by 19
- 2024Efficient Data Shapley for Weighted Nearest Neighbor Algorithmscited by 18
- 2024An Economic Solution to Copyright Challenges of Generative AIcited by 18
- 2019Efficient Per-Example Gradient Computations in Convolutional Neural Networkscited by 17
- 2019Estimation of the Shapley value by ergodic samplingcited by 15
- 2023A Note on "Towards Efficient Data Valuation Based on the Shapley Value''cited by 11
- 2023A Note on "Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithms"cited by 11
50 most cited of 53 All 53 (JSON)
Built on by
Citing papers per year: 2024: 4; 2025: 33; 2026: 39.
76 papers.
Sort:
By area:
- 2024Data Selection via Optimal Control for Language Modelscited by 38
- 2024Capturing the Temporal Dependence of Training Data Influencecited by 34
- 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Valuescited by 19
- 2026OPUS: Towards Efficient and Principled Data Selection in Large Language Model Pre-training in Every Iterationcited by 17
- 2025DataRater: Meta-Learned Dataset Curationcited by 13
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learningcited by 11
- 2026Shaping capabilities with token-level data filteringcited by 11
- 2025Interpretation Meets Safety: A Survey on Interpretation Methods and Tools for Improving LLM Safetycited by 10
- 2025Fairshare Data Pricing via Data Valuation for Large Language Modelscited by 7
- 2025Fast Data Attribution for Text-to-Image Modelscited by 7
- 2026MergeMix: Optimizing Mid-Training Data Mixtures via Learnable Model Mergingcited by 7
- 2025Train Once, Answer All: Many Pretraining Experiments for the Cost of Onecited by 6
38 more
- 2025CAMs as Shapley Value-based Explainerscited by 5
- 2025DATE-LM: Benchmarking Data Attribution Evaluation for Large Language Modelscited by 5
- 2024Targeted synthetic data generation for tabular data via hardness characterizationcited by 4
- 2025Scalable Valuation of Human Feedback through Provably Robust Model Alignmentcited by 4
- 2025Taming Hyperparameter Sensitivity in Data Attribution: Practical Selection Without Costly Retrainingcited by 4
- 2025Trade in Minutes! Rationality-Driven Agentic System for Quantitative Financial Tradingcited by 4
- 2026Motion Attribution for Video Generationcited by 4
- 2026How Faithful Is Trajectory-Based Data Attribution? Error Sources, Remedies, and Practical Guidelinescited by 4
- 2025Concept-TRAK: Understanding how diffusion models learn concepts through concept-level attributioncited by 3
- 2025MaxShapley: Towards Incentive-compatible Generative Search with Fair Context Attributioncited by 3
- 2026A Sustainable AI Economy Needs Data Deals That Work for Generatorscited by 3
- 2026OATS: Online Data Augmentation for Time Series Foundation Modelscited by 3
- 2026In-Run Data Shapley for Adam Optimizercited by 3
- 2025Do Data Valuations Make Good Data Prices?cited by 2
- 2025DICE: Data Influence Cascade in Decentralized Learningcited by 2
- 2025Newfluence: Boosting Model interpretability and Understanding in High Dimensionscited by 2
- 2025Z0-Inf: Zeroth Order Approximation for Data Influencecited by 2
- 2025Layer-Aware Influence for Online Data Valuation Estimationcited by 2
- 2026GAST: Gradient-aligned Sparse Tuning of Large Language Models with Data-layer Selectioncited by 2
- 2026ShapleyLaw: A Game-Theoretic Approach to Multilingual Scaling Lawscited by 2
- 2024LossVal: Efficient Data Valuation for Neural Networkscited by 1
- 2025Unifying and Optimizing Data Values for Selection via Sequential Decision-Makingcited by 1
- 2025KAIROS: Scalable Model-Agnostic Data Valuationcited by 1
- 2025For-Value: Efficient Forward-Only Data Valuation for finetuning LLMs and VLMscited by 1
- 2025AssayMatch: Learning to Select Data for Molecular Activity Modelscited by 1
- 2025Shapley-based Data Valuation for LLM Alignment via Sequential Preference Optimizationcited by 1
- 2026Imperfect Influence, Preserved Rankings: A Theory of TRAK for Data Attributioncited by 1
- 2026Local Shapley: Model-Induced Locality and Optimal Reuse in Data Valuationcited by 1
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuningcited by 1
- 2026In-Context Credit Assignment via the Corecited by 1
- 2026AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Trainingcited by 1
- 2026Measuring Task-Agnostic Training Data Influence Across Language Model Pretrainingcited by 1
- 2025Chunked Data Shapley: A Scalable Dataset Quality Assessment for Machine Learningcited by 0
- 2025Uncovering Scaling Laws for Large Language Models via Inverse Problemscited by 0
- 2025What Is The Performance Ceiling of My Classifier? Utilizing Category-Wise Influence Functions for Pareto Frontier Analysiscited by 0
- 2025f-INE: A Hypothesis Testing Framework for Estimating Influence under Training Randomnesscited by 0
- 2025Auditing Information Disclosure During Large-Scale Gradient-Based Training via Gradient Uniquenesscited by 0
- 2025Geometric Data Valuation via Leverage Scorescited by 0
50 most cited of 76 All 76 (JSON)
- 2026CoTrace: Data Recipes for Training Terminal Agents with Harness-Model Co-Evolution7 Oct 2026
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale30 Sep 2026
- 2026Not All Rollouts Are Worth Learning: On Trajectory Valuation for Post-Training Reinforcement Learning28 Sep 2026
- 2026Scalable Attribution and Control of Model Behavior During Training27 Sep 2026
- 2026Towards Scalable Data Diversification for Language Model Pretraining via Leverage Score Sampling26 Sep 2026
- 2026Traceprop: Training Data Attribution in a Single Pass26 Sep 2026
- 2026Past, Future, All at Once: Mitigating Stability-Plasticity Dilemma via Post-hoc JANUS Rectification17 Sep 2026
- 2026From Detrimental to Beneficial: Dynamic Influence-based Valuation and Editing23 Aug 2026
- 2026Measuring Task-Agnostic Training Data Influence Across Language Model Pretraining13 Aug 2026
- 2026Efficient Decentralized Multi-task Dataset Valuation via Model Merging3 Jul 2026
- 2026GRASP: Geometry-aware Residual Alignment for Scalable Pretraining Data Attribution5 Jun 2026
38 more
- 2026AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Training1 Jun 2026
- 2026De-attribute to Forget for LLM Unlearning29 May 2026
- 2026Efficient Banzhaf-Based Data Valuation for k-Nearest Neighbors Classification20 May 2026
- 2026Dynamic Shapley Computation20 May 2026
- 2026Shapley Neuron Values for Continual Learning: Which Neurons Matter Most?15 May 2026
- 2026On the Fragility of Data Attribution When Learning Is Distributed15 May 2026
- 2026How Faithful Is Trajectory-Based Data Attribution? Error Sources, Remedies, and Practical Guidelines12 May 2026
- 2026In-Context Credit Assignment via the Core7 May 2026
- 2026One Algorithm, Two Goals: Dual Scoring for Parameter and Data Selection in LLM Fine-Tuning7 May 2026
- 2026Data Attribution in Adaptive Learning6 Apr 2026
- 2026ShapE-GRPO: Shapley-Enhanced Reward Allocation for Multi-Candidate LLM Training31 Mar 2026
- 2026ShapleyLaw: A Game-Theoretic Approach to Multilingual Scaling Laws18 Mar 2026
- 2026GAST: Gradient-aligned Sparse Tuning of Large Language Models with Data-layer Selection10 Mar 2026
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuning8 Mar 2026
- 2026Local Shapley: Model-Induced Locality and Optimal Reuse in Data Valuation4 Mar 2026
- 2026Challenges in Enabling Private Data Valuation27 Feb 2026
- 2026OPUS: Towards Efficient and Principled Data Selection in Large Language Model Pre-training in Every Iteration5 Feb 2026
- 2026Imperfect Influence, Preserved Rankings: A Theory of TRAK for Data Attribution1 Feb 2026
- 2026In-Run Data Shapley for Adam Optimizer30 Jan 2026
- 2026SurrogateSHAP: Training-Free Contributor Attribution for Text-to-Image (T2I) Models29 Jan 2026
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filtering29 Jan 2026
- 2026Shaping capabilities with token-level data filtering29 Jan 2026
- 2026OATS: Online Data Augmentation for Time Series Foundation Models26 Jan 2026
- 2026MergeMix: Optimizing Mid-Training Data Mixtures via Learnable Model Merging25 Jan 2026
- 2026Select or Project? Evaluating Lower-dimensional Vectors for LLM Training Data Explanations23 Jan 2026
- 2026A Sustainable AI Economy Needs Data Deals That Work for Generators15 Jan 2026
- 2026Motion Attribution for Video Generation13 Jan 2026
- 2025Shapley-based Data Valuation for LLM Alignment via Sequential Preference Optimization12 Dec 2025
- 2025MaxShapley: Towards Incentive-compatible Generative Search with Fair Context Attribution5 Dec 2025
- 2025AssayMatch: Learning to Select Data for Molecular Activity Models20 Nov 2025
- 2025Online Data Curation for Object Detection via Marginal Contributions to Dataset-level Average Precision18 Nov 2025
- 2025Fast Data Attribution for Text-to-Image Models13 Nov 2025
- 2025Reliable and Private Utility Signaling for Data Markets11 Nov 2025
- 2025Error Estimate and Convergence Analysis for Data Valuation9 Nov 2025
- 2025Geometric Data Valuation via Leverage Scores3 Nov 2025
- 2025Layer-Aware Influence for Online Data Valuation Estimation14 Oct 2025
- 2025Z0-Inf: Zeroth Order Approximation for Data Influence13 Oct 2025
- 2025Auditing Information Disclosure During Large-Scale Gradient-Based Training via Gradient Uniqueness13 Oct 2025
50 newest of 76 All 76 (JSON)
- 2024Data Selection via Optimal Control for Language Modelscited by 38
- 2024Capturing the Temporal Dependence of Training Data Influencecited by 34
- 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Valuescited by 19
- 2026OPUS: Towards Efficient and Principled Data Selection in Large Language Model Pre-training in Every Iterationcited by 17
- 2025DataRater: Meta-Learned Dataset Curationcited by 13
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learningcited by 11
- 2026Shaping capabilities with token-level data filteringcited by 11
- 2025Fairshare Data Pricing via Data Valuation for Large Language Modelscited by 7
- 2026MergeMix: Optimizing Mid-Training Data Mixtures via Learnable Model Mergingcited by 7
- 2025Train Once, Answer All: Many Pretraining Experiments for the Cost of Onecited by 6
- 2025CAMs as Shapley Value-based Explainerscited by 5
- 2025DATE-LM: Benchmarking Data Attribution Evaluation for Large Language Modelscited by 5
12 most cited of 66 All 66 (JSON)
- 2026CoTrace: Data Recipes for Training Terminal Agents with Harness-Model Co-Evolution7 Oct 2026
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale30 Sep 2026
- 2026Not All Rollouts Are Worth Learning: On Trajectory Valuation for Post-Training Reinforcement Learning28 Sep 2026
- 2026Scalable Attribution and Control of Model Behavior During Training27 Sep 2026
- 2026Towards Scalable Data Diversification for Language Model Pretraining via Leverage Score Sampling26 Sep 2026
- 2026Traceprop: Training Data Attribution in a Single Pass26 Sep 2026
- 2026Past, Future, All at Once: Mitigating Stability-Plasticity Dilemma via Post-hoc JANUS Rectification17 Sep 2026
- 2026From Detrimental to Beneficial: Dynamic Influence-based Valuation and Editing23 Aug 2026
- 2026Measuring Task-Agnostic Training Data Influence Across Language Model Pretraining13 Aug 2026
- 2026Efficient Decentralized Multi-task Dataset Valuation via Model Merging3 Jul 2026
- 2026GRASP: Geometry-aware Residual Alignment for Scalable Pretraining Data Attribution5 Jun 2026
12 newest of 66 All 66 (JSON)