「知其然,亦知其所以然。」
这句儒家命题强调,真正的理解不仅在于结果,更在于推理过程。如今,在大型语言模型的后训练阶段,奖励模型承担着桥接模型行为与人类价值的重要职责;但现有模型往往只给出一个分数,却难以解释其依据。缺乏推理的奖励,就如「知其然而不知其所以然」,既难以建立信任,也难以指导更优的学习。
伊利诺伊大学香槟分校的研究团队提出了 RM-R1 框架,将奖励建模重新定义为推理任务,提出了推理奖励模型(Reasoning Reward Models, ReasRMs)。RM-R1 关注于如何通过整合推理能力来增强奖励模型,使其能够更准确地对模型输出进行评估和打分,从而更好地与人类偏好对齐。RM-R1 通过生成结构化的评估标准和推理过程,提升了奖励模型的可解释性和性能。
论文标题:RM-R1: Reward Modeling as Reasoning
论文链接:pdf/2505.02387
代码仓库:RM-R1-UIUC/RM-R1
开源模型:collections/gaotang/rm-r1-681128cdab932701cad844c8
文章验证了三个核心发现:
规模带来增益:随着模型变大、计算力增强,RM-R1 的推理链训练方法效果越好,性能几乎线性提升;
简单套用旧 RL 策略行不通:想让模型「会推理」,得精准划分问题类型、并对推理过程进行定向蒸馏训练,才能带来真正泛化的提升;
推理比直接输出答案更通用:相比传统的直接监督,RM-R1 的推理能力更稳健、更易扩展到新任务,即使训练数据少也有优势。
推理驱动的奖励建模:从评分到解释
RM-R1 引入了「推理奖励模型」(ReasRMs)的概念,将奖励建模视为一个推理过程。模型在评估候选响应之前,先生成结构化的评估标准或推理过程,然后根据这些标准进行评估。这一方法提升了模型的可解释性,使其能够更好地处理复杂的推理任务。
图 1: 直接用现有的 Instruction-tuned model 过拟合于 SFT 数据中的表层模式,无法识别被拒绝回答中的情感伤害和细节缺失。而右下角的推理奖励模型则能跳出表面特征,从回应所造成的深层影响出发进行评估。
核心机制:链式评估准则 (CoR) 如何引导模型「思考」
RM-R1 的核心创新之一在于其引入的链式评估准则(Chain-of-Rubrics, CoR)机制。该机制将奖励建模任务分解为一系列结构化的推理步骤,使得模型在给出最终偏好判断之前,能够像人类专家一样进行深层次的「思考」和评估。CoR 机制的精髓在于其「自适应性」和「内部专家」角色,这不仅提升了评估的准确性,更揭示了模型在复杂任务中进行「元推理」(meta-reasoning)的潜力。
CoR 机制能够根据任务类型动态调整其评估策略。系统提示会指导奖励模型(rθ)首先将每个偏好数据样本分类为「推理型」(Reasoning)或「对话型」(Chat)任务。
对于推理任务(如数学、编程、逻辑推理、多步推理等),模型被指示首先自行解决用户的问题并生成解决方案,并将其封装在 solution ... 标签内。这个内部生成的解决方案充当了「标准答案」 或「内部专家」的参考。模型随后会以此为基准,评估两个候选响应的正确性、完整性和推理质量。这种机制使得模型能够进行内部自洽性检查和「自我纠错」,因为它在判断外部答案之前,先独立地得出了自己的「真理」。这种「内部专家」机制使得模型能够进行更深层次的正确性验证,而不仅仅是表层模式匹配,从而显著提高了在复杂推理任务上的判断准确性。
对于对话任务(如开放式对话、安全问题、风格改写或一般性帮助请求),模型则被指示生成定制化的评估准则(rubric),并为每个准则分配权重,同时提供理由(justify),解释选择这些准则及其权重的原因。这些准则和理由被封装在 rubric ... 和 justify ... 标签内。随后,模型会根据这些明确定义的准则对两个候选响应进行详细比较和评估,并将评估结果放在 eval ... 标签中,最后给出最终判断,如 answer [[A]] 或 answer [[B]]。这种方法反映了人类偏好判断的复杂性和多维度性,模型不再是简单地进行二元判断,而是能够像人类一样权衡多个因素。
CoR 机制通过强制模型遵循预定义的逻辑序列、进行任务特定评估、提供透明的推理痕迹以及将评估基于实际内容而非表面特征,从而实现了深度推理并增强了可解释性。这种「自适应性」表明 RM-R1 不仅仅是学会了「如何推理」,更学会了「何时以及如何应用不同的推理策略」,这是一种更高层次的认知能力,即「元推理」能力。这种能力是其在多样化基准上取得最先进表现的关键,因为它能够为不同类型的任务提供最恰当、最细致的评估。
图 2: CoR 机制将输入样本划分为两类之一:对话类(chat)或推理类(reasoning)。这种任务感知机制使模型能够根据任务类型灵活调整推理策略,从而生成更契合、更有效的奖励信号。
训练流程:从推理蒸馏到强化学习
RM-R1 的训练包括两个关键阶段:
推理蒸馏:从高质量的推理链中提取知识,训练模型生成结构化的评估标准。
强化学习:使用可验证的奖励信号,进一步优化模型的推理能力。
这一训练流程使得 RM-R1 在 RewardBench, RM-Bench 和 RMB 等多个奖励模型基准测试中表现出色,超过了更大规模的开源模型(如 INF-ORM-Llama3.1-70B)和闭源模型(如 GPT-4o) 。
图 3: RM-R1 的训练流程分两步走。第一步:推理链的蒸馏(Distillation of Reasoning Chains),使用高质量的推理链数据对模型进行蒸馏,使其初步具备推理能力。第二步:强化学习(Reinforcement Learning, RL):在蒸馏的基础上,使用可验证的奖励进行强化学习,进一步提升模型的推理能力。
实验验证:RM-R1 如何刷新奖励模型性能
为了全面验证 RM-R1 在奖励建模任务中的有效性,研究团队在 RewardBench, RM-Bench 和 RMB 等多个权威基准上进行了系统性实验。实验设置使用了 7B/14B/32B 的 Qwen-2.5-Instruct 以及 DeepSeek-Distilled-Qwen 作为基础模型。对比了多种训练策略,包括原始模型、监督微调(SFT)以及 RM-R1 提出的推理蒸馏与强化学习结合的策略。主要评估维度是模型的准确率,以及在不同任务类别和难度下的泛化能力。
核心结果:
显著性能提升: RM-R1 模型在所有评估基准上均实现了最先进(SOTA)或接近 SOTA 的性能。例如,RM-R1-DeepSeek-Distilled-Qwen-14B 模型在平均表现上超越了 INF-ORM-Llama3.1-70B、Nemotron-4-340B-Reward 和 GPT-4o,尽管其模型规模远小于这些基线模型。RM-R1-Qwen-Instruct-32B 和 RM-R1-DeepSeek-Distilled-Qwen-32B 进一步扩大了领先优势,在某些基准上,性能提升高达 8.7%。这表明,在奖励建模任务中,模型的「推理能力」比单纯的「模型规模」或「参数数量」更为关键。传统奖励模型可能依赖于从海量数据中学习到的表层模式或统计关联,而 RM-R1 通过 CoR 机制,强制模型进行深层逻辑推理,从而能够更好地理解和判断复杂场景。
卓越的推理能力: 特别是在 RM-Bench(推理密集型基准)上,RM-R1-DeepSeek-Distilled-Qwen-32B 在数学和代码任务中取得了突破性进展,准确率分别达到 91.8% 和 74.1%,远超此前的最佳表现(数学 73%,代码 63%)。这直接证明了 RM-R1 通过其独特的推理训练范式,显著增强了模型的深层推理能力。
模型规模与推理计算的积极影响: 实验还发现,更大的 RM-R1 模型(7B、14B、32B)不仅最终性能更好,而且从推理训练中获得的性能增益也更大,呈现出近乎线性的趋势。同时,增加推理时的计算预算(允许更长的推理链)也能显著提升性能,这进一步验证了深度推理对奖励模型性能的重要性。
鲁棒性与泛化能力(消融研究):消融研究深入剖析了 RM-R1 成功的关键因素。结果表明,单独的强化学习训练不足以弥补与完整 RM-R1 模型之间的性能差距。CoR 提示工程,包括评估准则(Rubrics)和查询分类(Query Categorization),显著优化了奖励模型的推理过程,尤其是在对话、安全和推理任务上。推理蒸馏被证实是性能提升的关键因素,它为模型提供了强大的基础推理能力。
数据效率:基于 Instruct 的模型展现出惊人的数据效率。仅使用 8.7K 蒸馏示例,RM-R1 就达到了与使用 800K 示例训练的 DeepSeek-Distilled 模型相当的竞争性性能。这意味着,通过创新的架构和训练方法(如推理蒸馏和 CoR),可以在相对较小的模型上实现卓越的性能,这对于计算资源有限的团队和实际部署具有巨大的经济和效率优势。
这些实验结果共同表明,RM-R1 的成功暗示了未来奖励模型研究的一个重要方向:不仅仅是扩大模型规模,更要注重提升模型的内在认知和推理能力,使其能够像人类一样「理解」和「思考」偏好,而非仅仅「识别」偏好。
结语:RM-R1 开启奖励建模的新篇章
RM-R1 的提出,不仅提供了一种新的奖励模型训练方法,更开创了一种基于推理的可解释奖励范式,为大型语言模型与人类偏好对齐领域带来了深远的影响。通过大规模实验与深入对比分析,该研究验证了几个核心发现。首先,推理能力是奖励模型的关键要素。RM-R1 将奖励建模重构为一项推理任务,显著提升了模型的准确性,并提供了前所未有的可解释性。其次,两阶段训练范式展现出卓越的高效性。推理蒸馏与强化学习的协同作用,特别是高质量推理链的生成,被证实是其成功的基石。最后,链式评估准则(CoR)机制带来了深度可解释性与卓越性能。通过结构化推理和任务定制化评估,RM-R1 在性能上超越了现有最先进的模型,同时提供了透明的判断依据。通过大规模实验与深入对比分析,文章验证了三个核心发现:
1. 规模带来增益:随着模型变大、计算力增强,RM-R1 的推理链训练方法效果越好,性能几乎线性提升;
2. 简单套用旧 RL 策略行不通:想让模型「会推理」,得精准划分问题类型、并对推理过程进行定向蒸馏训练,才能带来真正泛化的提升;
3. 推理比直接输出答案更通用:相比传统的直接监督,RM-R1 的推理能力更稳健、更易扩展到新任务,即使训练数据少也有优势。
0 条