P-GenRM:基于测试时用户缩放的个性化生成式奖励模型 P-GenRM: Personalized Generative Reward Model with Test-time User-based Scaling
首个个性化生成式奖励模型,用原型聚类和双粒度缩放实现精准用户对齐
前置知识
奖励模型(Reward Model)
奖励模型是 RLHF(基于人类反馈的强化学习)流程中的核心组件,负责对大语言模型的输出进行打分。传统奖励模型通常采用 Bradley-Terry 模型,将用户对两个响应的偏好建模为成对比较问题,输出一个标量分数。在个性化对齐场景中,奖励模型需要根据特定用户的偏好对响应进行评估,而不仅仅是学习全局的平均偏好。P-GenRM 正是在此基础上提出了个性化版本的生成式奖励模型。
P-GenRM 本质上是一种个性化奖励模型,理解传统奖励模型的工作原理和局限性是理解本文动机的前提。
生成式奖励模型(Generative Reward Model, GenRM)
与传统输出标量分数的判别式奖励模型不同,生成式奖励模型通过生成自然语言的评估链(evaluation chain)来判断响应质量。这种方法利用了大语言模型强大的生成能力,通过链式思维(chain-of-thought)推理产生可解释的评分依据。GenRM 具有天然的测试时可扩展性——可以通过多次采样生成多个评分方案再进行聚合,从而提升评分的准确性和鲁棒性。这种范式近年来受到越来越多关注,因其强泛化能力和对 LLM 生成能力的有效利用。
P-GenRM 基于 GenRM 范式构建,其核心创新之一就是利用 GenRM 的测试时可扩展性实现用户级缩放。
测试时缩放(Test-time Scaling)
测试时缩放是一种在推理阶段通过增加计算资源来提升模型性能的策略。在 GenRM 场景中,这意味着对同一个查询生成多个独立的评分方案(scoring schemes),然后通过聚合(如取平均)得到更稳定的最终分数。这种方法的优势在于不需要重新训练模型,只需增加采样次数即可获得性能提升,且计算成本与采样数近似线性增长。P-GenRM 在此基础上创新性地提出了双粒度的用户级缩放策略。
本文的 Test-time User-based Scaling 是核心创新点,利用测试时缩放的范式同时在个体和原型两个粒度上聚合偏好信号。
协同过滤(Collaborative Filtering)
协同过滤是推荐系统中的经典方法,核心思想是「相似的用户对相似的物品有相似的偏好」。基于用户的协同过滤通过找到与目标用户历史行为相似的其他用户,利用这些相似用户的评分来预测目标用户对未见过物品的偏好。这种方法能有效缓解冷启动问题——即使新用户只有少量历史数据,也可以通过相似用户的偏好进行近似预测。经典算法包括 Goldberg 等人(1992)和 Sarwar 等人(2001)的工作。
P-GenRM 的原型级缩放机制直接借鉴了协同过滤的思想,通过用户原型聚类和相似用户偏好迁移来增强对新用户的泛化能力。
GRPO(Group Relative Policy Optimization)
GRPO 是一种强化学习算法,与 PPO 类似但不依赖价值函数网络。它通过对同一查询采样多个响应,计算组内相对优势(relative advantage)来更新策略。具体来说,GRPO 计算每个采样响应相对于组内平均表现的优势值,然后用这个优势值来指导策略梯度更新。这种方法更稳定、更高效,特别适合生成式奖励模型的训练场景,由 Shao 等人(2024)提出。
P-GenRM 的第二阶段训练(Criteria-based Reasoning Enhancement)使用 GRPO 算法来优化评估链的生成质量。
课程学习(Curriculum Learning)
课程学习模仿人类从易到难的学习过程,先用简单样本训练模型,然后逐渐引入更困难的样本。在 P-GenRM 的语境中,硬负样本(hard negatives)是指那些与正样本非常相似、难以区分的负样本。通过逐步增加硬负样本的比例,模型可以学到更精细的区分能力,提升在高度主观任务上的鲁棒性。该方法最早由 Bengio 等人(2009)提出。
P-GenRM 的第三阶段训练采用硬负样本感知的课程学习来增强模型处理困难案例的能力。
用户画像(User Persona)
用户画像是对用户特征的结构化描述,包括用户的身份、偏好、行为模式等。在个性化对齐中,用户画像可以作为先验知识帮助模型理解用户的隐式偏好信号。P-GenRM 中的用户画像是场景特定的(scenario-specific),即同一个用户在不同场景下可能有不同的画像,例如开车时偏好简洁回答、闲聊时偏好丰富表达。这种动态画像设计是 P-GenRM 区别于 SynthesizeMe 等静态方法的关键。
P-GenRM 的核心机制之一是根据用户的偏好信号动态推断场景特定的用户画像,而非使用静态的用户画像。
研究动机
个性化对齐是大语言模型对齐领域的重要研究方向,其目标是让模型的输出适应不同用户的独特偏好。在开放式任务(如对话)中,评估标准高度依赖主观判断,这使得获取准确的用户特定奖励信号变得极具挑战性。现有个性化奖励模型面临两个根本性缺陷。第一,偏好建模的静态化问题。大多数方法将用户多样且动态变化的偏好压缩为少量固定的评估维度。例如,Prism 数据集预定义了 {Style, Values, Fluency, Factuality, Safety, Diversity, Helpfulness} 等固定维度,但实际上用户的偏好远比这丰富,可能包括 {Philosophical Engagement, Openness, Structure, Depth, Nuance, Sensitivity, Breadth of Ideas} 等维度。更关键的是,即使是同一个用户,其偏好也会随场景变化——开车时偏好简洁回答,闲聊时则偏好表达丰富的回答,而静态建模无法捕捉这种场景依赖性。第二,对新用户的泛化能力不足。当新用户只有少量反馈数据时(冷启动场景),现有模型难以有效适应,限制了实际应用效果。现有的 GPO、VPL、PAL 等方法在这两个问题上均存在明显不足。
本文的目标是本文的目标是构建首个个性化生成式奖励模型(Personalized Generative Reward Model),能够将用户多样化的偏好信号(包括隐式交互历史和显式偏好标准)转化为结构化的评估链(evaluation chain),动态推断场景特定的用户画像和评分规则,并通过测试时用户级缩放(Test-time User-based Scaling)机制进一步提升评分准确性和对新用户的泛化能力。具体而言,P-GenRM 需要实现:在个体层面,通过多次采样探索用户偏好的多种假设,获得更丰富、更鲁棒的评分方案;在原型层面,利用相似用户的偏好信息优化推断结果,并为历史数据稀疏的新用户提供有效的偏好近似。最终目标是在广泛使用的个性化奖励模型基准测试上达到最先进水平。
与已有工作不同的是,本文的独特切入角度体现在三个方面。首先,在方法论层面,P-GenRM 将个性化对齐问题从传统的「固定维度偏好匹配」转化为「结构化评估链生成」,利用大语言模型的生成能力动态推断用户画像和评分标准,而非预定义少量评估维度。这种转化使得模型能够捕捉到远超固定维度的丰富偏好信息。其次,在技术路线上,本文借鉴推荐系统中协同过滤的思想(Goldberg 等人 1992 年和 Sarwar 等人 2001 年的经典工作),首次将用户原型聚类和双粒度缩放机制引入生成式奖励模型,在统一框架内同时解决偏好推断噪声和新用户泛化两个挑战。最后,在训练策略上,本文设计了三阶段训练框架——SFT 基础能力注入、RL 评估链质量提升、课程学习鲁棒性增强——系统性地构建个性化评分能力,而非单一阶段的端到端训练。这种渐进式的训练策略使得每个阶段专注于解决特定的能力短板。
核心方法
P-GenRM 的方法可以分为两个相互关联的部分:三阶段训练框架和测试时用户级缩放机制。直觉上,个性化评分需要先理解用户是谁、在意什么,然后根据这些理解制定评分标准,最后对候选响应进行打分。三阶段训练就是系统性地构建这种能力:第一阶段(SFT)教会模型将用户的偏好信号转化为结构化评估链,包括场景特定的用户画像和评分标准;第二阶段(RL)通过强化学习提升评估链的质量,特别是在缺乏显式偏好信息时的推断能力;第三阶段(课程学习)通过逐步引入硬负样本增强模型处理困难案例的鲁棒性。训练完成后,测试时缩放机制利用 GenRM 天然的可扩展性,在个体层面通过多次采样探索用户偏好的多种可能,在原型层面借鉴协同过滤思想融合相似用户的偏好信息,从而同时提升评分准确性和对新用户的泛化能力。整体技术路线从训练到推理形成了完整的闭环,每一环节都针对个性化对齐的特定挑战设计了相应的解决方案。
P-GenRM 的核心创新在于将 GenRM 的测试时可扩展性与用户原型聚类相结合,实现双粒度的偏好信号聚合。与已有方法的本质区别在于:第一,SynthesizeMe 等方法将用户画像视为静态先验,在整个交互过程中保持不变,而 P-GenRM 动态推断场景特定的用户画像和评分标准,能够捕捉同一用户在不同场景下的偏好差异。例如,一个用户在工作中偏好专业正式的回答风格,在闲聊中偏好轻松幽默的风格,P-GenRM 能够根据当前场景动态调整画像。第二,现有个性化奖励模型(如 GPO、VPL、PAL)通常为每个用户训练独立的参数或使用固定的偏好维度组合,无法有效利用用户间的相似性信息。P-GenRM 通过用户原型聚类,将相似用户组织在一起,在测试时利用同一原型内其他用户的偏好信息来优化对目标用户的评分,这直接借鉴了推荐系统中协同过滤的核心思想。第三,在生成式奖励模型的框架下,P-GenRM 首次将显式评估链生成与测试时用户级缩放结合,实现了可解释性和准确性的统一。
方法步骤详情
P-GenRM 的完整技术流程包含以下步骤。第一步,构建结构化评估链(SEC)数据集。给定用户的当前查询 $q_t$ 和偏好信号 $\{H_t^{(u)}, E^{(u)}\}$(其中 $H_t^{(u)}$ 是历史交互,$E^{(u)}$ 是显式偏好标准),使用指令型 LLM(原文使用 OpenAI o3)推断用户画像 $P_t^{(u)}$,然后基于画像为当前场景生成评分标准和权重,最后对每个候选响应 $y_i^t$ 进行评估得到分数 $s_i^t$。第二步,SFT 阶段使用经过拒绝采样过滤的高质量 SEC 数据对 P-GenRM 进行监督微调,使其获得基本的个性化评分能力。第三步,RL 阶段使用 GRPO 算法进行训练,奖励函数包含过程奖励 $PR_t$(由 LLM-as-a-Judge 评估评估链对用户偏好的覆盖程度,得分 0-1)和结果奖励 $OR_t = \mathbf{1}\{s_c^t > s_r^t\}$(判断排序正确性),总奖励 $R_t = \alpha \cdot PR_t + \beta \cdot OR_t$,原文中 $\alpha=0.5$、$\beta=1.0$。第四步,课程学习阶段逐步增加硬负样本比例,同时关闭过程奖励以扩大探索空间。第五步,离线原型初始化:使用 Qwen3-Embedding-0.6B 计算用户画像嵌入,通过 K-means 聚类得到 $k=50$ 个用户原型,再通过历史感知的注意力原型精炼机制优化原型表示。精炼过程使用原型增强注意力机制 $\alpha_\tau = \text{softmax}_\tau(\frac{o_\tau^T q_t}{\sqrt{d}} + \rho \frac{o_\tau^T a_j}{\sqrt{d}})$ 来加权历史记录,并通过成对损失和正则化更新原型。第六步,测试时双粒度缩放:个体级缩放对同一查询生成 $m$ 个评分方案取平均;原型级缩放将用户分配到最近原型,选择 $n$ 个最相似用户的偏好信号进行融合。最终分数为 $s_i^t = \frac{1}{m}\sum_{x=1}^{m} \text{Extract}(S_{i,x}^t) + \frac{1}{n}\sum_{w=1}^{n} \text{Extract}(S_i^t(u_w))$。
技术新颖性
P-GenRM 的技术新颖性体现在多个层面。首先,它是首个将生成式奖励模型应用于个性化对齐场景的工作,将个性化评分从传统的标量回归问题转化为结构化文本生成问题,天然获得了可解释性和测试时可扩展性。其次,三阶段训练框架的设计具有独到之处:SFT 阶段通过拒绝采样保证基础能力,RL 阶段创新性地将过程奖励(评估链质量)与结果奖励(排序正确性)结合,并通过权重系数 $\alpha$ 和 $\beta$ 控制模型对显式偏好标准的敏感度,防止过拟合到特定偏好维度;课程学习阶段关闭过程奖励以扩大探索空间,这一设计细节体现了对不同训练阶段目标的深入理解。再次,用户原型聚类机制将推荐系统的协同过滤思想引入生成式奖励模型,通过 PCA 分析确定原型数量(保留方差比),并通过历史感知的注意力机制和正则化更新策略(包括簇中心正则 $\lambda_{\text{cent}} \|a_j - \mu_j\|_2^2$ 和时序平滑正则 $\lambda_{\text{tr}} \|a_j - p_j\|_2^2$)确保原型的有效性和稳定性。最后,双粒度缩放的统一框架在个体级探索用户偏样的多种假设、在原型级利用相似用户的集体智慧,两个层次的协同效应带来了超越单独使用任一缩放策略的性能提升。
实验结果
P-GenRM 在多个基准测试上取得了最先进的性能,并验证了其各组件的有效性。在 PersonalRewardBench 上,P-GenRM-8B 在 Chatbot Arena-Personalized 和 PRISM-Personalized 上分别达到 72.68% 和 65.32%,相比此前 SOTA(SynthesizeMe + FT RM 的 69.78% 和 62.84%)平均提升 2.77%。P-GenRM-70B(使用 LoRA 训练)进一步提升至 73.42% 和 66.21%。值得注意的是,P-GenRM-8B 甚至超越了此前最好的 70B 模型(72.68% vs 72.05% 在 Chatbot Arena 上),平均超越 1.04%。在测试时缩放方面,最佳设置(Ind-16, Pro-8)在 Chatbot Arena 和 PRISM 上分别达到 75.92% 和 68.06%,相比 P-GenRM 本身平均提升 2.99%,且仅需 24 次缩放步骤(16+8),少于单独使用 Ind-32 的 32 次步骤却获得了更好的性能(75.92% vs 75.59%),证明了原型级缩放的有效性。在与 OpenAI o3 的对比中,P-GenRM-8B 大幅超越 o3(72.68% vs 64.47% 在 Chatbot Arena 上),甚至 o3 + PSI 也仅为 69.14%。消融实验表明,移除课程学习、过程奖励、结果奖励或整个 RL 阶段分别导致 1.61%、2.46%、3.63% 和 5.92% 的性能下降,验证了每个组件的必要性。在 LaMP-QA 的 OOD 泛化测试中,P-GenRM-8B(Ind-8, Pro-4)的平均 Spearman 相关系数达到 0.638,超越了更大的 Qwen3-235B-A22B(0.619)和 LLaMA3.1-70B(0.600),展示了在稀疏反馈场景下的强泛化能力。此外,P-GenRM 在宏准确率(macro accuracy)指标上也达到 65.21%,保证了少数群体偏好不被忽视。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Chatbot Arena-Personalized (8B) | Accuracy (%) | P-GenRM: 72.68 ± 1.85% | FT RM + SynthesizeMe: 69.78 ± 1.98% | +2.90%(平均提升 2.77% over SOTA) |
| PRISM-Personalized (8B) | Accuracy (%) | P-GenRM: 65.32 ± 0.56% | FT RM + SynthesizeMe: 62.84 ± 0.85% | +2.48%(平均提升 2.77% over SOTA) |
| Chatbot Arena-Personalized (70B) | Accuracy (%) | P-GenRM: 73.42 ± 1.74% | FT RM + SynthesizeMe: 72.05 ± 2.24% | +1.37%(平均提升 1.99% over SOTA) |
| PRISM-Personalized (70B) | Accuracy (%) | P-GenRM: 66.21 ± 0.76% | FT RM + SynthesizeMe: 63.74 ± 0.66% | +2.47%(平均提升 1.99% over SOTA) |
| Chatbot Arena (Ind-16, Pro-8) | Accuracy (%) | 75.92 ± 1.70% | P-GenRM base: 72.68 ± 1.85% | +3.24%(测试时缩放平均提升 2.99%) |
| PRISM (Ind-16, Pro-8) | Accuracy (%) | 68.06 ± 0.69% | P-GenRM base: 65.32 ± 0.56% | +2.74%(测试时缩放平均提升 2.99%) |
| LaMP-QA OOD 泛化 | Spearman correlation (avg) | P-GenRM-8B (Ind-8, Pro-4): 0.638 | Qwen3-235B-A22B: 0.619 | +0.019(超越 235B 参数模型) |
局限与改进
尽管 P-GenRM 取得了显著的性能提升,但仍存在以下局限性。首先,测试时缩放的计算成本问题:虽然 Ind-16 + Pro-8 的 24 次缩放步骤少于 Ind-32 的 32 次,但相比不需要缩放的基线模型,推理成本仍然显著增加。作者在附录中报告了推理时间的比较,但这种成本在大规模部署场景下可能成为瓶颈。其次,原型数量的选择基于 PCA 方差分析确定为 50,但这一选择可能对不同数据分布不够鲁棒,且当用户群体发生显著变化时需要重新计算原型。第三,三阶段训练流程较为复杂,需要构建高质量的 SEC 数据集(依赖 OpenAI o3 等强大指令模型)、进行 GRPO 强化学习训练、以及课程学习的渐进式硬负样本引入,整体训练成本远高于传统的 Bradley-Terry 奖励模型。第四,论文仅在英文数据集上进行了评估(Chatbot Arena、PRISM、LaMP-QA),未验证方法在其他语言或文化背景下的适用性。第五,偏好信号的推断仍然存在噪声问题,虽然原型级缩放可以在一定程度上缓解,但在极端情况下(如用户偏好高度独特或矛盾),模型可能产生不准确的评估链。
独立分析的弱点
P-GenRM 存在几个值得深入分析的弱点。第一,训练数据构建对强模型的依赖:SEC 数据集的构建依赖 OpenAI o3 等强大的指令模型来生成高质量的评估链,这引入了外部依赖和潜在的级联误差——如果教师模型的偏好推断存在偏差,这些偏差会被传递到 P-GenRM 中。改进方向可以是探索自举(bootstrapping)策略或使用多个教师模型的集成来构建更稳健的训练数据。第二,原型更新的离线性质:当前原型的初始化和精炼是离线完成的,无法实时适应用户偏好的变化。未来可以探索在线原型更新机制,使原型能够随着用户交互的积累而持续演化。第三,原型级缩放的用户相似度计算基于嵌入空间的最近邻搜索,但嵌入质量受限于 Qwen3-Embedding-0.6B 模型的能力。对于偏好高度细粒度的用户,简单的最近邻搜索可能无法找到真正相似的用户。改进方向包括引入更精细的相似度度量或层次化的原型结构。第四,过程奖励和结果奖励的权重 $\alpha=0.5$、$\beta=1.0$ 是通过附录中的网格搜索确定的,但这些超参数可能对不同数据集不够稳定,缺乏自适应调整机制。
未来方向
基于 P-GenRM 的研究成果,未来研究可以从多个方向延伸。第一,作者提出的方向包括将 P-GenRM 应用于策略模型的训练(附录 A.12 已初步验证了 DPO 和 GRPO 设置下的有效性,P-GenRM 将 8B 策略模型提升到超越 70B 模型的水平),进一步探索端到端的个性化 RLHF 流程。第二,可以将用户原型聚类机制扩展为层次化结构,支持从粗粒度(如文化背景、年龄段)到细粒度(如具体偏好维度)的多层级偏好建模。第三,探索动态原型更新策略,使模型能够在部署过程中持续学习用户偏好的变化,实现真正的在线个性化。第四,将 P-GenRM 的评估链生成能力应用于其他主观评估任务,如创意写作评估、个性化内容推荐、教育辅导等场景。第五,探索更高效的测试时缩放策略,如自适应缩放(根据查询的难度动态决定采样数量)或基于检索的缩放(预先缓存相似查询的评分方案)。第六,将方法扩展到多模态场景,结合用户的视觉、听觉等多模态偏好信号进行个性化评估。
复现评估
在复现方面,P-GenRM 的代码已在 GitHub 上开源(https://github.com/Tongyi-ConvAI/Qwen-Character/tree/main/Character-GenRM),这为复现提供了良好的基础。训练数据方面,PersonalRewardBench 基于公开的 Chatbot Arena 和 PRISM 数据集构建,LaMP-QA 也是公开可用的。然而,完整的复现面临几个挑战。首先,三阶段训练流程需要大量计算资源:8B 模型使用 8 张 GPU,70B 模型使用 32 张 GPU 进行 LoRA 训练,这对大多数研究团队来说成本较高。其次,SEC 数据集的构建依赖 OpenAI o3 API,需要相应的 API 访问权限和费用,且 o3 是闭源模型。第三,原型初始化使用 Qwen3-Embedding-0.6B,需要额外的嵌入计算。第四,论文报告了 5 次独立运行的标准误差,说明结果具有一定的随机性,完整复现需要多次实验。总体而言,虽然代码开源降低了复现门槛,但计算资源和外部 API 依赖仍是主要障碍。
论文图表