arXiv:2502.01456cs.LGcs.AI2025-02被引 410

用隐式奖励让大模型在线优化推理过程,提升数学编程能力。

Process Reinforcement through Implicit Rewards

  • 通过策略回放和结果标签生成隐式过程奖励,免去人工标注。
  • 在多个推理基准上较SFT模型平均提升15.1%准确率。
  • 仅需10%数据量即可超越原模型,适合资源有限的强化学习场景。

密集的过程奖励在大型语言模型(LLM)的推理时扩展中比稀疏的结果级奖励更有效,尤其适用于需要复杂多步推理的任务。尽管密集奖励在强化学习中具有潜力,能缓解结果奖励带来的训练效率与信用分配问题,但其实际应用受限于在线训练过程奖励模型(PRM)的困难:高质量过程标签收集成本过高,易引发奖励劫持。为此,我们提出PRIME(Process Reinforcement through IMplicit rEwards),通过仅使用策略回放和结果标签生成隐式过程奖励,实现在线更新PRM。PRIME可与多种优势函数结合,无需专门的奖励模型训练阶段,显著降低开发开销。我们在多个竞争性数学与编程任务上验证了其有效性:从Qwen2.5-Math-7B-Base出发,PRIME在多个关键推理基准上相比SFT模型平均提升15.1%。值得注意的是,我们的模型Eurus-2-7B-PRIME仅用原模型10%的训练数据,就在七个推理基准上超越Qwen2.5-Math-7B-Instruct。

原文摘要 · Abstract (English)

Dense process rewards have proven a more effective alternative to the sparse outcome-level rewards in the inference-time scaling of large language models (LLMs), particularly in tasks requiring complex multi-step reasoning. While dense rewards also offer an appealing choice for the reinforcement learning (RL) of LLMs since their fine-grained rewards have the potential to address some inherent issues of outcome rewards, such as training efficiency and credit assignment, this potential remains largely unrealized. This can be primarily attributed to the challenges of training process reward models (PRMs) online, where collecting high-quality process labels is prohibitively expensive, making them particularly vulnerable to reward hacking. To address these challenges, we propose PRIME (Process Reinforcement through IMplicit rEwards), which enables online PRM updates using only policy rollouts and outcome labels through implict process rewards. PRIME combines well with various advantage functions and forgoes the dedicated reward model training phrase that existing approaches require, substantially reducing the development overhead. We demonstrate PRIME's effectiveness on competitional math and coding. Starting from Qwen2.5-Math-7B-Base, PRIME achieves a 15.1% average improvement across several key reasoning benchmarks over the SFT model. Notably, our resulting model, Eurus-2-7B-PRIME, surpasses Qwen2.5-Math-7B-Instruct on seven reasoning benchmarks with 10% of its training data.

强化学习大模型推理隐式奖励数学生成

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