arXiv:2412.01981cs.LGcs.CL2024-12被引 166

无需中间步骤标注,用结果标签就能训练出高效的过程奖励模型。

Free Process Rewards without Process Labels

  • 用响应级标签训练输出奖励模型,隐式获得过程奖励能力。
  • 仅需1/38的训练数据就超越强基线,单响应训练也有效。
  • 适合资源有限但想提升推理模型的开发者使用。

与评估完整输出的结果奖励模型(ORM)不同,过程奖励模型(PRM)逐步评分推理轨迹,提供更密集、精细的奖励。然而,训练PRM需要在每个中间步骤标注标签,人工和自动收集都面临巨大挑战。本文提出,只需在廉价的响应级标签上训练一个ORM,即可无额外成本获得隐式PRM。理论与实证均表明,只要将结果奖励参数化为策略模型与参考模型的对数似然比,无论损失函数如何选择,该方法均可优化。实验中,我们以多种目标实例化隐式PRM,在MATH数据集上验证其性能:优于基于MCTS的Math-Shepherd基线,且训练数据不足其1/38;通过多数投票可进一步提升性能。研究还发现,扩大指令和响应规模有助于性能提升,后者增益更大。特别地,当使用交叉熵损失时,该隐式PRM具有极高的数据效率,即使每条指令仅有一个响应,仍能持续改进生成模型。此外,指令应与下游任务相关,响应多样性不带来增益。令人惊讶的是,额外使用Math-Shepherd的步骤标签并未提升仅基于结果数据训练的隐式PRM性能。本工作希望推动对PRM训练方式的重新思考,促进其更广泛的应用。

原文摘要 · Abstract (English)

Different from its counterpart outcome reward models (ORMs), which evaluate the entire responses, a process reward model (PRM) scores a reasoning trajectory step by step, providing denser and more fine grained rewards. However, training a PRM requires labels annotated at every intermediate step, presenting significant challenges for both manual and automatic data collection. This paper aims to address this challenge. Both theoretically and empirically, we show that an \textit{implicit PRM} can be obtained at no additional cost, by simply training an ORM on the cheaper response-level labels. The only assumption is to parameterize the outcome reward as the log-likelihood ratios of the policy and reference models, which can be optimized regardless of the specific choice of loss objectives. In experiments, we instantiate our implicit PRMs with various objectives and evaluate their performance on MATH. We show that our implicit PRM outperforms a strong MCTS-based baseline \textit{á la} Math-Shepherd using less than $1/38$ of the training data. Its performance can be further improved with majority voting. We further find that scaling up instructions and responses benefits our implicit PRM, and the latter brings a larger gain. Particularly, we find that our implicit PRM, when instantiated with the cross-entropy (CE) loss, is more data-efficient and can keep improving generation models even when trained with only one response per instruction, the setup that suffers from extreme data scarcity and imbalance. Further, instructions should be relevant to downstream tasks while the diversity of responses does not bring gains. Surprisingly, training on extra Math-Shepherd step labels brings no further improvements to our implicit PRM trained on only outcome data. We hope that our work will encourage a rethinking of PRM training approaches and contribute to making training PRMs more accessible.

奖励模型推理生成数据效率强化学习

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