用最小值分配奖励,解决大模型推理中的奖励欺骗问题。
Stop Summation: Min-Form Credit Assignment Is All Process Reward Model Needs for Reasoning
- 采用最小值形式的奖励分配机制,抑制奖励欺诈
- 仅需30%步骤即可达到与验证奖励相当的推理性能
- 适合需要高效可靠训练的大模型推理微调场景
过程奖励模型(PRMs)在提升大语言模型(LLMs)复杂推理任务表现方面表现出色。然而,其在强化学习微调中常因奖励欺骗问题而受限。本文识别出根本原因:强化学习中经典的求和形式奖励分配(累积折扣未来奖励)易诱导大模型操纵高奖励步骤。为此,提出PURE(Process sUpervised Reinforcement lEarning):核心创新是使用未来奖励的最小值作为价值函数,显著缓解奖励欺骗,合理分配优势。在3个基础模型上的实验表明,采用最小值分配的PRM方法仅需30%步骤即可达到验证奖励方法的推理性能;而传统求和形式在训练初期即导致崩溃。此外,若额外引入10%可验证奖励,进一步缓解奖励欺骗,在Qwen2.5-Math-7B上实现AMC23 82.5%准确率及5个基准测试平均53.3%准确率。本文还总结了观测到的奖励欺骗案例并分析训练崩溃原因。代码与模型权重已开源。
原文摘要 · Abstract (English)
Process reward models (PRMs) have proven effective for test-time scaling of Large Language Models (LLMs) on challenging reasoning tasks. However, reward hacking issues with PRMs limit their successful application in reinforcement fine-tuning. In this paper, we identify the main cause of PRM-induced reward hacking: the canonical summation-form credit assignment in reinforcement learning (RL), which defines the value as cumulative gamma-decayed future rewards, easily induces LLMs to hack steps with high rewards. To address this, we propose PURE: Process sUpervised Reinforcement lEarning. The key innovation of PURE is a min-form credit assignment that formulates the value function as the minimum of future rewards. This method significantly alleviates reward hacking by limiting the value function range and distributing advantages more reasonably. Through extensive experiments on 3 base models, we show that PRM-based approaches enabling min-form credit assignment achieve comparable reasoning performance to verifiable reward-based methods within only 30% steps. In contrast, the canonical sum-form credit assignment collapses training even at the beginning! Additionally, when we supplement PRM-based fine-tuning with just 10% verifiable rewards, we further alleviate reward hacking and produce the best fine-tuned model based on Qwen2.5-Math-7B in our experiments, achieving 82.5% accuracy on AMC23 and 53.3% average accuracy across 5 benchmarks. Moreover, we summarize the observed reward hacking cases and analyze the causes of training collapse. We release our code and model weights at https://github.com/CJReinforce/PURE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。