新方法让大模型奖励模型同时具备相对排序和绝对质量评估能力。
Learning Ordinal Probabilistic Reward from Preferences
- 将奖励建模为概率分布,用有序评分捕捉响应质量
- 在多个基准上提升准确率2.9%~7.4%,且数据效率高
- 适合需要精准质量判断的对齐任务,如生成内容评估
奖励模型对对齐大语言模型与人类价值观至关重要。现有方法分为生成式(GRMs)和判别式(DRMs),但各有局限:GRMs需昂贵的逐点标注,而DRMs生成的相对得分未经校准且无概率解释。为此,我们提出一种新范式——概率奖励模型(PRM),将奖励视为随机变量,学习每个响应质量的完整概率分布。为实用化,我们给出其闭式离散实现:有序概率奖励模型(OPRM),将质量评分离散化为有限有序等级。基于OPRM,我们提出数据高效的训练策略区域泛洪调优(RgFT),通过质量等级标注引导模型将概率质量集中在对应评分区间。在多个奖励模型基准上的实验表明,该方法相比先前模型提升准确率2.9%~7.4%,表现出更强性能与数据效率。对得分分布的分析证实,该方法不仅能捕捉相对排序,还能反映绝对质量。
原文摘要 · Abstract (English)
Reward models are crucial for aligning large language models (LLMs) with human values and intentions. Existing approaches follow either Generative (GRMs) or Discriminative (DRMs) paradigms, yet both suffer from limitations: GRMs typically demand costly point-wise supervision, while DRMs produce uncalibrated relative scores that lack probabilistic interpretation. To address these challenges, we introduce a novel reward modeling paradigm: Probabilistic Reward Model (PRM). Instead of modeling reward as a deterministic scalar, our approach treats it as a random variable, learning a full probability distribution for the quality of each response. To make this paradigm practical, we present its closed-form, discrete realization: the Ordinal Probabilistic Reward Model (OPRM), which discretizes the quality score into a finite set of ordinal ratings. Building on OPRM, we propose a data-efficient training strategy called Region Flooding Tuning (RgFT). It enables rewards to better reflect absolute text quality by incorporating quality-level annotations, which guide the model to concentrate the probability mass within corresponding rating sub-regions. Experiments on various reward model benchmarks show that our method improves accuracy by $\textbf{2.9%}\sim\textbf{7.4%}$ compared to prior reward models, demonstrating strong performance and data efficiency. Analysis of the score distribution provides evidence that our method captures not only relative rankings but also absolute quality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。