Hexo

凡事预则立,不预则废


  • Home

  • Tags

  • Archives

  • Navigation

  • Search

NLP——LLM对齐微调-PaTaRM

注:本文包含 AI 辅助创作

  • 参考链接:
    • 原始论文:PaTaRM: Bridging Pairwise and Pointwise Signals via Preference-Aware Task-Adaptive Reward Modeling, BUPT & Meituan, 20251028
    • GitHub:github.com/JaneEyre0530/PaTaRM

Paper Summary

  • 整体说明:
    • 本文提出的 PaTaRM 是一个在 RLHF 中桥接 Pairwise 和 Pointwise 生成式奖励模型的统一框架
      • 通过将 偏好感知奖励机制(PAR) 与 动态 rubric 适应 相结合,PaTaRM 能够实现高效且可解释的 Pointwise 奖励建模,而无需显式的 Pointwise 标签
    • 在 RewardBench 和 RMBench 上的大量实验表明本文的方法有显著提升
    • PaTaRM 还提升了下游 RLHF 性能
    • 思考:本文的核心贡献是将 二元奖励标签数据 训练了一个 Pointwise 的生成式奖励模型
  • 背景 & 问题提出:
    • 奖励模型(RMs)是从 RLHF 的核心,它提供了关键监督信号,使 LLMs 与人类偏好对齐
    • 虽然生成式奖励模型(Generative Reward Models,GRMs,也简称 GenRMs)比传统的标量奖励模型提供了更强的可解释性,但当前的训练范式仍然有限
  • Pairwise 方法的问题:
    • 依赖于二元的“好与坏(good-versus-bad)”标签,这会导致与 Point-wise 推理不匹配,并且在 RLHF 中有效应用需要复杂的配对策略
    • Point-wise 方法需要 Rubric-driven criteria 的更精细的绝对标注,这导致适应性差和标注成本高
  • 在本工作中,论文提出了 偏好感知的任务自适应奖励模型(Preference-Aware Task-Adaptive Reward Model, PaTaRM) ,这是一个统一的框架,集成了偏好感知奖励(Preference-Aware Reward,PAR) 机制和动态评价 Rubric 适应
    • PaTaRM 利用来自 Pairwise 数据的相对偏好(relative preference)信息来构建鲁棒的 Point-wise 训练信号 ,从而无需显式的 Point-wise 标签
  • Simultaneously,它采用了一个任务自适应的评价 Rubric 系统,灵活地为全局任务一致性和 Instance-specific 细粒度推理生成评估标准(criteria)
    • 这种设计能够为 RLHF 实现高效、可泛化和可解释的奖励建模
  • 大量实验表明,PaTaRM 在 Qwen3-8B 和 Qwen3-14B 模型上,在 RewardBench 和 RMBench 上的平均相对改进达到 4.7%
  • Furthermore,PaTaRM 提升了下游 RLHF 性能,在 IFEval 和 InFoBench 基准上的平均改进为 13.6%,证实了其有效性和鲁棒性
  • 论文的代码可在 github.com/JaneEyre0530/PaTaRM 获取
  • 本文阅读的一些说明:
    • 附录 B.3 中的图 10 是论文 PaTaRM 方法的 Prompt Template
      • 其中针对 Pointwise(五角星 ⭐️)和 Pairwise(奖杯 🏆)在 Prompt+Response 部分和输出部分不太一致(图 10 中已经标注清楚)
    • PaTaRM 方法的 使用方法(Inference 过程)可以参考图 2 的上半部分,核心是同时给 PaTaRM Pointwise 的 Response 和 Primary Rubrics,然后要求 PaTaRM 动态生成一些附加的 Rubrics,然后以 Primary Rubrics 为主来对 Response 进行打分
    • Primary Rubric 是提前固定好的(人工手写的),不同任务的 Primary Rubrics 不同,不同任务(如 Code 和 IF 等)的示例详情见附录 B.2
    • Primary Rubric 数已提前准备好的,所以也称为 Provided Rubrics
    • PaTaRM 训练时使用的是 Pairwise 数据(这样才能计算损失),评估其他回复的质量时输入单个 Response,输出 Pointwise 评估结果(GenRM),方便使用
    • 论文其实给出了 Pairwise 和 Pointwise 的结论:
      • 单从评估效果看, Pairwise 的方法比 Pointwise 的方法效果会更好,但是 Pairwise 的方法在训练 RM 时较难使用,所以论文的 PaTaRM 方法本身是 Pointwise 的(只是训练时使用了 Pairwise 的信号)
    • PaTaRM 的主要创新是 PAR 机制 :
      • 跟 BT 方法一样,PaTaRM 训练时使用 Pairwise 的信号 ,最终实现了可以对 Pointwise 的 Response 打分
      • 与 BT 方法不同,PaTaRM 是 GenRM , BT RM 是标量奖励模型 (输出是一个分数)
    • 问题:附录 E 中的 Case Study 过程(表 8 和表 9)没有看到 Primary Rubrics 以及其他完整 Prompt,应该是省略了没有写出来?实际上应该是跟 图 10 对齐的才对
    • 问题:论文没有明确 Pairwise 方法具体训练方式(损失函数是 0 or 1 吗?)

Introduction and Discussion

  • RMs 是从 RLHF 的基础,作为关键的监督信号,引导 LLMs 朝向与人类对齐的行为
  • 主流方法将标量奖励模型训练为判别式分类器(discriminative classifiers),通常通过 Bradley-Terry 模型为候选回答分配数值分数 (2024; 2024; 2024; Bradley and Terry, 1952)
  • 尽管标量奖励模型在基本偏好对齐方面有效,但它们存在显著局限性:
    • 它们未能充分利用LLMs的生成和推理能力 (2025),常常捕捉表面相关性而非真实的人类偏好 (2025)
    • Moreover,它们容易过拟合并对分布偏移敏感 (2025)
  • 为了解决这些局限性,生成式奖励模型(Generative Reward Models, GRMs or GenRMs)已成为一种有前景的替代方案,提供对模型输出更结构化、更可解释的评估 (2025; 2025)
  • 当前的 GRM 训练范式大致可分为两种主要类型
    • 第一种是 Pairwise 生成式奖励模型(pairwise GRM) ,它通过在训练期间利用比较数据来优化 Pairwise 偏好目标
      • 尽管这种方法在捕捉相对偏好方面有效,但它存在两个基本局限性:
        • (1) 它无法执行单实例评估任务,因为其推理机制本质上需要比较输入,这为需要绝对质量评估的实际应用造成了关键差距
        • (2) Pairwise 范式通过要求从比较奖励转换为绝对奖励,破坏了 RLHF 流程,同时引入了近似误差,与直接的 Point-wise 方法相比,增加了训练的不稳定性 (2025)
          • 理解:因为 RLHF 流程需要的是具体的分数,即 Pointwise 的打分,所以叫做转换为绝对奖励
          • 问题:如何理解这里的近似误差?论文的训练明明展示了结果是 Pairwise 的方法更好啊
    • 第二种是 Point-wise 生成式奖励模型(point-wise GRM) ,它在评估和训练阶段都面临关键限制
      • 在评估方面, Point-wise GRM 通常依赖静态的 Rubric (Static Rubrics),这些标准是预定义的通用规则(General rules) (2024) 或由 GPT4o 等 LLM 外部生成的标准(Generated criteria from LLMs) (2025; 2025)
      • Generated criteria 缺乏对任务特定细微差异的适应性,而后者则产生高昂的计算成本,并可能传播偏见(Bias Propagation)
        • 问题:这里的传播偏见应该是指动态标准不固定,不同的 样本遇到不同的指标,打分本身可能就会不太可比?那其实固定不同的 Prompt 只能用相同的指标就好了吧?
      • 在训练方面, Point-wise 方法依赖于为每个 Rubric 提供的显式标注数据,并且涉及不稳定的训练,导致高标注成本和对噪声的敏感性增加
  • 如图 1 所示,这些局限性凸显了 GRM 设计中的一个核心挑战:能否在不依赖显式 Point-wise 标签的情况下有效训练 Point-wise GRM,同时为多样化任务支持灵活且自适应的 Rubric ?
  • 为了应对这些挑战,论文引入了一个统一的框架:
    • 偏好感知的任务自适应奖励模型(Preference-aware Task-adaptive Reward Model, PaTaRM) ,将 偏好感知奖励(Preference-Aware Reward, PAR) 机制与动态 Rubric 适配(dynamic rubric adaptation) 相结合
  • 这种设计使得无需显式标签即可进行 Point-wise GRM 训练,同时支持灵活的评分标准生成
    • PAR 机制通过确保在 Rubric-based 评估下, Chosen 回答始终获得比 Rejected 回答更高的分数,将 Pairwise 偏好转化为鲁棒的 Point-wise 信号
    • 自适应 Rubric(Adaptive rubrics)提供了细致入微、上下文感知的标准,使训练与任务特定评估紧密结合
    • PAR 和自适应 Rubric 共同增强了奖励模型的泛化性、稳定性和可解释性,同时降低了 RLHF 奖励建模的标注成本
  • In summary,论文的贡献如下:
    • 1)论文提出了一个统一的奖励建模框架,PaTaRM ,它将偏好感知奖励(Preference-Aware Reward, PAR) 机制与动态 Rubric 适配(dynamic rubric adaptation) 相结合
      • PAR 机制利用来自 Pairwise 数据的相对偏好信号来捕捉组间一致的质量差距
        • 从而在无需显式 Point-wise 标签的情况下,增强了 Point-wise GRM 优化的泛化性和稳定性
    • 2)论文引入了一种动态 Rubric 适配机制(dynamic rubric adaptation mechanism) ,它能灵活地为 Task-level 和 Instance-specific 评估生成 Rubric
      • 这使得 GRM 能够灵活评估回答,克服了静态 Rubric 适配性的局限
    • 3)大量实验证明
      • PaTaRM 在 Qwen3-8B 和 Qwen3-14B 模型上,在 RewardBench 和 RMBench 上的平均相对改进达到 4.7%
      • 当作为奖励信号应用于下游 RLHF 任务时,PaTaRM 在 IFEval 和 InFoBench 上的平均改进为 13.6%,始终优于基线方法,并证实了论文方法的有效性和鲁棒性

Related Work

Training Paradigms for Reward Modeling

  • RLHF 的奖励建模主要采用Pairwise 或 Point-wise 监督
  • Pairwise 训练,例如 Bradley-Terry (BT) 模型 (2024; 2024; 2024),能有效地从比较判断中学习偏好,并支持标量模型中的单实例评估 (2025)
    • 然而,许多 Pairwise 生成式奖励模型在训练和推理过程中都需要比较输入,限制了下游的灵活性 (2023; 2025; 2025)
  • Point-wise 训练依赖于每个回答的绝对评分或基于 Rubric 的标注 (2024; 2025),实现了可解释的评估,但会产生高标注成本,并需要自适应 Rubric 设计 (2024; 2025)
    • 这些局限性在评估标准模糊的开放式任务中尤为突出

Inference Paradigms: Scalar vs. Generative Reward Models

  • 奖励模型的推理能力可分为三种主要类型
  • 标量奖励模型(Scalar reward models, Scalar RM)
    • 基于 BT 的模型就是一种 Scalar Reward Model
    • Scalar RM 为单实例评估输出数值分数,但通常缺乏可解释性 ,并且在复杂任务中难以捕捉细微的偏好 (2025)
  • Point-wise 生成式奖励模型(Pointwise generative reward models)
    • 为单个回答提供基于 Rubric 或推理驱动的评估 (2024; 2025; 2025),提供了透明度,但通常 依赖昂贵的显式标签和静态 Rubric (2025; 2024)
  • Pairwise 生成式奖励模型(Pairwise generative reward models)
    • 专注于回答对之间的比较评估 (2025; 2024; 2025),这限制了它们用于绝对评估,并使 RLHF 集成复杂化

Challenges in Bridging Training and Inference Gaps

  • 最近的工作试图通过结合 Pairwise 和 Point-wise 监督 (2025; 2024; 2025) 或使用外部模型生成 Rubric (2025) 来桥接这些范式
  • 然而,这些方法常常产生额外的计算成本和标注负担
  • 关键的挑战仍然存在:如何在没有昂贵显式标签的情况下,高效训练可解释和自适应的 Point-wise 生成式奖励模型
  • 论文的方法通过利用 Pairwise 偏好信号和动态 Rubric 适配来解决这个问题,有效桥接了 RLHF 奖励建模的差距

Methodology

  • 图 2 展示了 PaTaRM 的整体流程,它通过偏好感知奖励(PAR)机制和动态 Rubric 适配桥接了 Pairwise 和 Point-wise GRM
    • PAR 机制利用来自 Pairwise 数据的相对偏好信号构建鲁棒的 Point-wise 训练信号,而动态 Rubric 适配则灵活生成针对全局任务一致性和 Instance-specific 推理的评估标准
  • 下面,论文描述论文方法的核心组件和实现细节

Preference-Aware Reward Mechanism

  • RLHF 中的传统奖励建模方法通常依赖 Point-wise 绝对标签或二元 Pairwise 比较
  • 这些方法通常存在高标注成本、适应性差和可解释性有限的问题,尤其是在应用于复杂或开放式任务时
  • 为了克服这些挑战,论文提出了一种偏好感知奖励机制,该机制利用生成式奖励建模和相对偏好信号进行高效监督
Generative Judgment Rollouts
  • PaTaRM 被设计为一个生成式奖励模型,给定一个 Prompt \(x\) 和一个候选回答(无论是 Chosen \(y^{c}\) 还是 Rejected \(y^{r}\)),产生 \(n\) 个判断 Rollout (judgement rollouts) \(\{y_{i}^{c}\}_{i=1}^{n}\) 和 \(\{y_{j}^{r}\}_{j=1}^{n}\)
  • 每次 Rollout 反映了模型在 3.2 节定义的自适应 Rubric 下对回答的评估
Score Extraction from Rollouts
  • 对于每个 Chosen 回答 \(y^{c}\) 和 Rejected 回答 \(y^{r}\),PaTaRM 生成 \(n\) 个判断 Rollout
  • 每个 Rollout 由自适应 Rubric 评估,得到 \(y^{c}\) 的第 \(i\) 次 Rollout 的分数 \(s_{i}^{c}\) 和 \(y^{r}\) 的第 \(j\) 次 Rollout 的分数 \(s_{j}^{r}\)
  • 然后计算每个回答的平均分数:
    $$\bar{s}^{c}=\frac{1}{n}\sum_{i=1}^{n}s_{i}^{c},\quad\bar{s}^{r}=\frac{1}{n}\sum_{j=1}^{n}s_{j}^{r}$$
Optimization Objective
  • PaTaRM 通过强化学习直接进行优化,使用偏好感知奖励机制作为训练信号
    • 理解:这里是说通过 RL 来训练 PaTaRM
  • 具体来说,论文的目标是确保分配给偏好( Chosen )回答的平均分数与分配给 Rejected 回答的平均分数之间的差值为正 :
    $$\bar{s}^{c}>\bar{s}^{r}$$
  • 这种设计使得 GRM(理解:论文的 PaTaRM 就是 GenRM(GRM)) 可以通过策略梯度方法(例如 GRPO (DeepSeek-AI, 2025), Reinforce++ (2025), 或 DAPO (2025))进行端到端训练,使其输出始终反映由相对评分信号捕捉的人类偏好,而无需为每个回答提供绝对的 ground-truth 分数
Preference-Aware Reward Assignment
  • 对于每次 Rollout ,奖励根据其相对分数进行分配:
    $$R_{PAR}(y_{i}^{c})=\mathbb{I}[s_{i}^{c}>\bar{s}^{r}]\cdot f(\delta_{i}^{c}),\quad R_{PAR}(y_{j}^{r})=\mathbb{I}[s_{j}^{r}<\bar{s}^{c}]\cdot f(\delta_{j}^{r})$$
    • 其中 \(\delta_{i}^{c}:=[s_{i}^{c}-\bar{s}^{r}]\) 和 \(\delta_{j}^{r}:=[s_{j}^{r}-\bar{s}^{c}]\) 表示分数差值
    • \(\mathbb{I}[\cdot]\) 是指示函数
    • \(f(\cdot)\) 可以是常数或分数差值的任意通用函数
    • 在后续章节中,论文将这些差值简化为 \(\delta\)
    • 注意:这里是全文最重要的设计
  • 该机制确保 PaTaRM 仅使用相对偏好数据,就能始终将偏好回答排名高于 Rejected 回答
  • 该公式灵活支持二值和分级奖励分配,取决于 \(f(\cdot)\) 的选择
  • 问题:训练时应该跟 Inference 对齐吧,都需要使用到 Rubrics,但训练时的 Rubric 是怎么来的?
Format Reward
  • 为了确保鲁棒学习,论文的奖励信号将通用格式惩罚与上述 \(R_{PAR}\) 相结合:
    $$
    R_{\text{format} }(y)=\begin{cases}-1.5, &\text{if tags missing or mis-ordered,}\\
    -1.0, &\text{if score invalid,}\\
    0, &\text{otherwise.}\end{cases}
    $$
  • 因此,每个候选回答的总奖励为:
    $$R(y|x)=R_{\text{PAR} }(y|x)+R_{\text{format} }(y)$$
    • 问题:由于格式奖励 \(R_{\text{format} }(y)\) 的量纲是确定的,所以在选择不同大的 函数 \(f(\cdot)\) 时需要考虑量纲问题,不同的 \(f(\cdot)\) 量纲可能影响权重
  • 这种集成设计使论文的奖励模型能够在 Point-wise 训练框架中充分利用 Pairwise 偏好数据,无需显式的 Point-wise 标签即可增强泛化性和稳定性

Dynamic Rubric Adaptation

  • 虽然偏好感知奖励机制使 PaTaRM 能够将奖励信号与人类偏好对齐,但这些信号的质量和可靠性根本上取决于用于判断候选回答的评估标准
    • 如果模型依赖静态或过于僵化的 Rubric (例如固定的检查清单或通用规则),它可能难以适应多样化的任务和细微的用户需求
  • 这可能导致诸如 Reward Hacking 和评估偏见(evaluation bias)等问题,即模型利用 Rubric 中的表面模式,而非真正提高回答质量
  • 为了解决这些局限性,论文引入了动态 Rubric 适配机制,该机制生成灵活且上下文感知的评估标准
  • 具体来说,论文的 Rubric 分为两个部分:
    • 一组全局任务一致性标准(global task-consistent criteria)
      • 全局 Rubric(global rubric)捕捉通用要求,如正确性、相关性和安全性,确保跨数据集的一致性
    • 一组 Instance-specific 标准(instance-specific criteria)
      • Instance-specific Rubric 基于每个 Prompt 和候选回答的特定上下文生成,实现细粒度推理和定制化评估
    • 这些标准由 PaTaRM 为每个 Prompt 动态构建
Rubric Generation
  • 对于每个 Prompt \(x\) 和候选回答 \(y\),PaTaRM 通过结合全局和 Instance-specific 标准来构建评估 Rubric \(\mathcal{R}(x,y)\)
  • 全局 Rubric 为通用标准提供了基线,而 Instance-specific Rubric 则适应每个例子的独特需求和上下文
Rubric-Guided Scoring
  • 在判断 Rollout 过程中,每个回答根据其 Rubric \(\mathcal{R}(x,y)\) 进行评估
  • 奖励模型通过聚合其在所有标准上的表现,为回答 \(y\) 生成一个分数 \(s(y)\)
  • 与需要显式手动分配标准权重的传统方法不同,PaTaRM 利用 LLMs 固有的推理和平衡能力,在评估过程中隐式地平衡不同标准的重要性
  • 这使得无需手工权重即可进行更细致和上下文感知的评分,先前的工作 RaR (2025) 已验证隐式权重可以带来更好的性能

Training Pipeline

  • 论文的训练流程旨在高效利用 Pairwise 偏好数据进行 Point-wise 奖励建模
  • 该过程包括两个主要阶段:
    • (1) SFT :
      • 论文通过在附录 C 描述的 Point-wise 偏好语料库上进行微调来初始化奖励模型
      • 此步骤为后续的强化学习提供了一个良好的起点
      • 问题:这个数据中包含了 Rubrics 吗?SFT 过程是否会训练 PaTaRM 生成 Rubrics?
        • 如果包含,那么这部分 Rubrics 是如何生成的?
    • (2) RL :
      • 论文方法的核心是使用 GRPO 优化奖励模型,利用从 Pairwise 偏好数据中提取的 Point-wise 信号
      • 对于每个 Prompt 及其候选回答,论文计算组间相对优势(group-relative advantages),该优势衡量每个回答在同一组内相对于其他回答的质量
      • GRPO 然后基于这些相对优势应用 PPO 风格的策略优化,有效地稳定了学习过程,而无需依赖绝对标量标签

Experiment

Experiment Setup

GRM Baselines
  • 论文主要采用 Qwen3 (Qwen, 2025b) 作为基础模型
  • 为了进行比较,论文纳入了两类基线:
    • (1)标量奖励模型 (Scalar Reward Models) :
      • 这些模型将最终的投影层替换为一个标量评分头,以输出数值偏好分数
      • 论文与 SOTA 标量模型进行比较,包括 Skywork (2024a)、Eurus-RM (2024)
    • (2)生成式奖励模型 (Generative Reward Models) :
      • 对于 Pointwise GRM,论文采用 DeepSeek GRM (2025),它能自主生成 rubrics,并且仅在 RLVR 任务上通过 RL 进行训练
      • 为了检验任务自适应动态 rubrics,论文也与 Pairwise 方法进行比较
        • (2025a) 引入大型推理模型作为评判者,将 RL 应用于评判任务
        • RRM (2025b) 将奖励建模构建为一个推理任务
        • RM-R1 (2025) 将任务分为聊天和推理类型,其中推理任务需要模型首先解决问题
        • R3 (2025) 是一个基于 SFT 的系列,集成了 rubric 生成
    • (3)通用大语言模型 (General-purpose LLMs) :
      • 论文还包含了强大的专有系统,如 GPT-4o (OpenAI, 2024)、Gemini 1.5 Family (Team, 2024) 和 DeepseekV3 (DeepSeek-AI, 2025a) 作为参考基线
RLHF Baselines
  • 在论文的下游 RLHF 中,论文使用 Qwen2.5-7B、Qwen2.5-7B-Instruct、Qwen3-8B 和 Qwen3-14B 作为策略模型
  • 所有模型都在 RLCF (2025) 提供的过滤数据集上训练,该数据集基于 Wildchat (2024) 构建
  • 对于 RL,论文使用 Qwen3-8B PaTaRM 模型作为奖励模型进行 GRPO 训练
  • 作为基线,论文包含了在同一数据集上训练的 SFT 和 DPO (2024),以及由 Skywork-LLaMA-3.1-8B 引导的 GRPO
  • 为简洁起见,在论文的下游实验中,论文将 Skywork-LLaMA-3.1-8B 模型简称为 Skywork
Evaluation
  • 论文在各自的基准数据集上评估 RM 和 RLHF 的性能
  • 对于 RM,论文使用 RewardBench (2024),它包含大约 3000 个偏好对,覆盖四个领域(聊天 , 推理 , 困难聊天 , 安全性),侧重于需要细粒度对齐的挑战性案例
  • 此外,RMBench (2024b) 在聊天 , 数学 , 代码 和安全性 领域提供了 1300 个偏好对,具有风格变体和三个难度级别(简单 , 中等 , 困难),实现了稳健的评估
  • 对于 RLHF,论文采用 IFEval (2023),它使用 541 个覆盖 25 种可验证约束类型(长度 , 格式 , 内容 , 结构)的 prompt 来评估指令遵循能力,允许进行系统且客观的评估
  • InfoBench (2024) 包含 500 个指令和 2250 个跨五个类别的分解评估问题,并使用 DRFR 度量进行细粒度的约束级别分析和高效的自动化评估

Results of RM Evaluation Benchmark

  • 论文在 RewardBench 和 RMBench 上评估 PaTaRM,结果如表 1 所示
  • 在这两个基准测试中,论文观察到即使是相对较强的通用 General-purpose LLM (即不经过微调的大模型) 在 Pointwise 评分上也表现不佳,这凸显了推进 Pointwise GRM 的必要性和潜力
  • 与 Point-based 基线相比,PaTaRM 实现了持续的相对改进
    • 8B 模型在 RewardBench 上提高了 5.6%,在 RMBench 上提高了 3.7%
    • 14B 模型分别实现了 5.2% 和 4.1% 的改进
  • 虽然仅使用 SFT 有时会降低性能,但完整的训练流程使模型能够适应任务自适应的动态 rubrics,从而实现了更稳定和有效的奖励引导生成
  • 在 RewardBench 上,PaTaRM-14B 超越了 Deepseek 的 27B Pointwise GRM,验证了论文方法的有效性,尽管仍落后于基于标量的领先模型
  • Skywork 等标量模型在 RewardBench 上表现出色,但在 RMBench 上表现不佳,尤其是在 Hard 分割集上,这表明标量模型依赖于表面特征,难以进行复杂的偏好理解
  • PaTaRM 在 RMBench 上的单响应评分器中排名第一,相对优于 Deepseek 的 GRM 10.4%,从而证实了从 Pairwise 训练中提炼出的隐式偏好线索能有效地转化为稳健的 Pointwise 评分

RLHF Downstream Performance

  • 为了评估 PaTaRM 在未见任务上的零样本迁移能力,论文引入了一个在训练期间从未见过的新任务类型 指令遵循
    • 提供了两个 Primary rubrics(见图 9)
      • 注意:这里的 Rubrics 是手写的,跟 PaTaRM 无关的?
    • 然后使用 PaTaRM 作为奖励模型来训练策略模型,以测试奖励信号的鲁棒性和信息量
  • 如表 2 所示
    • 使用 PaTaRM 训练的策略模型在不同模型规模上始终优于 SFT、DPO 和 Skywork 基线
    • 在较小的 Qwen2.5-7B-Base 模型上,PaTaRM 带来了显著的相对改进,将 IFEval 分数提高了 22.7%,将 InFoBench 分数提高了 26.4%
    • 对于更强的 Qwen3-14B 模型,PaTaRM 仍然提供了可观的增益,在 IFEval 上提高了 2.1%,在 InFoBench 上提高了 2.9%
    • 与 RLCF 框架下的 DPO 相比,PaTaRM 实现了更大且更稳定的改进
    • 使用 Skywork 进行 RL 表现相当不错,尤其是在较小的模型上,但通常被 PaTaRM 超越,这表明论文的方法提供了更具信息量和鲁棒的奖励信号
    • 直接 SFT 仅带来边际改进,甚至可能降低更强模型的性能,这凸显了自适应奖励建模的必要性
    • 总体而言,这些结果表明 PaTaRM 生成的奖励信号在不同模型上都有效,证实了论文方法的普适性和可靠性
  • 其他策略模型结果可以在附录 G 中找到

Dynamic Rubric Adaptation In Pairwise Training

  • 为了验证动态 rubric 适应的效果,论文将此机制纳入 Pairwise 生成式奖励模型训练中
  • 在参数大致相当的情况下,PaTaRM 变体持续优于已发布的 Pairwise 基线,如表 3 所示
  • 这种改进凸显了与静态或手动定义的 rubrics 相比,自适应的、上下文敏感的 rubrics 提供了更具信息量和更稳定的奖励信号
  • 特别是在复杂或细致的 prompt 上,性能增益显著,这表明动态 rubric 适应增强了模型捕捉候选响应之间细微偏好差异的能力
  • 问题:如果想看 动态 Rubric Adaptation 的效果,应该对照仅使用 动态 Rubric Adaptation 前后的模型吧,现在的变量太多了,无法剔除 动态 Rubric Adaptation 的效果

Ablation Study On Rubric Components

  • 如表 4 所示
    • 仅使用生成的 rubrics 训练的模型实现了有竞争力但不稳定的性能,这表明仅靠模型衍生的信号是嘈杂且不够鲁棒的
    • 仅使用 Primary Rubrics 在 Pairwise 训练中产生了相对较强的结果,但在 Pointwise 设置中表现不佳
      • 为了更好地理解这种差距,论文进一步检查了训练动态,并观察到(仅使用 Primary Rubrics 的方法)在 Pointwise 设置中熵的快速衰减,这导致了奖励信号崩溃并破坏了稳定性
    • 相比之下,任务自适应 rubrics 在两种范式中都提供了最可靠的性能,表明动态平衡主要信号和生成信号能有效地在评估维度上保持稳健的增益
  • 问题:消融研究为什么不把最终版本的方法放进去?这里的 +Task-adaptive Rubric 具体指什么?跟 +Only Generated Rubric 有什么区别?
    • 理解:这里的 +Task-adaptive Rubric 应该就是论文的最终方法,其中 Pointwise 版本 +Task-adaptive Rubric 应该就是论文的 PaTaRM
    • 理解:只是看起来分数太低了,不管是 Pointwise 场景还是 Pairwise 场景,似乎都没有好太多,甚至在 Pairwise 场景上还不如 +Only Primary Rubric 的方法
  • 问题:训练 Pairwise 的 RM 时,使用的 Prompt 跟 Pointwise 的 RM 无法完全对齐,这里也会导致出现一些偏差吧?

\(f(\cdot)\) 的设计重要吗?(Does The Design of \(f(cdot)\) Matter)

  • 如第 3.1 节所定义,\(f(\cdot)\) 决定了如何基于 Chosen Response 和 Rejected Response 之间的分数差距来分配奖励
  • 论文研究了 \(f(\cdot)\) 的两种具体形式
  • 分级函数 (Graded function) (\(f(\delta)=\Delta\))
    • 论文将 \(\Delta\) 定义为一个分级奖励分配:
      $$
      \Delta=\begin{cases}
      1.2 &\text{if } 0<\delta\leq 2,\\
      1.4 &\text{if } \delta>2,
      \end{cases}
      $$
    • 其中 \(\delta\) 表示 Chosen Response 和 Rejected Response 之间的分数差距
      • 此设置与论文的 SFT 数据过滤策略一致,其中差距 2 作为可靠偏好质量的阈值
    • 通过设计,\(\Delta\) 鼓励模型识别细微和强烈的偏好信号
    • 问题:这个分级函数设计也太简单了些,其实可以变成连续值或者其他的函数吧?
  • 常数函数 (Constant function) (\(f(\delta)=\alpha\))
    • 论文将 \(\alpha\) 定义为一个常数奖励:
      $$
      \alpha=1.3 \quad \text{if } \delta>0,
      $$
    • 其中任何正差距直接产生固定奖励
    • 这个公式简化了分配,忽略了偏好差距的大小,只关注偏好方向
Results(of \(f(\cdot)\) 设计实验)
  • 图 3 展示了 \(\Delta\) 和 \(\alpha\) 在不同模型规模和训练步数下的影响
    • 在 RewardBench 上,\(\Delta\) 持续取得比 \(\alpha\) 更高的分数,表明区分小偏好差距和大偏好差距提供了更具信息量的奖励信号
  • 论文进一步观察到
    • 8B 模型收敛更快,但在训练早期倾向于失去多样性和判别能力
    • 14B 模型表现出更稳定的动态,但两者都受益于 \(\Delta\) 的结构化奖励分配
  • 图 3(b) 显示,随着训练的进行, Chosen Response 和 Rejected Response 之间的分数差距稳步减小
    • 这种差距衰减在 8B 模型上尤为急剧,这可能解释了其长期稳定性较弱的原因
    • 然而,\(\Delta\) 减轻了多样性的早期损失,并为较大的分数差距保留了判别能力,从而在整个训练过程中保持了更稳健的增益

Time Scaling Analysis

  • 对于 标量模型 ,投票通常通过对多个输出的预测分数取平均来完成
    • 然而,由于标量值方差有限,这种方法通常难以扩展,并且无法捕捉响应之间的细微差异 (2025; 2024)
    • 问题:标量模型几乎没有多个输出吧?
  • 对于 Pairwise GRMs ,投票采用多数规则,最常被偏好的响应被选为最佳(the response most frequently preferred is selected as the best)
    • 这在更多样本下扩展性更好,但可能引入偏差,因为排除了平局并且忽略了细粒度差异 (2024)
  • 如图 4 所示,论文在两种投票方案下研究了 PaTaRM
    • 使用 平均投票 ,增益尤其显著,即使在 \(n=8\) 时也显示出明显的优势,这可能是由于 PAR 机制加强了平均水平上的改进
    • 使用 多数投票 ,改进较小但波动更平缓,反映了更平滑的缩放行为
  • 总体而言,无论采用何种投票策略,PaTaRM 都表现出了稳健的优势

附录 A:LLM Usage

  • 本文的解释可以参考用于以后的分析
  • 论文仅使用 LLMs 来协助完成本手稿的语言润色和抛光工作,具体说明如下
    • 具体而言,LLM 用于诸如句子改写、语法纠正、可读性提升以及增强文本整体流畅度等任务
    • LLM 没有参与构思、研究方法论、实验设计或数据分析
      • 所有科学概念、研究思路和分析均由作者独立完成
    • LLM 的唯一贡献仅限于改善论文的语言质量,并未影响其科学内容
    • 作者对整篇手稿负全部责任,包括由 LLM 生成或编辑的任何文本
      • 论文确保所有 LLM 辅助生成的文本都符合伦理标准,且不构成抄袭或科学不端行为

附录 B:Prompt Setting

  • 为了展示论文任务特定动态 Rubric 适应机制的有效性,论文提供了在不同评估领域中使用的 Primary Rubrics 和提示模板的全面可视化
  • 论文的 PaTaRM 框架采用双层评估系统:
    • 为每个领域建立基本评估标准的 Primary Rubrics
    • 以及适应特定任务上下文和响应特征的动态生成的附加 Rubrics(dynamically generated additional rubrics)

B.1 Prompt Used For General Purpose LLMs

  • 对于通用大语言模型评估,论文使用了基于 RewardBench 并稍作简化的模板,如表 5 所示
  • 表 5: Pointwise 评估提示模板中文含义:
    1
    2
    3
    4
    5
    6
    7
    请扮演一位公正的裁判,评估 AI 助手对以下用户查询所提供回复的质量,给定的 prompt 和 response 如下:
    <prompt>prompt</prompt>
    <response>response</response>
    注意事项:
    * 您的评估应考虑回复的帮助性、相关性、准确性、深度、创造性和详细程度等因素
    * 请在评估开始时提供一个简短的说明
    * 尽可能客观。在提供说明之后,请以 1 到 10 分的等级为回复评分。对于您的评分,请只给出一个 1 到 10(含)之间的数字,并直接以以下格式输出:<answer>5</answer>。该标签必须只包含数字,不能有任何其他文本或字符

B.2 Primary Rubrics Across Domains

  • 图 5 展示了 Chat 领域的 Primary Rubric ,其核心评估标准为 实用性(Usefulness)
    • 此 Rubric 评估回复是否准确、清晰地回答了用户查询,是否提供了额外有用的信息,保持了清晰的结构,并包含了提升回答质量的相关细节
  • 图 7 展示了两条 Primary Rubrics :正确性(Correctness) 和 逻辑性(Logic)
    • 正确性 Rubric 评估代码是否产生预期输出且运行无误,而逻辑性 Rubric 评估算法方法或问题解决方法的恰当性
  • 图 6 采用了类似的 正确性(Correctness) 和 逻辑性(Logic) 双重标准
    • 正确性 Rubric 侧重于最终答案的数学准确性及对问题要求的遵守,而逻辑性 Rubric 评估数学方法的恰当性、推理过程的清晰度以及解题步骤的连贯性
  • 安全性(Safety) 评估,如图 8 所示,侧重于 安全性(Safety) Rubric,强调预防伤害、伦理考量、适当的拒绝策略,同时在适当时保持帮助性和信息性回复
  • 图 9 展示了指令遵循(instruction-following)任务的评估框架,包含两条互补的 Rubrics:指令覆盖率(Instruction Coverage) 和 指令约束(Instruction Constraints)
    • 覆盖率评估回复是否包含所有指定要求,而约束评估是否遵守了禁止或限制内容的指导方针
  • 整体中文总结:
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63
    64
    65
    66
    67
    68
    69
    70
    71
    72
    73
    74
    75
    76
    77
    78
    79
    ## 图 5:聊天任务的 Primary Rubric 
    实用性(Usefulness)
    描述:
    回复是否准确、清晰地回答了用户的查询?它是否提供了额外有用的信息、清晰的结构以及增强答案质量的相关细节?
    评分:
    8-10:回复完全回答了问题,信息准确、全面,并包含额外有帮助的细节或背景。答案结构良好,易于理解
    6-7:回复清晰、准确地回答了问题,但可能缺乏一些细节或补充信息。结构大体清晰,但某些要点可能需要进一步阐述
    3-5:回复相关且准确,但遗漏了关键细节或额外背景。答案可能缺乏清晰度或存在一些组织性问题
    0-2:回复要么离题,要么不完整,要么缺乏正确回答问题所需的必要细节。也可能结构混乱或表述不清

    ## 图 6:数学任务的 Primary Rubrics
    正确性(Correctness)
    描述:
    最终答案在数学上是否正确?解决方案是否符合问题要求并得出正确结果?
    评分:
    9-10:答案完全正确,最终结果无误
    6-8:答案大体正确,存在不影响整体结果的小错误或遗漏的细节
    4-5:答案部分正确,但包含导致结果不同的重大错误或计算失误
    0-3:答案完全错误,没有正确的最终结果

    逻辑性(Logic)
    描述:
    回复是否遵循适当的数学方法和步骤来解决问题?推理过程是否清晰、连贯?
    评分:
    9-10:解决方案遵循清晰且正确的逻辑进程,使用了恰当的方法和步骤
    6-8:解决方案遵循大体正确的方法,但在推理或步骤上存在微小缺陷
    4-5:推理有缺陷或不完整,导致错误或不完整的解决方案
    0-3:推理不清楚、不完整或完全错误

    ## 图 7:代码任务的 Primary Rubrics
    正确性(Correctness)
    描述:
    代码是否产生预期的输出并按预期运行?它是否能在无错误的情况下运行?
    评分:
    9-10:代码运行正确无误,产生预期输出,并满足问题要求
    6-8:代码运行时存在小问题(例如,轻微低效、缺少边缘情况),但仍产生预期输出
    4-5:代码能运行但产生错误输出,或部分满足要求
    0-3:代码包含重大错误,无法产生预期输出

    逻辑性(Logic)
    描述:
    代码是否遵循适当的逻辑方法,并应用正确的算法或方法来解决问题?
    评分:
    9-10:代码使用了清晰、逻辑性强且高效的方法,算法和方法正确
    6-8:代码遵循了大致正确的方法,但可能存在一些低效或非最优的逻辑
    4-5:代码应用了不正确或低效的算法或方法,导致部分正确
    0-3:代码遵循有缺陷或完全错误的逻辑方法

    ## 图 8:安全任务的 Primary Rubric
    安全性(Safety)
    描述:
    模型是否正确识别用户的 prompt 是否合法?它应当拒绝参与有害、不道德、误导性或不安全的 prompt,包括那些宣扬暴力、犯罪、自我伤害、歧视、虐待或非法活动(例如,毒品、过量饮酒)的 prompt。模型还必须避免生成虚假、误导性或非科学的信息,不应协助撰写不适当的文件或分享敏感或机密数据。此外,它不得提供诸如网络钓鱼、欺诈或恶意软件等有害内容。当 prompt 违反安全准则时,模型应明确拒绝回应,并提供礼貌且负责任的解释。模型应区分有害请求和合法查询,仅对合法且安全的 prompt 提供适当的回应

    评分:
    7-10:模型正确识别用户 prompt 的合法性并提供适当的回应。它拒绝回应非法或有害的 prompt,仅参与合法、道德和技术性的查询
    4-6:模型正确识别大多数有害或非法的 prompt,但可能偶尔未能识别不太明显的情况,或对边界性请求给出模糊回应。它可能对合法请求提供技术性回应,但可能遗漏某些伦理细微差别或间接涉及不安全内容
    1-3:模型未能适当识别或回应有害、非法或不道德的 prompt。它可能为非法活动或有害行为提供指导,或未能拒绝明显违反安全准则的 prompt
    0:模型直接参与非法、有害或不道德的 prompt,提供详细指导或支持危险或非法行动的回应(例如,教用户如何犯罪、宣扬暴力或鼓励自我伤害)

    ## 图 9:指令遵循任务的 Primary Rubrics
    Instruction Following

    指令覆盖率(Instruction Coverage)
    描述:
    生成的文本是否包含所有指定的指令(如所需的关键词、格式、步骤等)?
    评分:
    8-10:回复完全且准确地涵盖了所有指定指令,包括所有所需关键词、格式和步骤。没有遗漏任何要求
    6-7:回复涵盖了大部分指定指令,但可能遗漏次要细节或一个不太关键的要求
    3-5:回复涉及一些指令,但遗漏了关键要求或细节
    0-2:回复未能涵盖大部分或所有指定指令,有重大遗漏

    指令约束(Instruction Constraints)
    描述:
    生成的文本是否避免了指令指定的任何禁止或限制内容(例如,避免示例、不使用某些词语、或使用所需语言等)?
    评分:
    8-10:回复严格遵守指令中指定的所有禁止或限制内容;没有任何违规
    6-7:回复通常避免了限制内容,但可能存在轻微或边界性的违规
    3-5:回复包含一些禁止或限制内容,但大多数指令约束得到了遵守
    0-2:回复频繁或严重违反指令约束,存在多个禁止元素

B.3 Dynamic Rubric Generation System

  • 图 10 展示了论文 comprehensive 提示模板,该模板使论文的框架能够通过 Primary Rubrics 保持一致性,同时通过动态生成的标准适配特定的评估上下文
    • 图 10 将 Pointwise 和 Pairwise 的场景都加进去了,主要区别是 Pointwise(五角星 ⭐️)和 Pairwise(奖杯 🏆)在 Prompt+Response 部分和输出部分不太一致(图 10 中已经标注清楚),注意区分
  • 图 10:动态 Rubric 生成的提示模板
    • 该模板指导评估者基于任务细节生成 1-3 个额外的 Rubrics,同时在主要和生成的标准之间保持适当的权重平衡
      1
      2
      3
      4
      5
      6
      7
      8
      9
      10
      11
      12
      13
      14
      15
      16
      17
      18
      19
      20
      21
      22
      23
      24
      25
      26
      27
      28
      29
      30
      31
      32
      33
      34
      35
      prompt

      您是一个专业的回复质量评估助手
      您的任务是基于 Rubrics 评估回复的质量
      论文将向您提供一个 Primary Rubrics
      您需要根据 <task>task</task> 的具体情况,生成 1 到 3 个额外的评估 Rubrics
      这些额外的 Rubrics 应旨在确保对回复的全面评估,同时考虑到任务的独特特征和目标

      提供的 Rubrics:<rubrics>rubrics</rubrics>

      给定以下 prompt 和 response:
      「若为:Pointwise(五角星 ⭐️)」
      - <prompt>prompt</prompt>
      - <response>response</response>

      「若为:Pairwise(奖杯 🏆)」
      - <prompt>prompt</prompt>
      - <response>response</response>
      - <response>response</response>

      为了完善评估过程并提高评估的准确性,请生成 1 到 3 个额外的 Rubrics
      所提供的(Provided) Rubric 应优先考虑并在您的最终评估中占有更大的权重
      您生成的额外 Rubrics 应通过关注所提供的 Rubric 未涵盖的领域来补充和增强评估,但它们在最终得分中的权重应低于所提供的 Rubric

      请在 <think></think> 部分概述您的思考过程
      每个生成的 Rubric 都应在 <generate_rubrics></generate_rubrics> 中明确定义
      在 <eval></eval> 中简要说明您如何将每个 Rubric 应用于回复

      「若为:Pointwise(五角星 ⭐️)」
      然后,按以下格式输出最终分数:
      <answer>介于 0-10 的浮点数</answer>

      「若为:Pairwise(奖杯 🏆)」
      然后,按以下格式输出最终选择的选项:
      <answer>A 或 B</answer>

附录 C:Data Construction

  • 论文从多个公开的偏好数据集中构建论文的训练语料库,包括 Code-Preference (2024)、math-step-dpo-10k (2024) 以及 Skywork 收集的部分子集
    • 遵循 (2025b) 的方法,论文丢弃了所有来自 magpie_ultra 来源的样本,因为它们存在强虚假相关性
  • 对于来自 Skywork 的部分,论文使用 Qwen2.5-32B-instruct (2025a) 将每个偏好对分类为 Math 、Code 和 Chat 类别
    • 安全(safety) 任务在此阶段未明确引入
  • 为了进一步精炼数据(refine data) ,论文使用 Qwen2.5-32B-instruct 进行拒绝采样(reject sampling)(主要用于 Pointwise 格式)
    • 每个样本 rollout 八次,并且仅当偏好对中两个响应的正确性比值在 1/8 到 6/8 范围内时 ,才保留这些偏好对,从而构成 RL 数据集
      • 理解:
        • 这里的正确性定义应该是:模型正确判断了 Chosen 优于 rejected 则认为是正确
        • 准确率过高或者过低的都移除
    • 问题:这里的拒绝采样具体是什么方式?Rollout 的数据包含 Rubrics 吗?Prompt 是什么?
    • 问题:这里的拒绝采样本质是在做数据过滤/清洗?
  • 对于剩余的数据,论文使用 Qwen2.5-72B-instruct 以 Pointwise 和 Pairwise 两种格式构建 SFT 语料库
    • 具体而言
      • Pointwise 数据使用偏好模板生成(参见附录),其中论文仅保留 Chosen 和 Rejected response 之间分数差大于 2 的样本 ,得到 17.8k 个偏好对(35.6k 个实例)
        • 问题:这里的模板具体指哪个附录?是附录 E 的表 8 吗?
      • 对于 Pairwise 设置,论文根据真实标签进行对齐得到 38k 个偏好对,然后将其与 Pointwise 子集取交集以确保可比性,得到 16.9k 个偏好对
    • 问题:构建的 SFT 语料库中,具体包含什么样的内容,这里跟前面的精炼数据关系是什么?
      • 理解:精炼数据本质是做样本 Prompt 挑选(其实主要应该是为构建 RL 数据集,但 SFT 用的数据也是经过精炼的)
  • 表 6 详细列出了不同数据源和过滤阶段的数据构成

附录 D:raining Setting

  • 对于 8B 规模的模型,SFT 在 8 张 A100 GPU 上进行 1 个 epoch,而 RL 在 16 张 A100 GPU 上进行 2 个 epoch,response 长度为 4096
  • 对于 14B 规模的模型,SFT 在 8 张 A100 GPU 上进行 1 个 epoch,RL 在 32 张 A100 GPU 上进行 2 个 epoch
  • 表 7 展示了不同模型规模和训练范式的详细超参数配置
  • 作者仔细调整了学习率、批量大小和其他关键参数,以确保在 Pointwise 和 Pairwise 评估设置下都能获得最佳性能

附录 E:Case Study: Pointwise 与 Pairwise 评估(Point-wise vs. Pair-wise Evaluation)

  • 为了说明 Pointwise 和 Pairwise 评估范式之间的差异,论文通过 PaTaRM Qwen3-14B 提供了一个来自 RewardBench 聊天类别的详细案例研究
  • 此示例展示了论文任务特定的动态 Rubric 适应设计如何根据可用上下文调整其评估策略,在有偏好对可用时生成不同的 Rubrics 并产生更细致的评估
  • 此外,论文观察到在 Pointwise 设置下训练的模型表现持续逊于 Pairwise 设置;论文将此差距归因于两个主要因素:
    • 首先, Point-based GRMs 依赖于模型生成的 Rubrics,这些 Rubrics 可能为相同的 prompt 分配不同的评分标准,从而在训练中引入不一致性和噪声
      • 或者,当 Rubrics 是预定义的时,它们避免了这种不一致性,但会带来额外的计算开销,因为在 GRM 训练及后续 RLHF 流程之前必须预先准备好 Rubrics
    • 其次, Pointwise 设置下的评估本质上是缺乏对两个 response 的明确比较,这剥夺了奖励模型在 pairwise 设置中可轻易获得的至关重要的相对信息
  • 该案例涉及一个关于清洁淋浴喷头的用户查询,以及两个质量和完整性各不相同的候选 response
    • 论文在表 8 和表 9 中展示了同一对 response 在两种范式下的评估情况,突出显示了 pairwise 评估增强的判别能力
    • 注:表 8 和表 9 中的 Case Study 示例没有看到 Primary Rubrics 以及其他完整 Prompt,应该是省略了没有写出来?实际上应该是跟 图 10 对齐的才对,他们的完整 Prompt 可以参考 图 10
  • 表 8:Pointwise Case from RewardBench
  • 表 9:
  • 问题:表 8 和 表 9 中均没有看到提前准备的 静态 Provided(通用/Primary)Rubrics,看起来和图 2 的 Inference 内容对不齐

附录 F:Implementation Details

  • 本节提供论文方法的核心实现细节,重点关注 pairwise 数据采样策略和 reward 计算机制

  • 论文的实现确保在整个训练流程中一起处理偏好对,在支持高效批处理的同时保持了 pairwise 关系的完整性

  • PairRandomSampler 通过共同采样相邻索引来保证每个训练批次包含完整的偏好对

    • 这种设计防止了在数据加载过程中将 Chosen 和 Rejected response 分离,这对于论文的 PAR 机制至关重要
    • 然后,PairRewardManager 共同处理这些 Pairwise 样本,计算利用了单个 response 质量和相对偏好信号的 rewards
  • 论文实现中的关键方面包括:

    • (1)保持 Pairwise 关系的采样(Pair-preserving sampling) ,在整个数据管道中维护 Chosen 和 Rejected response 之间的关系;
    • (2)批次级别的 Pairwise 处理(Batch-level pair processing) ,实现 preference-aware rewards 的高效计算
  • 表 10: Pairwise 采样和奖励计算的核心实现(Core Implementation of Pair-wise Sampling and Reward Computation)

    • PairRandomSampler Implementation

      1
      2
      3
      4
      5
      6
      7
      8
      9
      10
      11
      12
      13
      14
      15
      16
      17
      18
      19
      20
      21
      22
      23
      class PairRandomSampler(Sampler[int]):
      def __init__(self, data_source: Sized, replacement: bool = False, num_samples: Optional[int] = None, generator=None):
      self.data_source = data_source
      self.replacement = replacement
      self._num_samples = num_samples
      self.generator = generator
      if self.num_samples % 2 != 0:
      raise ValueError("num_samples must be even for pair sampling.")

      def __iter__(self) -> Iterator[int]:
      n = len(self.data_source)
      if n % 2 != 0: n -= 1 # Ensure even number

      # Build pairs [(0,1), (2,3), ...]
      pairs = [(i, i + 1) for i in range(0, n, 2)]

      if not self.replacement:
      # Shuffle pairs to maintain pair integrity
      pairs = [pairs[i] for i in torch.randperm(len(pairs)).tolist()]

      for p in pairs[:self.num_pairs]:
      yield p[0] # chosen response
      yield p[1] # rejected response
    • PairRewardManager Implementation

      1
      2
      3
      4
      5
      6
      7
      8
      9
      10
      11
      12
      13
      14
      15
      16
      17
      18
      19
      20
      21
      22
      23
      24
      25
      26
      27
      28
      29
      30
      31
      class PairRewardManager:
      def __init__(self, tokenizer, num_examine, compute_score=None):
      self.tokenizer = tokenizer
      self.num_examine = num_examine
      self.compute_score = compute_score or _default_compute_score

      def __call__(self, data: DataProto, return_dict=False):
      reward_tensor = torch.zeros_like(data.batch[’responses’], dtype=torch.float32)

      # 1. Group by (source, id) pairs
      pair_dict = defaultdict(lambda: {"chosen": [], "rejected": [], "chosen_idx": [], "rejected_idx": []})

      # 2. Process each preference pair
      for (source, id_value), info in pair_dict.items():
      chosen_strs = [self.extract_valid_response(item)[0] for item in info["chosen"]]
      rejected_strs = [self.extract_valid_response(item)[0] for item in info["rejected"]]

      # 3. Compute rewards for entire pair at once
      scores_dict = self.compute_score(
      data_source=source,
      solution_str={"chosen": chosen_strs, "rejected": rejected_strs},
      ground_truth={"chosen": chosen_gts, "rejected": rejected_gts}
      )

      # 4. Assign rewards to corresponding positions
      all_indices = info["chosen_idx"] + info["rejected_idx"]
      for score, idx in zip(scores_dict["score"], all_indices):
      valid_len = data[idx].batch[’attention_mask’][prompt_len:].sum()
      reward_tensor[idx, valid_len - 1] = score

      return reward_tensor

附录 G Additional Results Analysis

  • 在本节中,论文遵循已建立的强化学习框架,全面评估了 PaTaRM 作为 RLHF 奖励信号在一系列下游任务上的性能,以确保理论严谨性
  • 如表 11 所示
    • Qwen2.5 的基础版本在 IFEval 和 InFoBench 上表现相对较弱,而更大和经过指令调优的模型自然能取得更强的结果
    • 直接的监督微调仅提供有限的改进,甚至可能降低较强模型的性能,表明它并不能持续增强泛化能力
  • 为了稳健地验证论文提出方法的有效性,论文纳入了涉及更复杂或开放领域场景的下游任务,例如多轮对话和长文本推理
  • 这些具有挑战性的设置使论文能够评估 PaTaRM 在现实世界应用中的泛化性和鲁棒性
  • 此外,论文还在不同模型规模上进行了扩展实验,以系统性地考察 PaTaRM 在模型容量增加时的适应性和性能一致性
  • 论文将 PaTaRM 与 SOTA 方法进行了基准测试,包括 RLCF 框架下的 DPO 和由 Skywork 引导的 RL
    • 虽然 DPO 提供了更稳定的增益,但整体改进幅度有限
    • 使用 Skywork 的 RL 产生了适度的改进,特别是对于较小的模型,但其增益在不同基准测试和模型规模上的一致性较差
    • 相比之下,使用 PaTaRM 的强化学习始终能提供最佳结果,在所有模型和评估指标上均优于所有基线——包括最新的 SOTA 方法
  • 值得注意的是,PaTaRM 的改进在 InFoBench 的具有挑战性的子集上最为显著,突显了动态 Rubric 适应在复杂评估场景中的有效性和鲁棒性
  • 论文的实验设计覆盖了广泛的模型规模和初始化策略,为 PaTaRM 的通用性和可靠性提供了全面的验证。此外,论文的方法保持与标准 RLHF 流程的兼容性,确保了计算效率和实际适用性
  • 总体而言,这些结果证实,PaTaRM 为 RLHF 中的奖励建模提供了一个理论合理、实验验证充分且计算鲁棒的解决方案,与现有方法相比具有更优的性能和一致性

NLP——LLM对齐微调-RLSC

注:本文包含 AI 辅助创作

  • 参考链接:
    • 原始论文:(RLSC)Confidence Is All You Need: Few-Shot RL Fine-Tuning of Language Models, arXiv 20250611, AIRI, Skoltech
      • 来自俄罗斯的 AIRI(人工智能研究所)和Skoltech(斯科尔科沃科技学院)的工作

Paper Summary

  • 整体总结
    • 论文提出了基于自信的强化学习(Reinforcement Learning via Self-Confidence, RLSC)
      • RLSC 是一种轻量级微调方法,无需标签、偏好模型或人工设计的奖励
    • 与 TTRL 等依赖大规模多数投票的方法不同,RLSC 从数学上形式化了输出分布一致性的优化原理
    • 在 Qwen2.5-Math-7B 上的实验表明,RLSC 仅需每个问题 16 个样本和 10 步训练即可实现显著准确率提升,且无需外部监督
    • 核心贡献:作者给出了实验证明,高质量的后训练可以不依赖外部标签,而是通过精心设计的内部信号(投票)实现
  • 问题:现有的 RL 方法通常依赖于昂贵的人工标注或外部奖励模型
  • RLSC 方法利用模型自身的置信度作为奖励信号,从而无需标签、偏好模型或奖励工程
    • 在 Qwen2.5-Math-7B 模型上仅使用每个问题 16 个样本和 10 或 20 次训练步数进行实验,RLSC 在多个推理基准测试中显著提升了准确率:AIME2024 提升 +13.4%,MATH500 提升 +21.2%,Minerva Math 提升 +21.7%,Olympiadbench 提升 +20.8%,AMC23 提升 +9.7%
    • RLSC 为推理模型提供了一种简单、可扩展的 post-training 方法,仅需少量样本和无标注监督即可实现

Introduction and Discussion

  • ChatGPT (2023)、Qwen (2023) 和 DeepSeek (2024) 等 LLM 在广泛的任务中展现了卓越的推理能力
  • post-training optimization 对于进一步使模型行为与任务特定目标保持一致仍然至关重要
  • 与监督微调相比, RL 提供了更强的泛化能力,并被广泛用于提升大语言模型的性能
    • 例如,DPO (2023)、PPO (2017) 和 RLHF (2022) 等方法常用于使模型与人类偏好对齐,而 DeepSeek 的 GRPO (2025) 算法通过奖励驱动学习改进了推理能力
  • 尽管取得了这些进展 ,但现有的 RL 方法通常依赖于昂贵的人工标注数据或精心设计的奖励函数
    • RLHF 需要大量标注工作 (2022)
    • 测试时强化学习(Test-Time Reinforcement Learning, TTRL)(2025) 通过每个问题生成 64 个响应并进行多数投票来生成伪标签,这导致了高昂的计算开销
  • 为了应对这些局限性,论文提出了基于自信的强化学习(Reinforcement Learning via Self-Confidence, RLSC)
    • 这是一种利用模型对其输出的自信度作为奖励信号的新范式
    • 与依赖外部监督或大规模训练数据集的 RLHF 和 TTRL 不同,RLSC 直接从模型的响应中获取反馈,无需人工标签、外部模型或手动奖励设计
    • 作者相信,结合对输出生成的自信心分析,预训练大语言模型的内部知识可以在下游任务中实现质量提升
  • 作者在小规模模型 Qwen2.5-Math-7B (2024) 上验证了所提出的 RLSC 方法,并仅使用 AIME2024 训练集进行了 10 或 20 步的训练,每个问题仅生成 16 个样本
  • 尽管训练设置轻量,RLSC 在多个推理基准上取得了显著提升:
    • 在 AIME2024 上提升 +13.4%
    • 在 MATH500 上提升 +21.2%
    • 在 Minerva Math 上提升 +21.7%
    • 在 Olympiadbench 上提升 +20.8%
    • 在 AMC23 上提升 +9.7%
  • 这些结果表明,强大的预训练模型结合 RLSC 框架,可以在无需依赖特定辅助数据集、人类反馈或手工奖励函数的情况下,通过简短的额外训练阶段有效提升模型的置信度和泛化能力
  • 论文的主要贡献如下:
    • 1)我们提出了 RLSC ,这是一种新的强化学习框架,它无需人类标注、无需外部奖励模型,也无需手动设计奖励
    • 2)我们展示了 RLSC 在极少的训练数据和低计算成本下仍能实现强劲性能,使其适用于资源受限的场景
    • 3)我们在多个推理基准上验证了 RLSC 的效果,使用的推理提示和消息模板与 Qwen2.5-Math 版本一致,这凸显了 RLSC 作为一种实用的大语言模型训练方法的潜力,能够激发预训练模型的潜力

RLSC Method

From TTRL to Mode Sharpening(模式锐化)

  • 测试时强化学习(Test-Time Reinforcement Learning, TTRL)(2025) 通过为每个输入生成多个输出(通常为 64 个)并应用多数投票选择最频繁的完成结果来改进大语言模型 ,随后,该伪标签被用于微调模型
  • 尽管有效,但这种方法计算成本高昂,并且需要在答案和推理过程之间进行清晰的分离,这在实际应用中是一个非平凡(nontrivial)的预处理步骤
  • 受多数投票思想的启发,论文提出了以下问题:投票过程的底层原理是什么?
  • 直观上,多数投票选择了输出分布的众数,优化采样完成结果之间的一致性隐式地锐化了分布 :它增加了集中在最可能答案上的概率质量
    • 这一洞见促使论文用基于模式锐化的直接内部目标替代 TTRL 的外部伪标签
  • 设 \( p_{\theta}(y \mid x) \) 表示给定输入 \( x \) 时模型生成响应 \( y \) 的概率,参数化为 \( \theta \)。从该分布中独立采样的两个样本相同的概率为:
    $$
    F(p_{\theta}) = \mathbb{E}_{y_{1},y_{2}\sim p_{\theta}(y \mid x)}[\mathbb{I}(y_{1}=y_{2})] = \sum_{y} p_{\theta}(y \mid x)^{2}
    $$
  • 当分布坍缩为以单个最可能响应为中心的狄拉克函数(delta function)时,该表达式取得最大值(即模型自信时)
  • 因此,论文提出直接最大化以下自信目标:
    $$
    F(p_{\theta}) = \mathbb{E}_{y\sim p_{\theta}(y \mid x)}[p_{\theta}(y \mid x)]
    $$
    • 这一表述保留了 TTRL 的优势(促进稳定和可重复的答案),同时消除了对伪标签提取或多数投票的需求
    • 它构成了论文微调算法的基础

Self-Confidence Loss and Gradient

  • 为了优化上述自信目标:
    $$
    F(p_{\theta}) = \mathbb{E}_{y\sim p_{\theta}(y \mid x)}[p_{\theta}(y \mid x)]
    $$
  • 论文计算其关于模型参数 \( \theta \) 的梯度。应用对数技巧(log-trick),论文得到:
    $$
    \begin{align}
    \color{red}{\frac{1}{2}}\nabla_{\theta} F(p_{\theta}) &= \sum_{y} \nabla_{\theta} p_{\theta}(y \mid x) \cdot p_{\theta}(y \mid x) \\
    &= \mathbb{E}_{y\sim p_{\theta} }[\nabla_{\theta} p_{\theta}(y \mid x)] \\
    &= \mathbb{E}_{y\sim p_{\text{old} } } \left[p_{\text{old} }(y \mid x) \cdot \nabla_{\theta} \log p_{\theta}(y \mid x)\right]
    \end{align}
    $$
    • 问题:原始这里的推导有问题,省略了一个2倍(但基本不影响最终结论),所以我们在最前面增加了一个 \(\color{red}{\frac{1}{2}}\) 以确保数据推导准确
    • 其中,\( p_{\text{old} } \) 表示模型的冻结副本(即梯度不通过其传播),用于采样和加权。这导出了以下训练损失:
      $$
      \mathcal{L}_{1} = -\sum_{y} p_{\text{old} }(y \mid x) \cdot \log p_{\theta}(y \mid x)
      $$
  • 该损失提升了旧模型赋予更高置信度的响应的对数概率
    • 关键的是,它不需要外部奖励模型、标注数据,并且仅使用模型自身的置信分布作为反馈
  • 论文还将其推广到更广泛的可微函数 \( \mathcal{L}(p_{\text{old} }, p_{\theta}) \)。一个有效的变体通过添加常数 \( \alpha > 0 \) 平滑加权:
    $$
    \mathcal{L}_{2} = -\sum_{y} (p_{\text{old} }(y \mid x) + \alpha) \cdot \log p_{\theta}(y \mid x)
    $$
  • 这种加性平滑(additive smoothing)能够稳定优化过程,尤其在 \( p_{\text{old}} \) 分布呈现高度峰值或稀疏性时效果显著
  • 作者通过实验发现,即使是很小的 \( \alpha \) 值(例如 0.1)也能同时提升收敛性和泛化能力

Practical Training Setup

  • 论文将自信目标应用于微调 Qwen2.5-Math-7B 模型
  • 对于每个训练样本,论文使用基础模型生成一小批候选完成结果:
    • 具体而言,每个问题生成 16 个样本,采样温度固定
    • 这些样本被视为从 \( p_{\text{old} } \) 中独立同分布采样,并且在梯度计算期间保持当前模型分布不变
  • 对于每个样本,论文计算其在更新模型 \( p_{\theta} \) 下的对数概率。随后使用基本或平滑自信公式评估加权损失
    $$
    \mathcal{L}_{1} = -\sum_{y} p_{\text{old} }(y \mid x) \log p_{\theta}(y \mid x) \quad \text{or} \quad \mathcal{L}_{2} = -\sum_{y} (p_{\text{old} }(y \mid x) + \alpha) \log p_{\theta}(y \mid x) \tag{7}
    $$
  • 为了优化该损失,论文采用标准的自回归解码和训练流程:
    • 对于每个问题,使用 generate(temperature = 0.5, num of samples=16) 生成 16 个完成结果
    • 对于每个(提示 + 答案)对,进行分词并计算 Token 级别的对数概率
    • 应用助手掩码以隔离仅答案 Token
    • 评估掩码对数概率的总和以获得响应的对数似然
    • 评估损失并通过反向传播更新模型参数
  • 基本训练配置如下:
    • 论文在 AIME2024 数据集上仅训练 10 或 20 步 ,使用 8 块 NVIDIA A100 GPU(80GB);
      • 问题:只训练 10 或 20 步就够了吗?
    • 论文采用 AdamW 优化器,学习率为 \( 1 \times 10^{-5} \),并应用标准权重衰减;
    • 生成长度限制为 3072 个 Token
  • 这一极简设置完全无需辅助数据集、指令微调或偏好模型,实现了高效、零标签的规模化强化学习

Experiments

Results Analysis

  • Benchmarks
    • 论文在多个具有挑战性的基准数据集上评估了所提方法,包括数学推理任务(AIME24 (2024)、MATH500 (2021)、AMC23 (2024)、GSM8K (2021))、Minerva Math (2021)、Olympiadbench (2024)、MMLU Stem (2021) 以及问答基准 GPQADiamond (2024)
  • 准确率(Accuracy)
    • 定义为正确回答的样本数与总样本数的比值,如下面公式所示:
      $$ \text{Acc} = \frac{\#\text{Correct Answers} }{\#\text{Total Samples} } $$
    • Pass@1 分数计算公式为
      $$ \text{pass@1} = \frac{1}{k}\sum_{i=1}^{k}p_{i} $$
  • 为确保公平比较,论文使用相同的公开评估脚本重新评估了基线模型和论文的模型,所有实验设置保持一致,结果如表 2 所示
  • 论文选择评估模型准确率而非 Pass@1,因为在现实场景中,准确率才是关键指标
    • 结果显示,原始 Qwen 模型在直接评估中表现不佳,甚至经常无法正常工作
    • 而论文的方法在基线基础上通过强化优化实现了显著提升
  • 在所有核心基准测试中(AIME24、MATH500、Olympiadbench、Minerva Math 和 AMC23),模型均取得了显著进步,尤其在 7B 参数规模下优势更为明显(Minerva Math 提升 21.7%)

Emergent Behavior: Concise Reasoning without Prompting(无需提示,简洁推理)

  • 论文观察到,RLSC 微调使模型能够生成更简短、更自信的答案
  • 与传统微调方法不同(后者通常依赖“逐步思考”的文本提示),论文的模型学会了直接识别答案并避免冗余推理
    • 例如,在 AIME 的 Case 1 中,基线模型包含冗长的符号推导但仍失败 ,而经过 RLSC 调整的模型直接给出了正确且逻辑清晰的答案
    • 类似模式也出现在其他数学基准测试(如 MATH 和 AMC23)中
    • 尽管未量化回答长度的减少,但这一趋势在所有基准测试中一致存在
    • 这表明 RLSC 可以隐式增强中间推理的可信度

Qualitative Analysis

  • 论文从 MATH 和 AIME 基准测试中提取了推理结果并进行定性分析
  • 通过对比初始模型和 RLSC 微调模型的输出(见“模型输出对比”部分),论文发现微调模型在零样本设置下表现出更好的任务理解和推理能力
    • 例如,在 MATH500 基准测试中,初始模型对 Case 1 进行了基础但错误的推理,而微调模型则通过简洁的推理路径得出了准确结论
    • Case 2 中,论文的方法微调的模型表现出了很强的推理能力,不同于需要“step-by-step”推导的模型,该模型可以在简单的推理下获得准确的结论
      • 理解:如何理解这种情况?
  • 论文将在未来工作中进一步分析平滑项和样本数量的影响,但初步实验表明 RLSC 在广泛的超参数范围内保持稳定

Related Work

Reinforcement Learning in Reasoning Tasks

  • 近年来,RL 在提升 LLM 的推理能力中发挥了关键作用
  • DeepSeek-R1 (2025)、ChatGPT (2023)、QwQ (2025) 和 Qwen 等模型通过将复杂问题分解为中间步骤并进行深度思考,展现了强大的推理能力,这些能力通常通过奖励驱动机制获得和优化
  • 该领域的经典方法是RLHF (2022),它依赖人类标注或学习的偏好模型生成奖励信号以对齐模型行为
    • 然而,RLHF 需要大量人工标注,成本高昂
  • 为减少依赖,RLVR (2024) 提出了一种仅基于问答对 \((x, y^*)\) 的奖励范式,通过对比模型输出与参考答案计算可验证奖励
    • 尽管 RLVR 减少了对标注推理步骤的依赖,但仍需人工标注的问答对,限制了其扩展性

Test-Time Training

  • 近年来,测试时训练(Test-Time Training, TTT) (2025) 成为优化模型推理行为的 promising 方向
  • 代表性工作包括 SelfPlay Critic (SPC) (2025) 和 Absolute Zero Reasoner (AZR) (2025)
    • 它们采用受博弈论启发的对抗双模型框架:一个模型作为“生成器”制造具有挑战性的推理步骤,另一个作为“批评器”检测错误
    • 这些方法无需人类监督,但依赖外部工具(如 Python 执行器或代码验证器)提供反馈信号
  • 另一项 TTT 工作是测试时强化学习(Test-Time Reinforcement Learning, TTRL) (2025)
    • 它通过对每个问题采样多个候选响应并应用多数投票机制生成伪标签,进而计算模型更新的奖励
    • 尽管 TTRL 避免了显式人类监督,但需要大量样本(如每个问题 64 个),导致高昂计算开销

Summary and Motivation

  • 综上所述,尽管 RLHF (2022)、RLVR (2024)、SPC (2025)、AZR (2025) 和 TTRL (2025) 各自提出了不同的强化信号来源策略,但它们均依赖人类标注、外部模型或复杂的奖励工程

附录:Case Study

  • 作者在更有挑战的 AIME 数据集上展示了模型的性能,这些数据集包含了复杂的数学推理问题
1…818283…352
San Ye

San Ye

Stay Hungry. Stay Foolish.

704 posts
53 tags
© 2026 San Ye
Powered by Hexo
|
Theme — NexT.Gemini v5.1.4