Steerable Multi-Objective Group-Relative Preference Optimization with Jointly Trained Low-Rank Adapters
一句话概括
LambdaMix 用一次训练得到一族可调策略:为每个目标各训一个低秩适配器(low-rank adapter),推理时用偏好向量选择适配器混合并加权奖励。它在帮助性/无害性与数学解题/指令遵循任务上,超体积(hypervolume)高于所评估的条件化方法与策略合并基线。
速览
- 任务:学习以偏好为条件的策略族,支持推理时改变目标权衡,无需再优化。
- 方法:冻结基座模型,联合训练每个目标一个低秩适配器;偏好向量同时决定适配器混合与归一化奖励权重。
- 结果:在帮助性/无害性、数学解题/指令遵循两类任务、4 个模型上,超体积超过所有评估的条件化与策略合并基线。
- 效率与开源:训练完成数少于一组 GRPO 专家策略,存储的适配器参数更少;代码与数据开源情况「摘要未给出」。
问题背景
语言模型部署常需在多个目标间取不同权衡。用户偏好会变,权衡也应随之变化。
但标准后训练(post-training)方法通常只产出一个策略,对应固定的目标加权。想换权衡,往往要重训或另训专家模型。
已有条件化策略与策略合并(policy merging)路线各有局限。摘要未展开具体缺陷,只将其作为对比基线。
方法要点
LambdaMix 属参数高效方法,基于组相对在线强化学习(group-relative online RL)。它在一个冻结基座模型上,为每个目标联合训练一个低秩适配器。
推理时,一个偏好向量选出适配器混合,由该混合生成回答。训练时,同一偏好向量为各目标分别归一化的奖励加权,用于更新适配器。
关键设计是“混合选择”与“奖励加权”共用同一偏好向量。一次训练即可支持推理时切换偏好,不需额外优化。
作者还比较了两种变化来源。固定架构与初始化后,同时改变适配器混合与目标权重,比只改其一得到更宽分数范围与更高覆盖。
关键结论
在帮助性/无害性与数学解题/指令遵循两类任务上,4 个模型全部取得更高超体积,优于所有评估的条件化与策略合并基线。
效率上,LambdaMix 比一组 GRPO 专家策略用更少训练完成数,存储更少适配器参数。
在第三类任务(改写评论:保留措辞、减少个人属性线索)上,仅评估了两个最大模型。摘要未给出该任务的具体数值结论。
消融方向明确:适配器混合与目标权重需协同变化,单独变化任一者覆盖更窄。
读后思考
这篇工作的核心不是新奖励,而是把“偏好”做成一个连续可调的控制轴。适配器混合提供表达力,奖励加权提供训练信号,两者共享同一向量,逻辑自洽。
对工程部署有直接意义:一个基座加一组小适配器,可按场景调权衡,省去维护多个专家模型。代价是推理时要管理混合与偏好向量。
评测用超体积衡量“整条权衡曲线”的质量,比单点指标更贴合多目标设定。这也解释了为何强调覆盖范围。
局限与开放问题
摘要只给出相对结论,未给绝对分数、模型规模与训练成本细节。复现需等代码与数据,开源情况「摘要未给出」。
第三类任务只测两个最大模型,结论外推需谨慎。小模型上的表现「摘要未给出」。
偏好向量的粒度与可解释性未讨论。用户能否直观设定权衡、混合是否稳定,都是开放问题。
超体积依赖评测点选择与奖励归一化方式。不同归一化下结论是否稳健,摘要未给出。