ExpRL: Exploratory RL for LLM Mid-Training
2026-06-15 • Machine Learning
Machine Learning
AI summaryⓘ
The authors explore a new way to help large language models improve their reasoning by using reinforcement learning (RL) during training. Instead of simply copying example solutions, their method, ExpRL, grades the model's thought process against human answers and rewards partial steps that show progress. This approach helps the model learn more effectively on tough math problems and works better than traditional training methods. They also find that ExpRL may be useful beyond math tasks.
Reinforcement LearningLarge Language ModelsSparse RewardsMid-TrainingReasoning TracesSelf-CorrectionMath Reasoning TasksPolicy OptimizationDense RewardsSelf-Distillation
Authors
Violet Xiang, Amrith Setlur, Chase Blagden, Nick Haber, Aviral Kumar
Abstract
Sparse reward reinforcement learning (RL) has become a standard tool for improving LLM reasoning, but its success depends critically on the coverage present in the base model. In practice, models are often primed for RL through \emph{mid-training} on curated reasoning traces that teach useful primitive skills such as decomposition, verification, or self-correction. Although effective, this strategy requires manually specifying what the model should learn, and it remains unclear whether such primitive coverage is enough for much harder problems, which require combining these skills into broader solution strategies. We study a more automated approach: \emph{RL-based mid-training} using large corpora of human-written question-answer data. Rather than treating reference solutions as targets to imitate, our method, ExpRL, uses them as \emph{reward scaffolds}: references are hidden from the policy and used only to construct problem-specific grading rubrics for judging on-policy reasoning traces. The policy samples from the original problem prompt, while an LLM judge compares the sampled reasoning trace against the reference solution and assigns outcome-level or process-level dense rewards. This lets ExpRL reinforce partial progress, useful intermediate reductions, and productive reasoning behaviors that sparse final-answer rewards often fail to upweight. On challenging math reasoning tasks, ExpRL yields stronger RL priming than SFT, sparse-reward GRPO, and self-distillation, and provides a better initialization for subsequent sparse-reward RL. Additional mixed-domain experiments further suggest that ExpRL can extend beyond the original math-only setting.