探索性强化学习:用于大型语言模型训练过程中的中期优化
ExpRL: Exploratory RL for LLM Mid-Training
摘要
稀疏奖励强化学习已成为提升大型语言模型推理能力的重要工具,但其效果的好坏在很大程度上取决于基础模型中所包含的“覆盖性”信息。实际上,通常会在训练过程中通过针对特定推理任务的数据来引导模型学习一些基本的技能,比如分解、验证或自我修正等。虽然这种策略很有效,但它需要人工指定模型应该学习的内容;而且目前还不清楚这种基本技能是否足以应对那些需要将这些技能整合到更综合解决方案中的复杂问题。我们提出了一种更为自动化的方法:利用大量人类编写的问答数据来进行基于强化学习的训练。与将参考解决方案作为模仿目标不同,我们的方法将参考解决方案作为奖励的基准:这些参考解决方案对策略来说是不可见的,它们仅用于构建针对特定问题的评估标准,以评判模型在训练过程中的表现。策略会根据原始问题提示来生成推理样本,而大型语言模型则会将生成的推理样本与参考解决方案进行比较,并据此给予相应的奖励。这样,ExpRL就能够强化那些具有实用价值的、能够带来积极效果的推理行为,而这些行为往往难以被稀疏的最终奖励所体现出来。在复杂的数学推理任务中,ExpRL所提供的强化效果优于其他方法,比如SFT、稀疏奖励的GRPO以及自我蒸馏方法。此外,更多的跨领域实验表明,ExpRL不仅可以应用于单纯的数学问题,还可以扩展到其他领域。
English 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 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: 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 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.