arXiv:2507.05197cs.CLcs.LG2025-07NeurIPS被引 13

用策略判别器思想训练通用奖励模型,性能显著超越传统方法。

Pre-Trained Policy Discriminators are General Reward Models

  • 将奖励建模视为策略判别任务,通过对比不同策略生成奖励信号。
  • POLAR-7B在STEM和创意写作任务上准确率分别提升至81.0%和85.5%。
  • 适用于RLHF,对多个大模型有显著性能提升,适合追求强奖励模型的研究者。

我们提出一种新的奖励建模视角:将其视为策略判别任务,通过量化两个策略的差异生成奖励信号,引导训练策略向目标策略演进。基于此,我们提出可扩展的预训练方法POLAR,让奖励模型(RM)学会区分相同与不同的策略。与依赖绝对偏好传统方法不同,POLAR捕捉策略与任意目标策略间的相对差异,具备高可扩展性,适合建模通用排序关系。我们构建了参数量从1.8B到7B的系列RM。实验表明,POLAR显著优于非预训练方法,例如在STEM任务上,偏好准确率从54.8%提升至81.0%;在创意写作任务上从57.9%升至85.5%。在强化学习人类反馈(RFT)中,POLAR展现出强泛化能力,显著提升策略表现——使LLaMa3.1-8B平均得分从47.36%提高到56.33%,Qwen2.5-32B从64.49%升至70.47%(20个基准测试)。规模实验显示计算量与性能呈明显幂律关系,线性相关系数接近0.99。优异性能、强泛化性与良好可扩展性表明,POLAR是构建通用强奖励模型的有力方向。

原文摘要 · Abstract (English)

We offer a novel perspective on reward modeling by formulating it as a policy discriminator, which quantifies the difference between two policies to generate a reward signal, guiding the training policy towards a target policy with desired behaviors. Based on this conceptual insight, we propose a scalable pre-training method named Policy Discriminative Learning (POLAR), which trains a reward model (RM) to discern identical policies and discriminate different ones. Unlike traditional reward modeling methods relying on absolute preferences, POLAR captures the relative difference between one policy and an arbitrary target policy, which is a scalable, high-level optimization objective suitable for modeling generic ranking relationships. Leveraging the POLAR pre-training paradigm, we present a series of RMs with parameter scales from 1.8B to 7B. Empirical results show that POLAR substantially outperforms traditional non-pre-trained methods, significantly enhancing RM performance. For instance, POLAR-7B could improve preference accuracy from 54.8% to 81.0% on STEM tasks and from 57.9% to 85.5% on creative writing tasks compared to SOTA baselines. POLAR also shows robust generalization capabilities in RLHF using Reinforcement Fine-tuning (RFT), providing reliable reward signals and markedly enhancing policy performance--improving LLaMa3.1-8B from an average of 47.36% to 56.33% and Qwen2.5-32B from 64.49% to 70.47% on 20 benchmarks. Moreover, scaling experiments reveal a clear power-law relationship between computation and performance, supported by linear correlation coefficients approaching 0.99. The impressive performance, strong generalization, and scaling properties suggest that POLAR is a promising direction for developing general and strong reward models.

奖励建模策略判别RLHF大模型

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