arXiv:2502.06781cs.CLcs.LG2025-02被引 51

用结果奖励训练数学推理模型,7B模型达到32B水平。

Exploring the Limit of Outcome Reward for Learning Mathematical Reasoning

  • 基于结果反馈设计新强化学习框架,仅需对错二元奖励
  • 7B模型在MATH-500上达94.0 pass@1,媲美32B模型
  • 适合想用小模型实现强推理能力的研究者

数学推理能力是通用智能的关键。尽管闭源模型如OpenAI o系列取得显著进展,但技术细节未公开,普遍认为使用了强化学习(RL)和长链思维。本文提出新框架OREAL,探索仅依赖二元结果奖励的强化学习在数学推理任务中的性能极限。理论上证明:在最佳-多选(BoN)采样中对正向轨迹进行行为克隆,足以学习到带KL正则化的最优策略;同时需重设负样本奖励以保证梯度一致性。为缓解稀疏奖励问题,尤其长链思维中部分正确带来的挑战,引入词粒度奖励模型,用于选择推理路径中的关键词。首次实现7B模型通过强化学习在MATH-500上达到94.0 pass@1,与32B模型相当;OREAL-32B更以95.0 pass@1超越此前蒸馏训练的32B模型。研究还揭示初始模型与训练题目的重要性。代码、模型与数据将开源。

原文摘要 · Abstract (English)

Reasoning abilities, especially those for solving complex math problems, are crucial components of general intelligence. Recent advances by proprietary companies, such as o-series models of OpenAI, have made remarkable progress on reasoning tasks. However, the complete technical details remain unrevealed, and the techniques that are believed certainly to be adopted are only reinforcement learning (RL) and the long chain of thoughts. This paper proposes a new RL framework, termed OREAL, to pursue the performance limit that can be achieved through \textbf{O}utcome \textbf{RE}w\textbf{A}rd-based reinforcement \textbf{L}earning for mathematical reasoning tasks, where only binary outcome rewards are easily accessible. We theoretically prove that behavior cloning on positive trajectories from best-of-N (BoN) sampling is sufficient to learn the KL-regularized optimal policy in binary feedback environments. This formulation further implies that the rewards of negative samples should be reshaped to ensure the gradient consistency between positive and negative samples. To alleviate the long-existing difficulties brought by sparse rewards in RL, which are even exacerbated by the partial correctness of the long chain of thought for reasoning tasks, we further apply a token-level reward model to sample important tokens in reasoning trajectories for learning. With OREAL, for the first time, a 7B model can obtain 94.0 pass@1 accuracy on MATH-500 through RL, being on par with 32B models. OREAL-32B also surpasses previous 32B models trained by distillation with 95.0 pass@1 accuracy on MATH-500. Our investigation also indicates the importance of initial policy models and training queries for RL. Code, models, and data will be released to benefit future research\footnote{https://github.com/InternLM/OREAL}.

强化学习数学推理小模型二元奖励

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。