DataRater: Meta-Learned Dataset Curation
Not on the map. Why?
Abstract (arXiv)
The quality of foundation models depends heavily on their training data. Consequently, great efforts have been put into dataset curation. Yet most approaches rely on manual tuning of coarse-grained mixtures of large buckets of data, or filtering by hand-crafted heuristics. An approach that is ultimately more scalable (let alone more satisfying) is to learn which data is actually valuable for training. This type of meta-learning could allow more sophisticated, fine-grained, and effective curation. Our proposed DataRater is an instance of this idea. It estimates the value of training on any particular data point. This is done by meta-learning using `meta-gradients', with the objective of improving training efficiency on held out data. In extensive experiments across a range of model scales and datasets, we find that using our DataRater to filter data is highly effective, resulting in significantly improved compute efficiency.
Key facts. Cited by 13 papers in the map (data through 13 Oct 2026). Builds on 41 of its 65 references that are in the map. First posted on arXiv 23 May 2025.
Builds on
41 of 65 references.
References · 41
- 2017Attention Is All You Needcited by 60,106
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,588
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,549
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,231
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,798
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,053
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,843
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,379
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,805
- 2018DARTS: Differentiable Architecture Searchcited by 1,789
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,729
- 2021Scaling Language Models: Methods, Analysis & Insights from Training Gophercited by 1,201
29 more
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,126
- 2023LIMA: Less Is More for Alignmentcited by 1,091
- 2016Bag of Tricks for Efficient Text Classificationcited by 872
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 621
- 2017Evolution Strategies as a Scalable Alternative to Reinforcement Learningcited by 562
- 2019CCNet: Extracting High Quality Monolingual Datasets from Web Crawl Datacited by 544
- 2024Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Researchcited by 477
- 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 391
- 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 359
- 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 303
- 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 234
- 2016Hyperparameter optimization with approximate gradientcited by 232
- 2024A Survey on Data Selection for Language Modelscited by 232
- 2018Meta-Gradient Reinforcement Learningcited by 158
- 2022A framework for bilevel optimization that enables stochastic and global variance reduction algorithmscited by 99
- 2024Perplexed by Perplexity: Perplexity-Based Data Pruning With Small Reference Modelscited by 84
- 2024Data Shapley in One Training Runcited by 79
- 2024SEAL: Safety-enhanced Aligned LLM Fine-tuning via Bilevel Data Selectioncited by 64
- 2019Optimizing Data Usage via Differentiable Rewardscited by 48
- 2023Meta-Learning Online Adaptation of Language Modelscited by 43
- 2024Data Selection via Optimal Control for Language Modelscited by 38
- 2021Podracer architectures for scalable Reinforcement Learningcited by 37
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweightingcited by 35
- 2025Optimizing ML Training with Metagradient Descentcited by 23
- 2025Predictive Data Selection: The Data That Predicts Is the Data That Teachescited by 20
- 2025Dynamic Loss-Based Sample Reweighting for Improved Large Language Model Pretrainingcited by 16
- 2021Robust Optimization for Multilingual Translation with Imbalanced Datacited by 15
- 2024Data, Data Everywhere: A Guide for Pretraining Dataset Constructioncited by 11
- 2025Scalable Meta-Learning via Mixed-Mode Differentiationcited by 4
Built on by
Citing papers per year: 2025: 5; 2026: 8.
13 papers.
Sort:
By area:
- 2025Language Self-Play For Data-Free Trainingcited by 42
- 2025Language Models Improve When Pretraining Data Matches Target Taskscited by 18
- 2026Towards Agentic Intelligence for Materials Sciencecited by 3
- 2026Synthetic Data for any Differentiable Targetcited by 2
- 2025Holdout-Loss-Based Data Selection for LLM Finetuning via In-Context Learningcited by 1
- 2025Alchemist: Unlocking Efficiency in Text-to-Image Model Training via Meta-Gradient Data Selectioncited by 1
- 2026Generalization Guarantees on Data-Driven Tuning of Gradient Descent with Langevin Updatescited by 1
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Trainingcited by 1
- 2025Uncovering Competency Gaps in Large Language Models and Their Benchmarkscited by 0
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filteringcited by 0
- 2026Tula: Optimizing Time, Cost, and Generalization in Distributed Large-Batch Trainingcited by 0
- 2026AMO: Operator-level Adaptive Muon Orthogonalizationcited by 0
- 2026Scalable, Transferable Meta-network for Data Selection Requires a Different Loss (and Why the Obvious Choice is Problematic)1 Oct 2026
- 2026AMO: Operator-level Adaptive Muon Orthogonalization18 May 2026
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Training2 May 2026
- 2026Generalization Guarantees on Data-Driven Tuning of Gradient Descent with Langevin Updates13 Apr 2026
- 2026Synthetic Data for any Differentiable Target9 Apr 2026
- 2026Tula: Optimizing Time, Cost, and Generalization in Distributed Large-Batch Training18 Mar 2026
- 2026Towards Agentic Intelligence for Materials Science29 Jan 2026
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filtering29 Jan 2026
- 2025Alchemist: Unlocking Efficiency in Text-to-Image Model Training via Meta-Gradient Data Selection18 Dec 2025
- 2025Uncovering Competency Gaps in Large Language Models and Their Benchmarks6 Dec 2025
- 2025Holdout-Loss-Based Data Selection for LLM Finetuning via In-Context Learning16 Oct 2025
- 2025Language Self-Play For Data-Free Training9 Sep 2025
1 more
- 2025Language Self-Play For Data-Free Trainingcited by 42
- 2026Synthetic Data for any Differentiable Targetcited by 2
- 2026Synthetic Data for any Differentiable Target9 Apr 2026
- 2025Language Self-Play For Data-Free Training9 Sep 2025
- 2026Towards Agentic Intelligence for Materials Sciencecited by 3
- 2026Towards Agentic Intelligence for Materials Science29 Jan 2026
- 2025Language Models Improve When Pretraining Data Matches Target Taskscited by 18
- 2026Generalization Guarantees on Data-Driven Tuning of Gradient Descent with Langevin Updatescited by 1
- 2025Uncovering Competency Gaps in Large Language Models and Their Benchmarkscited by 0
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filteringcited by 0
- 2026Tula: Optimizing Time, Cost, and Generalization in Distributed Large-Batch Trainingcited by 0
- 2026AMO: Operator-level Adaptive Muon Orthogonalizationcited by 0
- 2026Scalable, Transferable Meta-network for Data Selection Requires a Different Loss (and Why the Obvious Choice is Problematic)cited by 0
- 2026Scalable, Transferable Meta-network for Data Selection Requires a Different Loss (and Why the Obvious Choice is Problematic)1 Oct 2026
- 2026AMO: Operator-level Adaptive Muon Orthogonalization18 May 2026
- 2026Generalization Guarantees on Data-Driven Tuning of Gradient Descent with Langevin Updates13 Apr 2026
- 2026Tula: Optimizing Time, Cost, and Generalization in Distributed Large-Batch Training18 Mar 2026
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filtering29 Jan 2026
- 2025Uncovering Competency Gaps in Large Language Models and Their Benchmarks6 Dec 2025
- 2025Language Models Improve When Pretraining Data Matches Target Tasks16 Jul 2025