← 返回 2026-01-27

一人适配万众:面向个性化大模型对齐的元奖励建模 One Adapts to Any: Meta Reward Modeling for Personalized LLM Alignment

Hongru Cai, Yongqi Li, Tiezheng Yu, Fengbin Zhu, Wenjie Wang, Fuli Feng, Wenjie Li 📅 2026-01-26 👍 7 2026-07-13 08:35
RLHF 个性化对齐 元学习 奖励建模 少样本学习

元学习驱动的个性化奖励建模,少样本快速适应任意用户偏好

前置知识

RLHF(基于人类反馈的强化学习)

RLHF 是当前大模型对齐的主流范式,核心流程包括三个阶段:首先用人类标注的偏好对(对同一 prompt 的两个回复进行排序)训练一个奖励模型,该模型为每个回复输出一个标量分数;然后用这个奖励模型的信号通过 PPO 等强化学习算法优化语言模型的生成策略。奖励模型的训练通常采用 Bradley-Terry 模型,将偏好建模为两个回复分数之差的 sigmoid 函数,损失函数为负对数似然形式。这个过程让模型学会判断什么样的回复更符合人类偏好。在本文的背景下,MRM 是在 RLHF 框架内的奖励建模阶段进行个性化改进。

本文提出的 Meta Reward Modeling 是在 RLHF 框架内的奖励建模阶段做个性化改进,理解 RLHF 的基本流程是理解本文定位的前提

MAML(模型无关元学习)

MAML 是一种经典的优化式元学习方法,核心思想是学习一组好的初始参数,使得面对新任务时只需少量梯度更新就能快速适应。其双层优化结构为:内循环从初始参数出发,在任务的支持集上做少量梯度更新得到适应后的参数;外循环在任务的查询集上评估适应后的参数表现,然后更新初始参数以改善跨任务的泛化能力。这种学习如何学习的范式让模型获得快速适应能力,已被广泛应用于图像分类、语言建模和强化学习等领域。

本文将每个用户视为一个独立任务,将 MAML 应用于个性化奖励建模,是全文方法论的核心基石

个性化奖励建模

与传统奖励模型假设统一的人类偏好标准不同,个性化奖励模型旨在捕捉每个用户的独特偏好。现有方法分为两大类:第一类是个性化输入方法,将用户画像、历史偏好或学习到的嵌入作为条件输入共享模型;第二类是个性化参数方法,为每个用户分配独立参数如 LoRA 适配器或独立模型。前者依赖输入上下文质量,稀疏反馈下表现粗糙;后者需要大量用户数据训练,面对未见用户难以泛化。

本文正是针对这两类方法的局限性提出了第三条路径即元学习范式,理解现有方法的痛点是理解本文贡献的关键

基函数组合表示

本文将每个用户的奖励模型表示为 K 个共享基奖励函数的加权组合,其中用户特有的权重向量决定了每个基函数的贡献。这种低秩分解的思想将个性化建模简化为学习少量权重参数,既保证了表达能力(通过组合不同基函数捕捉偏好多样性),又大幅降低了参数量(每个用户只需维护 K 个权重,实验中 K 为 2)。这种表示方式使得元学习的内循环只需优化轻量级的权重向量,而非整个奖励模型。

这种表示方式使得元学习的内循环只需优化轻量级的权重向量,而非整个奖励模型,是方法高效性的关键设计

研究动机

现有个性化奖励建模面临两个根本性挑战。第一个是个性化反馈的稀缺性:通用奖励模型通常依赖海量偏好数据来捕捉偏好模式,但对于单个用户来说,收集如此大规模的反馈数据是不现实的。简单地聚合其他用户的数据不仅无效,反而会因为偏好冲突而混淆模型。例如,PRISM 数据集在过滤掉对话数少于 6 条的用户后,每个用户平均仅有约 6 轮对话数据;Reddit TLDR 数据集虽然每个用户约有 3,750 个标注对,但用户总数仅 40 人。第二个挑战是对未见用户的适应:不可能预先收集所有潜在用户的数据,因此个性化奖励模型必须能够在测试时快速适应那些在训练阶段未出现过的用户,且只能使用少量示例。实验显示,现有方法在 seen 和 unseen 用户之间的性能差距虽小,但整体改进有限,大多数个性化方法相比非个性化的 BT 基线仅获得微弱提升,甚至在 PRISM 上出现个性化输入方法无法超越 BT 的情况。

本文的目标是本文的具体目标是构建一个能够在稀疏反馈条件下实现高效个性化适应的奖励建模框架。具体而言,框架需要同时满足三个要求:一是在每个用户仅有少量偏好数据时仍能有效建模个性化偏好;二是能够快速泛化到训练时未见过的新用户;三是对多样化的用户偏好保持鲁棒性,不因优化平均性能而忽略难以建模的特殊用户。最终目标是在 PRISM 和 Reddit TLDR 等数据集上超越现有个性化奖励建模方法的性能,同时保持参数效率和计算可扩展性。

与已有工作不同的是,本文的独特切入角度在于提出了从拟合数据学习用户偏好到学习偏好适应过程的范式转换。现有方法本质上都是静态拟合:个性化输入方法通过固定模型参数依赖输入上下文来实现个性化,在稀疏反馈下无法区分细粒度的用户意图;个性化参数方法为每个用户分配独立参数,但从未见用户的少量示例训练新参数容易过拟合。本文认为,这两种范式都不足以同时实现对未见用户的高效适应和在有限反馈下的细粒度个性化。因此,作者引入元学习视角,将每个用户的偏好建模视为一个独立的学习任务,通过学习一个高度可适应的模型初始化来实现学会学习的目标。这种视角转换让模型不是去记忆每个用户的具体偏好,而是去学习跨用户的偏好适应模式,从而获得从少量示例快速收敛到任何用户意图的能力。

核心方法

MRM 的整体思路可以这样理解:假设你要训练一个家教,目标是让它能快速适应任何学生的学习风格。传统方法是给家教一份详细的学生档案(个性化输入),或者给每个学生配备专门的家教(个性化参数)。但前者在学生信息很少时无法准确判断其风格,后者面对新学生需要重新培训。MRM 的做法不同:先训练家教掌握如何根据少量信息快速适应不同学生的通用能力。具体技术路线是将每个用户的奖励模型表示为 K 个共享基奖励函数的加权组合,用户间的差异仅体现在权重向量上。通过 MAML 风格的双层优化框架学习这些权重的初始化:内循环在每个用户的支持集上做少量梯度更新来模拟个性化适应过程,外循环在查询集上评估适应效果并更新初始化,使模型学会从共享模式出发快速收敛到任何用户的偏好。

MRM 的核心创新在于将个性化奖励建模重新定义为元学习问题,而非传统的数据拟合问题。这与已有方法的本质区别体现在三个方面。第一,建模目标不同:现有方法直接学习用户的偏好函数,而 MRM 学习的是偏好适应的过程,即如何从共享初始化出发,通过少量梯度更新就能捕获特定用户的偏好。第二,参数结构不同:个性化输入方法依赖固定模型加动态输入,MRM 是共享基函数加可学习初始化的权重。第三,优化视角不同:标准元学习均匀对待所有用户以最大化平均性能,但 MRM 引入了 Robust Personalization Objective(RPO),通过动态加权将更多注意力分配给难以建模的用户。具体而言,RPO 通过计算用户的查询损失来识别困难用户,设定阈值为损失的特定分位数,并使用 sigmoid 平滑函数对用户损失重新加权,确保优化不会被简单用户主导。

方法步骤详情

MRM 的方法分为训练和推理两个阶段。训练阶段采用双层优化:外循环采样一批用户任务;对每个用户,内循环从共享初始化出发,使用支持集上的 Bradley-Terry 损失进行梯度更新,先将用户权重设为共享初始化值,然后执行一步或多步梯度下降;在查询集上评估适应后的权重计算用户级别的查询损失;应用鲁棒个性化目标对查询损失重新加权,使用 sigmoid 平滑公式控制权重过渡的锐度;外循环更新共享初始化和基函数以最小化重加权后的查询损失之和。推理阶段:给定未见用户及其少量训练数据,从共享初始化出发执行一步或几步梯度更新得到个性化权重,然后构建个性化奖励函数用于评估或排序回复。这种设计使得每个用户只需维护少量权重参数,推理时的适应成本极低。

技术新颖性

MRM 的技术新颖性体现在几个层面。首先,这是首次将元学习范式系统性地引入个性化奖励建模领域,开辟了从学习偏好函数到学习偏好适应过程的新方向。虽然 MAML 已在推荐系统和对话系统中有应用,但将其用于奖励建模的权重初始化是原创的。其次,基函数组合表示与元学习的结合是精巧的设计,每个用户只需适应 K 个权重(实验中 K 为 2),大大降低了内循环的计算成本,使得元学习在奖励建模场景下变得可行。第三,RPO 是对标准 MAML 的重要改进,标准 MAML 均匀对待所有任务,而 RPO 通过动态识别和强调困难用户来提升鲁棒性,这种关注最差案例的思想对个性化场景尤为重要。第四,实验设计也体现了新颖性,通过将用户分为 seen 和 unseen 两组,严格测试了方法的泛化能力,这比仅在训练集上评估更有实际意义。

个性化奖励建模方法对比:(a) 个性化输入;(b) 个性化参数;(c) 元奖励建模
Figure 1: 个性化奖励建模方法对比:(a) 个性化输入;(b) 个性化参数;(c) 元奖励建模
Meta Reward Modeling 概览
Figure 2: Meta Reward Modeling 概览
阈值比例对 PRISM 性能的影响
Figure 4: 阈值比例对 PRISM 性能的影响
元批次大小和平滑参数 gamma 对性能的影响
Figure 5: 元批次大小和平滑参数 gamma 对性能的影响

实验结果

实验结果表明 MRM 在多个维度上一致性地超越了所有基线方法。在总体性能方面,MRM 基于 Skywork-Reward V2 在 PRISM 数据集上达到 65.3% 的准确率,比最佳基线 SynthesizeMe FT 的 64.4% 提升了 1.4%;在 Reddit TLDR(100 示例)上达到 69.6%,比 LoRe 的 68.3% 提升了 1.9%;在 Reddit TLDR(150 示例)上达到 69.7%,比 LoRe 的 68.6% 提升了 1.6%。这些提升在统计上显著(p 值小于 0.05)。关于鲁棒性,MRM 在最差 10%、20% 和 50% 用户上的表现均优于所有基线。例如在 PRISM 上,个性化输入方法和参数方法在最差用户上都出现严重性能下降,PRISM 上的结果甚至低于随机猜测的 33%,而 MRM 在最差用户上仍保持约 37-38% 的准确率。消融实验证明了各组件的贡献:移除元学习框架导致最大性能下降(PRISM 上总体准确率从 64.8% 降至 63.1%),移除 RPO 导致轻微下降至 64.5%,移除基函数组合也降低性能至 63.8%。在少样本适应实验中,MRM 在仅有 1-2 个示例时就展现出明显优势。参数效率方面,MRM 的可训练参数数量在所有用户规模下均为最小。

MRM 与基线方法的性能对比
Table 1: MRM 与基线方法的性能对比
MRM 关键组件的消融研究
Table 2: MRM 关键组件的消融研究
不同方法随用户数量增加的可训练参数数量
Figure 7: 不同方法随用户数量增加的可训练参数数量
查看结构化数据
任务指标本文基线提升
PRISM 个性化偏好预测 用户级准确率 (%) 65.3 ± 0.3 64.4 ± 1.0 (SynthesizeMe FT) +1.4%
Reddit TLDR (100 examples) 个性化偏好预测 用户级准确率 (%) 69.6 ± 0.3 68.3 ± 0.5 (LoRe) +1.9%
Reddit TLDR (150 examples) 个性化偏好预测 用户级准确率 (%) 69.7 ± 0.3 68.6 ± 1.0 (LoRe) +1.6%
PRISM 最差 10% 用户鲁棒性 用户级准确率 (%) 37.9 ± 0.8 约 35-37 (各基线大幅下降) MRM 保持稳定,基线低于随机
参数效率(用户规模扩展) 可训练参数数量 最小且增长缓慢 PAL/LoRe 线性增长,GPO/VPL 恒定 参数量最低

局限与改进

作者在讨论部分坦诚地指出了 MRM 的多项局限性。首先是静态用户偏好的假设:当前公式假设用户偏好在适应过程中保持不变,但实际场景中偏好可能随时间和情境变化,这需要持续学习或在线适应机制,而现有基准主要提供静态快照。其次是计算复杂度:MRM 依赖的双层元学习框架比单层方法引入更高的计算成本,虽然这一成本主要发生在元训练阶段且在推理时被摊销,但进一步提升训练效率仍是重要方向。第三是评估范围受限:实验仅在奖励建模层面使用偏好预测准确率评估,未延伸到下游策略优化或文本生成,该指标不能直接反映生成回复的实际质量或用户满意度。此外,作者还讨论了隐私、偏见、安全和奖励黑客等伦理问题。从独立分析来看,实验中的一些发现也值得关注:在 PRISM 上几乎所有个性化方法相比 BT 基线的改进都很有限,这暗示在极端数据稀缺场景下个性化本身的价值可能受限;MRM 的改进主要体现在鲁棒性而非平均性能上。

独立分析的弱点

基于对论文的深入分析,可以识别出几个可改进的弱点。第一,基函数数量的敏感性分析不够充分:虽然实验显示 K 等于 2 已足够,但缺乏理论解释说明为什么小 K 就有效,以及在什么条件下需要更大的 K。建议未来工作探索自适应基函数数量的学习方法,根据数据复杂度动态调整。第二,RPO 的阈值选择依赖超参数 rho(实验中固定为 0.5),缺乏自适应机制。当用户分布发生变化时,固定的 rho 可能不是最优的,可以考虑基于损失分布的自适应阈值。第三,内循环的梯度更新步数在实验中固定为 1,但不同用户的适应难度不同,可能需要不同的更新步数,可以探索自适应步数策略。第四,MRM 对基函数的初始化和结构选择未做详细讨论,这些基础函数的质量直接影响最终性能,可以探索使用预训练的多样化奖励模型作为基函数。第五,评估指标单一,仅使用准确率无法全面反映个性化质量。

未来方向

作者在结论部分提出了五个未来研究方向,这些方向都有很强的实际价值。第一,将 MRM 从奖励建模扩展到直接策略优化(例如结合 DPO),这将实现端到端的个性化生成并简化对齐流程。第二,建模演变的用户偏好,通过持续学习或在线适应机制处理动态环境中的偏好漂移,这需要构建包含时间维度的偏好数据集。第三,从被动依赖稀疏历史数据转向主动查询用户以消除歧义,通过设计智能的查询策略降低适应所需的样本复杂度。第四,超越显式的成对偏好标注,利用隐式或嘈杂的行为信号(如浏览时间、点击模式)提供更丰富的个性化监督信号。第五,将 MRM 扩展到更表达力的模型参数化(如全模型元学习),在保证可扩展性和训练稳定性的前提下进一步提升个性化容量。此外,基于本文成果还可以延伸出更多方向:探索 MRM 在多语言多文化场景下的跨域泛化能力;引入联邦学习框架保护用户隐私;结合因果推断技术区分用户偏好中的因果关系和相关性。

复现评估

本文的复现性评估较为积极。代码方面,作者提供了项目页面链接 https://github.com/ModalityDance/MRM,表明有开源意向。数据方面,使用的 PRISM 和 Reddit TLDR 都是公开可获取的数据集,PRISM 包含 1,287 个用户(过滤后),Reddit TLDR 包含 40 个用户,数据预处理步骤在论文中有详细说明。算力方面,实验在 NVIDIA RTX A5000 GPU 上运行,这是相对常见的学术研究硬件,复现门槛不高。模型方面,MRM 基于 Skywork-Reward 的嵌入构建,这些模型是开源的,且每个用户仅需维护 K 等于 2 个基函数的权重,参数量极低。超参数方面,论文详细报告了学习率(PRISM 为 1e-3,Reddit TLDR 为 5e-3)、内循环步数(1 步)、外循环批次大小(2 个用户)、RPO 参数(rho 为 0.5,gamma 为 0.5)等关键设置。数据划分方式(20 次随机划分,50/50 seen/unseen 分割)也有明确说明。总体而言,复现难度中等偏低,主要挑战在于 MAML 风格的双层优化实现需要仔细处理梯度传播。