OPUS: Towards Efficient and Principled Data Selection in Large Language Model Pre-training in Every Iteration
Not on the map. Why?
Abstract (arXiv)
As high-quality public text approaches exhaustion, a phenomenon known as the Data Wall, pre-training is shifting from more tokens to better tokens. However, existing methods either rely on heuristic static filters that ignore training dynamics, or use dynamic yet optimizer-agnostic criteria based on raw gradients. We propose OPUS (Optimizer-induced Projected Utility Selection), a dynamic data selection framework that defines utility in the optimizer-induced update space. OPUS scores candidates by projecting their effective updates, shaped by modern optimizers, onto a target direction derived from a stable, in-distribution proxy. To ensure scalability, we employ Ghost technique with CountSketch for computational efficiency, and Boltzmann sampling for data diversity, incurring only 4.7% additional compute overhead. OPUS achieves remarkable results across diverse corpora, quality tiers, optimizers, and model scales. In pre-training of GPT-2 Large/XL on FineWeb and FineWeb-Edu with 30B tokens, OPUS outperforms industrial-level baselines and even full 200B-token training. Moreover, when combined with industrial-level static filters, OPUS further improves pre-training efficiency, even with lower-quality data. Furthermore, in continued pre-training of Qwen3-8B-Base on SciencePedia, OPUS achieves superior performance using only 0.5B tokens compared to full training with 3B tokens, demonstrating significant data efficiency gains in specialized domains.
Key facts. Cited by 17 papers in the map (data through 7 Oct 2026). Builds on 46 of its 63 references that are in the map. First posted on arXiv 5 Feb 2026.
Builds on
46 of 63 references.
References · 46
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2025Qwen3 Technical Reportcited by 8,195
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2024Qwen2 Technical Reportcited by 2,076
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,793
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
34 more
- 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,718
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
- 2019Adversarial NLI: A New Benchmark for Natural Language Understandingcited by 798
- 2018An Empirical Study of Example Forgetting during Deep Neural Network Learningcited by 669
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
- 2021Deep Learning on a Data Diet: Finding Important Examples Early in Trainingcited by 529
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2016A Corpus and Evaluation Framework for Deeper Understanding of Commonsense Storiescited by 413
- 2024DataComp-LM: In search of the next generation of training sets for language modelscited by 381
- 2023What Makes Good Data for Alignment? A Comprehensive Study of Automatic Data Selection in Instruction Tuningcited by 354
- 2018Not All Samples Are Created Equal: Deep Learning with Importance Samplingcited by 323
- 2023Data Selection for Language Models via Importance Resamplingcited by 322
- 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 269
- 2018Representer Point Selection for Explaining Deep Neural Networkscited by 187
- 2022Will we run out of data? Limits of LLM scaling based on human-generated datacited by 187
- 2024QuRating: Selecting High-Quality Data for Training Language Modelscited by 153
- 2022Training Data Influence Analysis and Estimation: A Surveycited by 134
- 2015Online Batch Selection for Faster Training of Neural Networkscited by 121
- 2020FastIF: Scalable Influence Functions for Efficient Model Interpretation and Debuggingcited by 120
- 2023MoDS: Model-oriented Data Selection for Instruction Tuningcited by 107
- 2024MATES: Model-Aware Data Selection for Efficient Pretraining with Data Influence Modelscited by 96
- 2024Perplexed by Perplexity: Perplexity-Based Data Pruning With Small Reference Modelscited by 83
- 2024Data Shapley in One Training Runcited by 76
- 2025Organize the Web: Constructing Domains Enhances Pre-Training Data Curationcited by 73
- 2024OlympicArena: Benchmarking Multi-discipline Cognitive Reasoning for Superintelligent AIcited by 70
- 2024SciAssess: Benchmarking LLM Proficiency in Scientific Literature Analysiscited by 40
- 2025Ultra-FineWeb: Efficient Data Filtering and Verification for High-Quality LLM Training Datacited by 35
- 2024Capturing the Temporal Dependence of Training Data Influencecited by 34
- 2024Most Influential Subset Selection: Challenges, Promises, and Beyondcited by 31
- 2025Group-Level Data Selection for Efficient Pretrainingcited by 12
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learningcited by 11
Built on by
Citing papers per year: 2026: 17.
17 papers.
Sort:
By area:
- 2026Socratic-SWE: Self-Evolving Coding Agents via Trace-Derived Agent Skillscited by 8
- 2026DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Modelscited by 4
- 2026Low-rank Optimization Trajectories Modeling for LLM RLVR Accelerationcited by 2
- 2026Test-Time Training with Next-Token Predictioncited by 2
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuningcited by 1
- 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selectioncited by 1
- 2026Data Agent: Learning to Select Data via End-to-End Dynamic Optimizationcited by 0
- 2026Let the Target Select for Itself: Data Selection via Target-Aligned Pathscited by 0
- 2026EvoPool: Evolutionary Programmatic Annotation for Label-Efficient Specialized Supervisioncited by 0
- 2026Reversible Foundations: Training a 120B Sparse MoE through State-Preserving Scalingcited by 0
- 2026INFUSER: Influence-Guided Self-Evolution Improves Reasoningcited by 0
- 2026Data-DPO: Direct Preference Optimization for Target Model Data Selection in LLM Post-Trainingcited by 0
5 more
- 2026Hierarchical Data Selection via Manifold Coverage and Sparse Feature Coverage in LLM Post-trainingcited by 0
- 2026Continual Data Unlearning in Diffusion Models via Transition-based Regularizationcited by 0
- 2026Towards Scalable Data Diversification for Language Model Pretraining via Leverage Score Samplingcited by 0
- 2026Relative Generalization Invariance of LLM Pretrainingcited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuningcited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026Towards Scalable Data Diversification for Language Model Pretraining via Leverage Score Sampling26 Sep 2026
- 2026Continual Data Unlearning in Diffusion Models via Transition-based Regularization26 Sep 2026
- 2026Hierarchical Data Selection via Manifold Coverage and Sparse Feature Coverage in LLM Post-training5 Aug 2026
- 2026Data-DPO: Direct Preference Optimization for Target Model Data Selection in LLM Post-Training5 Aug 2026
- 2026Test-Time Training with Next-Token Prediction19 Jun 2026
- 2026INFUSER: Influence-Guided Self-Evolution Improves Reasoning8 Jun 2026
- 2026Socratic-SWE: Self-Evolving Coding Agents via Trace-Derived Agent Skills5 Jun 2026
- 2026Reversible Foundations: Training a 120B Sparse MoE through State-Preserving Scaling5 Jun 2026
- 2026EvoPool: Evolutionary Programmatic Annotation for Label-Efficient Specialized Supervision1 Jun 2026
- 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selection27 May 2026
5 more
- 2026Let the Target Select for Itself: Data Selection via Target-Aligned Paths10 May 2026
- 2026Low-rank Optimization Trajectories Modeling for LLM RLVR Acceleration13 Apr 2026
- 2026DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Models27 Mar 2026
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuning8 Mar 2026
- 2026Data Agent: Learning to Select Data via End-to-End Dynamic Optimization8 Mar 2026
- 2026DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Modelscited by 4
- 2026Low-rank Optimization Trajectories Modeling for LLM RLVR Accelerationcited by 2
- 2026Test-Time Training with Next-Token Predictioncited by 2
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuningcited by 1
- 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selectioncited by 1
- 2026Data Agent: Learning to Select Data via End-to-End Dynamic Optimizationcited by 0
- 2026Let the Target Select for Itself: Data Selection via Target-Aligned Pathscited by 0
- 2026EvoPool: Evolutionary Programmatic Annotation for Label-Efficient Specialized Supervisioncited by 0
- 2026Hierarchical Data Selection via Manifold Coverage and Sparse Feature Coverage in LLM Post-trainingcited by 0
- 2026Towards Scalable Data Diversification for Language Model Pretraining via Leverage Score Samplingcited by 0
- 2026Relative Generalization Invariance of LLM Pretrainingcited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuningcited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026Towards Scalable Data Diversification for Language Model Pretraining via Leverage Score Sampling26 Sep 2026
- 2026Hierarchical Data Selection via Manifold Coverage and Sparse Feature Coverage in LLM Post-training5 Aug 2026
- 2026Test-Time Training with Next-Token Prediction19 Jun 2026
- 2026EvoPool: Evolutionary Programmatic Annotation for Label-Efficient Specialized Supervision1 Jun 2026
- 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selection27 May 2026
- 2026Let the Target Select for Itself: Data Selection via Target-Aligned Paths10 May 2026
- 2026Low-rank Optimization Trajectories Modeling for LLM RLVR Acceleration13 Apr 2026
- 2026DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Models27 Mar 2026
- 2026Filter-then-Weight: Online Data Selection and Reweighting for LLM Fine-Tuning8 Mar 2026
- 2026Data Agent: Learning to Select Data via End-to-End Dynamic Optimization8 Mar 2026