Skip to main navigation Skip to search Skip to main content

Learning to Reason with Curriculum I: Provable Benefits of Autocurriculum

  • Nived Rajaraman
  • , Audrey Huang
  • , Miro Dudik
  • , Robert Schapire
  • , Dylan J. Foster
  • , Akshay Krishnamurthy

Research output: Contribution to journal › Conference article › peer-review

Abstract

Chain-of-thought reasoning, where language models expend additional computation by producing thinking tokens prior to final responses, has driven significant advances in model capabilities. However, training these reasoning models is extremely costly in terms of both data and compute, as it involves collecting long traces of reasoning behavior from humans or synthetic generators and further post-training the model via reinforcement learning. Are these costs fundamental, or can they be reduced through better algorithmic design? We show that autocurriculum—where the model uses its own performance to decide which problems to focus training on—provably improves upon standard training recipes for both supervised fine-tuning (SFT) and reinforcement learning (RL). For SFT, we show that autocurriculum requires exponentially fewer reasoning demonstrations than non-adaptive fine-tuning (Joshi et al., 2025), by focusing teacher supervision on prompts where the current model struggles. For RL fine-tuning, autocurriculum decouples the computational cost from the quality of the reference model, reducing the latter to a burn-in cost that is nearly independent of the target accuracy. These improvements arise purely from adaptive data selection, drawing on classical techniques from boosting (Freund and Schapire, 1997) and learning from counterexamples (Angluin, 1987), and requiring no assumption on the distribution or difficulty of prompts.

Original languageEnglish (US)
JournalProceedings of Machine Learning Research
Volume336
StatePublished - 2026
Event39th Annual Conference on Learning Theory, COLT 2026 - San Diego, United States
Duration: Jun 29 2026 → Jul 3 2026

All Science Journal Classification (ASJC) codes

  • Software
  • Control and Systems Engineering
  • Statistics and Probability
  • Artificial Intelligence

Keywords

  • Autocurriculum
  • Language model reasoning
  • Reinforcement Learning

Fingerprint

Dive into the research topics of 'Learning to Reason with Curriculum I: Provable Benefits of Autocurriculum'. Together they form a unique fingerprint.

Cite this