AdaSTaR: Adaptive Data Sampling for Training Self-Taught Reasoners
On the map: LLM reasoning → Test-time compute and efficient thinking
also in 1 more
Abstract (arXiv)
Self-Taught Reasoners (STaR), synonymously known as Rejection sampling Fine-Tuning (RFT), is an integral part of the training pipeline of self-improving reasoning Language Models (LMs). The self-improving mechanism often employs random observation (data) sampling. However, this results in trained observation imbalance; inefficiently over-training on solved examples while under-training on challenging ones. In response, we introduce Adaptive STaR (AdaSTaR), a novel algorithm that rectifies this by integrating two adaptive sampling principles: (1) Adaptive Sampling for Diversity: promoting balanced training across observations, and (2) Adaptive Sampling for Curriculum: dynamically adjusting data difficulty to match the model's evolving strength. Across six benchmarks, AdaSTaR achieves best test accuracy in all instances (6/6) and reduces training FLOPs by an average of 58.6% against an extensive list of baselines. These improvements in performance and efficiency generalize to different pre-trained LMs and larger models, paving the way for more efficient and effective self-improving LMs.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 59 of its 75 references that are in the map. First posted on arXiv 22 May 2025.
Builds on
59 of 75 references.
Closest ancestors · 9
Same direction, newest first
- 2025Nemotron-H: A Family of Accurate and Efficient Hybrid Mamba-Transformer Models4 Apr
- 2025Stop Overthinking: A Survey on Efficient Reasoning for Large Language Models20 Mar
- 2025EXAONE Deep: Reasoning Enhanced Language Models16 Mar
- 2024B-STaR: Monitoring and Balancing Exploration and Exploitation in Self-Taught Reasoners23 Dec
- 2024Self-Training Meets Consistency: Improving LLMs' Reasoning with Consistency-Driven Rationale Evaluation10 Nov
- 2024ReGenesis: LLMs can Grow into Reasoning Generalists via Self-Improvement3 Oct
- 2024V-STaR: Training Verifiers for Self-Taught Reasoners9 Feb
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Models17 May
- 2022STaR: Bootstrapping Reasoning With Reasoning28 Mar
Other references · 50
- 2017Attention Is All You Needcited by 60,361
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2023Let's Verify Step by Stepcited by 3,789
38 more
- 2023Qwen Technical Reportcited by 3,344
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 991
- 2022Solving math word problems with process- and outcome-based feedbackcited by 901
- 2021Are NLP Models really able to Solve Simple Math Word Problems?cited by 878
- 2019Adversarial NLI: A New Benchmark for Natural Language Understandingcited by 807
- 2020Contrastive Learning with Hard Negative Samplescited by 628
- 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 427
- 2023Larger language models do in-context learning differentlycited by 394
- 2023Beyond Human Data: Scaling Self-Training for Problem-Solving with Language Modelscited by 349
- 2022Large Language Models Are Reasoning Teacherscited by 329
- 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 301
- 2019On The Power of Curriculum Learning in Training Deep Networkscited by 286
- 2024Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoningcited by 279
- 2021Scaling Laws for Transfercited by 272
- 2024Iterative Reasoning Preference Optimizationcited by 246
- 2025Inference-Time Scaling for Generalist Reward Modelingcited by 239
- 2022Towards Understanding Chain-of-Thought Prompting: An Empirical Study of What Matterscited by 222
- 2023Beyond Chinchilla-Optimal: Accounting for Inference in Language Model Scaling Lawscited by 159
- 2024QuRating: Selecting High-Quality Data for Training Language Modelscited by 157
- 2022Coverage-centric Coreset Selection for High Pruning Ratescited by 117
- 2023CLadder: Assessing Causal Reasoning in Language Modelscited by 107
- 2024GLoRe: When, Where, and How to Improve LLM Reasoning via Global and Local Refinementscited by 103
- 2024Systematic Evaluation of LLM-as-a-Judge in LLM Alignment Tasks: Explainable Metrics and Diverse Prompt Templatescited by 72
- 2024Lean-STaR: Learning to Interleave Thinking and Provingcited by 68
- 2025Z1: Efficient Test-time Scaling with Codecited by 35
- 2020DEAL: Difficulty-aware Active Learning for Semantic Segmentationcited by 32
- 2024AutoPSV: Automated Process-Supervised Verifiercited by 30
- 2024Dataset Decomposition: Faster LLM Training with Variable Sequence Length Curriculumcited by 26
- 2025Lightweight Dataset Pruning without Full Training via Example Difficulty and Prediction Uncertaintycited by 22
- 2024Enhancing Code Generation Performance of Smaller Models by Distilling the Reasoning Ability of LLMscited by 17
- 2024Automatic Curriculum Expert Iteration for Reliable LLM Reasoningcited by 16
- 2024Toward Adaptive Reasoning in Large Language Models with Thought Rollbackcited by 11
- 2024Rationale-Aware Answer Verification by Pairwise Self-Evaluationcited by 8
Built on by
Citing papers per year: 2025: 3; 2026: 3.
6 papers.
- 2025A Survey of Self-Evolving Agents: What, When, How, and Where to Evolve on the Path to Artificial Super Intelligencecited by 170
- 2026FailForge: Distilling Procedural Competence from Persistent Failures into Code Agentscited by 2
- 2025Aligning Reasoning LLMs for Materials Discovery with Physics-aware Rejection Samplingcited by 1
- 2026AnE: Pushing the Reasoning Frontier of Multimodal LLMs via Anchor Evolutioncited by 1
- 2026MMVistaReason: Toward Open-Data and Post-Training Recipes for Multimodal Reasoningcited by 1
- 2025Learning to Reason in LLMs by Expectation Maximizationcited by 0
- 2026MMVistaReason: Toward Open-Data and Post-Training Recipes for Multimodal Reasoning1 Oct 2026
- 2026FailForge: Distilling Procedural Competence from Persistent Failures into Code Agents9 Aug 2026
- 2026AnE: Pushing the Reasoning Frontier of Multimodal LLMs via Anchor Evolution25 May 2026
- 2025Learning to Reason in LLMs by Expectation Maximization23 Dec 2025
- 2025Aligning Reasoning LLMs for Materials Discovery with Physics-aware Rejection Sampling31 Aug 2025
- 2025A Survey of Self-Evolving Agents: What, When, How, and Where to Evolve on the Path to Artificial Super Intelligence28 Jul 2025
Similar, not cited
Close in wording, no citation either way
- 2025HS-STaR: Hierarchical Sampling for Self-Taught Reasoners via Difficulty Estimation and Budget Reallocation
- 2024RL-STaR: Theoretical Analysis of Reinforcement Learning Frameworks for Self-Taught Reasoner
- 2025STaR: Towards Effective and Stable Table Reasoning via Slow-Thinking Large Language Models
- 2024Star-Agents: Automatic Data Optimization with LLM Agents for Instruction Tuning
- 2025Learning to Reason Across Parallel Samples for LLM Reasoning
- 2025SoTA with Less: MCTS-Guided Sample Selection for Data-Efficient Visual Reasoning Self-Improvement
- 2025Continuous Self-Improvement of Large Language Models by Test-time Training with Verifier-Driven Sample Selection
- 2026Ada-RS: Adaptive Rejection Sampling for Selective Thinking