GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Training
Not on the map. Why?
Abstract (arXiv)
The great success of modern machine learning models on large datasets is contingent on extensive computational resources with high financial and environmental costs. One way to address this is by extracting subsets that generalize on par with the full data. In this work, we propose a general framework, GRAD-MATCH, which finds subsets that closely match the gradient of the training or validation set. We find such subsets effectively using an orthogonal matching pursuit algorithm. We show rigorous theoretical and convergence guarantees of the proposed algorithm and, through our extensive experiments on real-world datasets, show the effectiveness of our proposed framework. We show that GRAD-MATCH significantly and consistently outperforms several recent data-selection algorithms and achieves the best accuracy-efficiency trade-off. GRAD-MATCH is available as a part of the CORDS toolkit: urlhttps://github.com/decile-team/cords.
Key facts. Cited by 269 papers in the map (data through 7 Oct 2026). Builds on 16 of its 41 references that are in the map. First posted on arXiv 27 Feb 2021.
Builds on
16 of 41 references.
References · 16
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,242
- 2018An Empirical Study of Example Forgetting during Deep Neural Network Learningcited by 669
- 2019Green AIcited by 620
- 2019Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Boundscited by 602
- 2019Selection via Proxy: Efficient Data Selection for Deep Learningcited by 337
- 2020GLISTER: Generalization based Data Subset Selection for Efficient and Robust Learningcited by 224
- 2020The Cost of Training NLP Models: A Concise Overviewcited by 97
- 2018Bayesian Coreset Construction via Greedy Iterative Geodesic Ascentcited by 68
- 2019Learning From Less Data: A Unified Data Subset Selection and Active Learning Framework for Computer Visioncited by 64
4 more
- 2018Highly Scalable Deep Learning Training System with Mixed-Precision: Training ImageNet in Four Minutescited by 63
- 2016Restricted Strong Convexity Implies Weak Submodularitycited by 51
- 2019E2-Train: Training State-of-the-art CNNs with Over 80% Energy Savingscited by 36
- 2020Coresets for Robust Training of Neural Networks against Noisy Labelscited by 24
Built on by
Citing papers per year: 2021: 4; 2022: 19; 2023: 50; 2024: 55; 2025: 75; 2026: 66.
269 papers.
Sort:
By area:
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 385
- 2023Data Selection for Language Models via Importance Resamplingcited by 322
- 2022DeepCore: A Comprehensive Library for Coreset Selection in Deep Learningcited by 146
- 2022Coverage-centric Coreset Selection for High Pruning Ratescited by 115
- 2024What is in Your Safe Data? Identifying Benign Data that Breaks Safetycited by 114
- 2023Finding Support Examples for In-Context Learningcited by 104
- 2022Condensing Graphs via One-Step Gradient Matchingcited by 100
- 2022Quality Not Quantity: On the Interaction between Dataset Design and Robustness of CLIPcited by 97
- 2021GCR: Gradient Coreset Based Replay Buffer Selection For Continual Learningcited by 95
- 2023InfoBatch: Lossless Training Speed Up by Unbiased Dynamic Data Pruningcited by 90
- 2023Data Distillation: A Surveycited by 86
38 more
- 2023Anchor Points: Benchmarking Models with Much Fewer Examplescited by 84
- 2024Perplexed by Perplexity: Perplexity-Based Data Pruning With Small Reference Modelscited by 83
- 2022Zeus: Understanding and Optimizing GPU Energy Consumption of DNN Trainingcited by 78
- 2023Data Pruning via Moving-one-Sample-outcited by 67
- 2023Dataset Quantizationcited by 47
- 2023Towards Sustainable Learning: Coresets for Data-efficient Deep Learningcited by 44
- 2024SmallToLarge (S2L): Scalable Data Selection for Fine-tuning Large Language Models by Summarizing Training Trajectories of Small Modelscited by 43
- 2023Spanning Training Progress: Temporal Dual-Depth Scoring (TDDS) for Enhanced Dataset Pruningcited by 38
- 2023Loss-Curvature Matching for Dataset Selection and Condensationcited by 32
- 2024Get more for less: Principled Data Selection for Warming Up Fine-Tuning in LLMscited by 32
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
- 2024TAGCOS: Task-agnostic Gradient Clustered Coreset Selection for Instruction Tuning Datacited by 28
- 2025A Coreset Selection of Coreset Selection Literature: Introduction and Recent Advancescited by 28
- 2021TALISMAN: Targeted Active Learning for Object Detection with Rare Classes and Slices using Submodular Mutual Informationcited by 27
- 2023Repeated Random Sampling for Minimizing the Time-to-Accuracy of Learningcited by 26
- 2022AUTOMATA: Gradient Based Data Subset Selection for Compute-Efficient Hyper-parameter Tuningcited by 22
- 2023Data-Efficient Contrastive Self-supervised Learning: Most Beneficial Examples for Supervised Learning Contribute the Leastcited by 22
- 2024SelMatch: Effectively Scaling Up Dataset Distillation via Selection-Based Initialization and Partial Updates by Trajectory Matchingcited by 22
- 2022Mitigating Dataset Bias by Using Per-sample Gradientcited by 21
- 2023Regularizing Second-Order Influences for Continual Learningcited by 21
- 2022Infinite Recommendation Networks: A Data-Centric Approachcited by 20
- 2023You Only Condense Once: Two Rules for Pruning Condensed Datasetscited by 20
- 2023Robust Data Pruning under Label Noise via Maximizing Re-labeling Accuracycited by 20
- 2024A CLIP-Powered Framework for Robust and Generalizable Data Selectioncited by 20
- 2023Practical Differentially Private Hyperparameter Tuning with Subsamplingcited by 19
- 2024SHED: Shapley-Based Automated Dataset Refinement for Instruction Fine-Tuningcited by 19
- 2025Data Whisperer: Efficient Data Selection for Task-Specific LLM Fine-Tuning via Few-Shot In-Context Learningcited by 19
- 2024Importance-Aware Adaptive Dataset Distillationcited by 18
- 2024Pruning then Reweighting: Towards Data-Efficient Training of Diffusion Modelscited by 17
- 2026OPUS: Towards Efficient and Principled Data Selection in Large Language Model Pre-training in Every Iterationcited by 17
- 2022CVNets: High Performance Library for Computer Visioncited by 16
- 2025Hard Examples Are All You Need: Maximizing GRPO Post-Training Under Annotation Budgetscited by 16
- 2023Iterative Forward Tuning Boosts In-Context Learning in Language Modelscited by 15
- 2023Efficient Adversarial Contrastive Learning via Robustness-Aware Coreset Selectioncited by 14
- 2023Provable Data Subset Selection For Efficient Neural Network Trainingcited by 14
- 2024DELIFT: Data Efficient Language model Instruction Fine Tuningcited by 14
- 2025Data Pruning by Information Maximizationcited by 14
- 2023Less is More: Data Pruning for Faster Adversarial Trainingcited by 13
50 most cited of 269 All 269 (JSON)
- 2026SALD: Self-Referenced Advantage Learning for Diffusion Models1 Oct 2026
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026Learning What to Forget: Distributional Unlearning for LLM Representation Spaces30 Sep 2026
- 2026Not All Rollouts Are Worth Learning: On Trajectory Valuation for Post-Training Reinforcement Learning28 Sep 2026
- 2026GraphSelect for Budgeted Representation Selection in Multimodal Graph Inference27 Sep 2026
- 2026Refreshing Less, Selecting Better: Reusing Stale Gradient Features for Efficient Influence-Based Data Selection26 Sep 2026
- 2026Let Training Guide Selection: Online Synthetic Data Filtering via Real-Anchored Utility24 Sep 2026
- 2026Useful to Whom? Sample Value Is Defined Only Relative to the Learner22 Sep 2026
- 2026CRISP: Scalable Importance-Stratified Coresets for Imbalanced Tabular Learning22 Sep 2026
- 2026Are Coreset Selection Methods Worth Their Cost?19 Sep 2026
- 2026MemRiskBench: Trace-Aware Risk-Preserving Evaluation for Long-Horizon LLM Agents14 Sep 2026
- 2026When Safety Routing Breaks: Understanding Alignment Fragility under Benign Fine-Tuning1 Sep 2026
38 more
- 2026Learning Dynamics of Logits Debiasing for Long-Tailed Semi-Supervised Learning31 Aug 2026
- 2026What Memory Composition Does Not Tell Us About Anomaly Detection24 Aug 2026
- 2026From Detrimental to Beneficial: Dynamic Influence-based Valuation and Editing23 Aug 2026
- 2026Measuring Structured Predictability in Neural Training Dynamics: A Cross-Regime Study16 Aug 2026
- 2026When Do Surrogate Updates Improve Decisions? A Local Theory of Trajectory-Wise Transfer2 Aug 2026
- 2026ALIVE: Warnings Before Exclusion in Budgeted Multi-Source Learning31 Jul 2026
- 2026First-order Constrained Trilevel Optimization Over Distributed Networks for Robust Coreset Selection30 Jul 2026
- 2026PPL-Factory: Task-Aware and Budget-Aware Data Selection from Language Modeling to Reasoning20 Jul 2026
- 2026Selectivity Drives Efficiency: Dataset Pruning for Visual Place Recognition16 Jul 2026
- 2026Distribution-Aware Robust Bilevel Optimization: Quantile-Guided Huber Updates in Two-Timescale Stochastic Approximation21 Jun 2026
- 2026DRIFT: Data Selection for LLM Instruction Tuning via On-Policy Attribution16 Jun 2026
- 2026Rethinking Dataset Distillation for Classification: Do Distilled Sets Outperform Coresets?16 Jun 2026
- 2026TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins16 Jun 2026
- 2026Selecting Samples on Graphs: A Unified Dataset Pruning Framework for Lossless Training Acceleration11 Jun 2026
- 2026RCAP: Robust, Class-Aware, Probabilistic Dynamic Dataset Pruning10 Jun 2026
- 2026OrderDP: A Theoretically Guaranteed Lossless Dynamic Data Pruning Framework7 Jun 2026
- 2026Minibatch Selection for Language Models via Partition Matroid Constrained Gradient Matching6 Jun 2026
- 2026Data-efficient flood depth prediction through domain-aware coreset selection and tabular foundation models3 Jun 2026
- 2026AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Training1 Jun 2026
- 2026Efficient Test-Time Finetuning of LLMs via Convex Reconstruction and Gradient Caching28 May 2026
- 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selection27 May 2026
- 2026PlanE: Meta Planning of Data, Tuning, and Inference for Extractive-based LLMs22 May 2026
- 2026Multimodal Distribution Matching for Vision-Language Dataset Distillation22 May 2026
- 2026Label-Efficient Dataset Pruning via Semi-Supervised Pseudo-Labeling22 May 2026
- 2026CAST: Collapse-Aware multi-Scale Topology Fusion for Multimodal Coreset Selection12 May 2026
- 2026Let the Target Select for Itself: Data Selection via Target-Aligned Paths10 May 2026
- 2026Disagreement-Regularized Importance Sampling for Adversarial Label Corruption8 May 2026
- 2026Fair Dataset Distillation via Cross-Group Barycenter Alignment30 Apr 2026
- 2026Detecting and Suppressing Reward Hacking with Gradient Fingerprints17 Apr 2026
- 2026CoLA: A Choice Leakage Attack Framework to Expose Privacy Risks in Subset Training14 Apr 2026
- 2026UniPROT: Uniform Prototype Selection via Partial Optimal Transport with Submodular Guarantees13 Apr 2026
- 2026Data Warmup: Complexity-Aware Curricula for Efficient Diffusion Training8 Apr 2026
- 2026Beyond Loss Values: Robust Dynamic Pruning via Loss Trajectory Alignment8 Apr 2026
- 2026Training data generation for context-dependent rubric-based short answer grading30 Mar 2026
- 2026Is One-Shot In-Context Learning Helpful for Data Selection in Task-Specific Fine-Tuning of Multimodal LLMs?30 Mar 2026
- 2026PruneFuse: Efficient Data Selection via Weight Pruning and Network Fusion27 Mar 2026
- 2026Gradient-Informed Temporal Sampling Improves Rollout Accuracy in PDE Surrogate Training18 Mar 2026
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuning8 Mar 2026
50 newest of 269 All 269 (JSON)
- 2024What is in Your Safe Data? Identifying Benign Data that Breaks Safetycited by 114
- 2025GRAIT: Gradient-Driven Refusal-Aware Instruction Tuning for Effective Hallucination Mitigationcited by 11
- 2025LLM Unlearning Reveals a Stronger-Than-Expected Coreset Effect in Current Benchmarkscited by 11
- 2025UPCORE: Utility-Preserving Coreset Selection for Balanced Unlearningcited by 7
- 2025Distributional Machine Unlearning via Selective Data Removalcited by 5
- 2026Learning What to Forget: Distributional Unlearning for LLM Representation Spacescited by 0
- 2026Learning What to Forget: Distributional Unlearning for LLM Representation Spaces30 Sep 2026
- 2025Distributional Machine Unlearning via Selective Data Removal20 Jul 2025
- 2025LLM Unlearning Reveals a Stronger-Than-Expected Coreset Effect in Current Benchmarks14 Apr 2025
- 2025UPCORE: Utility-Preserving Coreset Selection for Balanced Unlearning20 Feb 2025
- 2025GRAIT: Gradient-Driven Refusal-Aware Instruction Tuning for Effective Hallucination Mitigation9 Feb 2025
- 2024What is in Your Safe Data? Identifying Benign Data that Breaks Safety1 Apr 2024
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
- 2026Detecting and Suppressing Reward Hacking with Gradient Fingerprintscited by 4
- 2025Holdout-Loss-Based Data Selection for LLM Finetuning via In-Context Learningcited by 1
- 2026ZPD Detector: Data Selection via Capability-Difficulty Alignment for Large Language Modelscited by 0
- 2026When Safety Routing Breaks: Understanding Alignment Fragility under Benign Fine-Tuningcited by 0
- 2026When Safety Routing Breaks: Understanding Alignment Fragility under Benign Fine-Tuning1 Sep 2026
- 2026Detecting and Suppressing Reward Hacking with Gradient Fingerprints17 Apr 2026
- 2026ZPD Detector: Data Selection via Capability-Difficulty Alignment for Large Language Models16 Jan 2026
- 2025Holdout-Loss-Based Data Selection for LLM Finetuning via In-Context Learning16 Oct 2025
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models4 Aug 2024
- 2024Pruning then Reweighting: Towards Data-Efficient Training of Diffusion Modelscited by 17
- 2025Fast Data Attribution for Text-to-Image Modelscited by 7
- 2024Data Pruning in Generative Diffusion Modelscited by 3
- 2026SALD: Self-Referenced Advantage Learning for Diffusion Modelscited by 0
- 2026SALD: Self-Referenced Advantage Learning for Diffusion Models1 Oct 2026
- 2025Fast Data Attribution for Text-to-Image Models13 Nov 2025
- 2024Data Pruning in Generative Diffusion Models19 Nov 2024
- 2024Pruning then Reweighting: Towards Data-Efficient Training of Diffusion Models27 Sep 2024
- 2025Hard Examples Are All You Need: Maximizing GRPO Post-Training Under Annotation Budgetscited by 16
- 2026Position: Modular Memory is the Key to Continual Learning Agentscited by 7
- 2025EPiC: Towards Lossless Speedup for Reasoning Training through Edge-Preserving CoT Condensationcited by 2
- 2026PPL-Factory: Task-Aware and Budget-Aware Data Selection from Language Modeling to Reasoningcited by 0
- 2026PPL-Factory: Task-Aware and Budget-Aware Data Selection from Language Modeling to Reasoning20 Jul 2026
- 2026Position: Modular Memory is the Key to Continual Learning Agents2 Mar 2026
- 2025Hard Examples Are All You Need: Maximizing GRPO Post-Training Under Annotation Budgets15 Aug 2025
- 2025EPiC: Towards Lossless Speedup for Reasoning Training through Edge-Preserving CoT Condensation4 Jun 2025
- 2025MM-GEN: Enhancing Task Performance Through Targeted Multimodal Data Curationcited by 4
- 2024Error-driven Data-efficient Large Multimodal Model Tuningcited by 2
- 2026Is One-Shot In-Context Learning Helpful for Data Selection in Task-Specific Fine-Tuning of Multimodal LLMs?cited by 0
- 2026CAST: Collapse-Aware multi-Scale Topology Fusion for Multimodal Coreset Selectioncited by 0
- 2026CAST: Collapse-Aware multi-Scale Topology Fusion for Multimodal Coreset Selection12 May 2026
- 2026Is One-Shot In-Context Learning Helpful for Data Selection in Task-Specific Fine-Tuning of Multimodal LLMs?30 Mar 2026
- 2025MM-GEN: Enhancing Task Performance Through Targeted Multimodal Data Curation7 Jan 2025
- 2024Error-driven Data-efficient Large Multimodal Model Tuning20 Dec 2024
- 2025Offline Behavioral Data Selection20 Dec 2025
- 2025Fewer May Be Better: Enhancing Offline Reinforcement Learning with Reduced Dataset26 Feb 2025
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 385
- 2023Data Selection for Language Models via Importance Resamplingcited by 322
- 2022DeepCore: A Comprehensive Library for Coreset Selection in Deep Learningcited by 146
- 2022Coverage-centric Coreset Selection for High Pruning Ratescited by 115
- 2023Finding Support Examples for In-Context Learningcited by 104
- 2022Condensing Graphs via One-Step Gradient Matchingcited by 100
- 2022Quality Not Quantity: On the Interaction between Dataset Design and Robustness of CLIPcited by 97
- 2021GCR: Gradient Coreset Based Replay Buffer Selection For Continual Learningcited by 95
- 2023InfoBatch: Lossless Training Speed Up by Unbiased Dynamic Data Pruningcited by 90
- 2023Data Distillation: A Surveycited by 86
- 2023Anchor Points: Benchmarking Models with Much Fewer Examplescited by 84
12 most cited of 241 All 241 (JSON)
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026Not All Rollouts Are Worth Learning: On Trajectory Valuation for Post-Training Reinforcement Learning28 Sep 2026
- 2026GraphSelect for Budgeted Representation Selection in Multimodal Graph Inference27 Sep 2026
- 2026Refreshing Less, Selecting Better: Reusing Stale Gradient Features for Efficient Influence-Based Data Selection26 Sep 2026
- 2026Useful to Whom? Sample Value Is Defined Only Relative to the Learner22 Sep 2026
- 2026CRISP: Scalable Importance-Stratified Coresets for Imbalanced Tabular Learning22 Sep 2026
- 2026Are Coreset Selection Methods Worth Their Cost?19 Sep 2026
- 2026Learning Dynamics of Logits Debiasing for Long-Tailed Semi-Supervised Learning31 Aug 2026
- 2026What Memory Composition Does Not Tell Us About Anomaly Detection24 Aug 2026
- 2026From Detrimental to Beneficial: Dynamic Influence-based Valuation and Editing23 Aug 2026
- 2026Measuring Structured Predictability in Neural Training Dynamics: A Cross-Regime Study16 Aug 2026
- 2026When Do Surrogate Updates Improve Decisions? A Local Theory of Trajectory-Wise Transfer2 Aug 2026
12 newest of 241 All 241 (JSON)