arXiv:2410.08146cs.LGcs.CL2024-10ICLR被引 271

用进度奖励提升大模型推理,让每一步都更智能。

Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoning

  • 用进度衡量每步改进,而非仅看最终结果
  • 测试时搜索准确率提升超8%,计算效率高1.5-5倍
  • 适合研究大模型推理与强化学习的学者

提升大语言模型推理能力的一种有前景方法是使用过程奖励模型(PRMs),在多步推理过程中每一步提供反馈,相比仅在最终步骤反馈的成果奖励模型(ORMs)能更好分配奖励。然而,密集的逐步人工标注难以扩展,而基于自动标注数据训练的PRMs此前成效有限。本文的核心洞见是:有效的过程奖励应衡量‘进度’——即采取某一步后对未来正确回答概率的影响变化,对应强化学习中的步骤级优势。关键在于,该进度应在与基础策略不同的证明者策略下评估。我们理论上刻画了优质证明者的特性,结果表明从这类证明者优化过程奖励可增强测试时搜索与在线强化学习中的探索能力。事实上,弱证明者策略也能显著提升强基础策略,这在实验中得到验证。我们训练出过程优势验证器(PAVs)来预测此类进度,并证明:相较于ORMs,基于PAVs的测试时搜索准确率提升超过8%,计算效率提高1.5至5倍;在线强化学习中,使用PAVs提供的密集奖励,实现样本效率提升5-6倍、准确率提升超6%。

原文摘要 · Abstract (English)

A promising approach for improving reasoning in large language models is to use process reward models (PRMs). PRMs provide feedback at each step of a multi-step reasoning trace, potentially improving credit assignment over outcome reward models (ORMs) that only provide feedback at the final step. However, collecting dense, per-step human labels is not scalable, and training PRMs from automatically-labeled data has thus far led to limited gains. To improve a base policy by running search against a PRM or using it as dense rewards for reinforcement learning (RL), we ask: "How should we design process rewards?". Our key insight is that, to be effective, the process reward for a step should measure progress: a change in the likelihood of producing a correct response in the future, before and after taking the step, corresponding to the notion of step-level advantages in RL. Crucially, this progress should be measured under a prover policy distinct from the base policy. We theoretically characterize the set of good provers and our results show that optimizing process rewards from such provers improves exploration during test-time search and online RL. In fact, our characterization shows that weak prover policies can substantially improve a stronger base policy, which we also observe empirically. We validate our claims by training process advantage verifiers (PAVs) to predict progress under such provers, and show that compared to ORMs, test-time search against PAVs is $>8\%$ more accurate, and $1.5-5\times$ more compute-efficient. Online RL with dense rewards from PAVs enables one of the first results with $5-6\times$ gain in sample efficiency, and $>6\%$ gain in accuracy, over ORMs.

大模型推理强化学习奖励设计

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