Curriculum Reinforcement Learning from Easy to Hard Tasks Improves LLM Reasoning
Shubham Parashar, Shurui Gui, Xiner Li, Hongyi Ling, Sushil Vemuri, Blake Olson, Eric Li, Yu Zhang, James Caverlee, Dileep Kalathil, Shuiwang Ji
TL;DR
This work addresses the challenge of improving reasoning in language models via reinforcement learning post-training, where sparse rewards hinder learning on hard tasks. It proposes E2H Reasoner, a curriculum RL framework that decomposes tasks into easy-to-proceed levels (trivial/easy/medium) and uses probabilistic schedulers (cosine or Gaussian) to interpolate between distributions $d_1$ and $d_K$ across curriculum steps, thereby enhancing generalization. Grounded in Approximate Policy Iteration, it provides convergence guarantees and finite-sample complexity bounds, showing curriculum-based learning can require fewer total samples than direct learning. Empirically, E2H improves reasoning in small LLMs (1.5B–3B) across multiple domains, with strong gains on hard and out-of-distribution tasks, and a public implementation is available.
Abstract
We aim to improve the reasoning capabilities of language models via reinforcement learning (RL). Recent RL post-trained models like DeepSeek-R1 have demonstrated reasoning abilities on mathematical and coding tasks. However, prior studies suggest that using RL alone to improve reasoning on inherently difficult tasks is less effective. Here, we draw inspiration from curriculum learning and propose to schedule tasks from easy to hard (E2H), allowing LLMs to build reasoning skills gradually. Our method is termed E2H Reasoner. Empirically, we observe that, although easy tasks are important initially, fading them out through appropriate scheduling is essential in preventing overfitting. Theoretically, we establish convergence guarantees for E2H Reasoner within an approximate policy iteration framework. We derive finite-sample complexity bounds and show that when tasks are appropriately decomposed and conditioned, learning through curriculum stages requires fewer total samples than direct learning. Experiments across multiple domains show that E2H Reasoner significantly improves the reasoning ability of small LLMs (1.5B to 3B), which otherwise struggle when trained with vanilla RL alone, highlighting the effectiveness of our method. Our code can be found on https://github.com/divelab/E2H-Reasoning.
