arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.09538cs.CL
收藏

UniARM: 向多目标测试时间对齐的统一自回归奖励模型迈进

UniARM: Towards a Unified Autoregressive Reward Model for Multi-Objective Test-Time Alignment

School of Computer, Beihang University(北京航空航天大学计算机学院);Institute of Artificial Intelligence (TeleAI), China Telecom(中国电信人工智能研究院)

Hongyan Xie, Yikun Ban, Ruiyu Fang, Zixuan Huang, Deqing Wang, Jianxin Li, Yitong Yao, Chao Wang, Shuangyong Song

AI总结:

UniARM通过统一自回归奖励模型实现多目标测试时间对齐,通过共享特征和偏好调节模块减少特征纠缠,提升对偏好权衡的控制能力。

摘要:

多目标对齐旨在将大语言模型(LLM)的响应与多个人类偏好目标对齐。在现有方法中,通过自回归奖励模型(ARMs)引导冻结LLM的生成以实现多目标测试时间对齐是一种低成本的解决方案。然而,这些方法通常依赖于每个偏好目标独立的参数,要么通过在偏好维度上独立训练ARMs,这忽视了偏好特征之间的交互,要么通过为每个偏好训练一个单独的特征提取模块来训练单个ARM,这可能导致特征纠缠。这两种策略都可能导致生成输出与用户偏好之间的不匹配。为了解决这一限制,我们提出了Preference-Modulated & Shared Low-Rank Adaptation(MoSLoRA)用于ARM训练,首先通过一个无偏见的模块提取共享特征,然后通过一个受混合偏好向量条件的偏好调节模块对共享特征应用仿射变换。这种设计缓解了特征纠缠,并在推理过程中实现了对偏好权衡的精确控制。在此基础上,我们引入了统一的自回归奖励模型(UniARM),这是一种新的多目标测试时间对齐框架。UniARM在一个参数空间内联合建模所有偏好维度,消除了对每个偏好目标独立参数的需要。在更大规模的LLM上进行训练,增强了其实用性。

英文摘要:

Multi-objective alignment aims to align LLM responses with multiple human preference objectives. Among existing methods, guiding the generation of frozen LLMs through autoregressive reward models (ARMs) to accomplish multi-objective test-time alignment is a low-cost solution. However, these methods typically rely on independent parameters for each preference objective, either by training ARMs independently across preference dimensions, which neglects interactions among preference features, or by training a single ARM with separate feature extraction modules for each preference, which can cause feature entanglement. Both strategies can result in misalignment between generated outputs and user preferences. To address this limitation, we propose Preference-Modulated \& Shared Low-Rank Adaptation (MoSLoRA) for ARM training, which first extracts shared features via a preference-agnostic module and then applies affine transformations to shared features via a preference modulation module conditioned on mixed preference vectors. This design mitigates feature entanglement and enables precise control over preference trade-offs during inference. Building on this, we introduce the Unified Autoregressive Reward Model (UniARM), a novel framework for multi-objective test-time alignment. UniARM jointly models all preference dimensions in a single parameter space, eliminating the need for independent parameters for each preference objective. es on larger-scale LLMs, enhancing its practical usability.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑