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

Linear Alignment:一种无需调优和反馈即可对齐人类偏好的闭式解决方案

Linear Alignment: A Closed-form Solution for Aligning Human Preferences without Tuning and Feedback

Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);School of Computer Science, Fudan University(复旦大学计算机科学学院);ByteDance Inc(字节跳动公司)

Songyang Gao, Qiming Ge, Wei Shen, Shihan Dou, Junjie Ye, Xiao Wang, Rui Zheng, Yicheng Zou, Zhi Chen, Hang Yan, Qi Zhang, Dahua Lin

AI总结:

本文提出 Linear Alignment 算法,通过散度约束下策略优化的新参数化和闭式解,在单次推理中实现 LLM 与人类偏好对齐,无需标注或训练。

摘要:

基于语言模型(LLM)的 AI 助手的成功取决于从人类反馈中进行强化学习(RLHF),以理解并对齐用户意图。然而,PPO 等传统对齐算法受到复杂标注和训练要求的阻碍。这种依赖限制了 RLHF 的适用性,并阻碍了为多样化人类偏好定制的专业助手的发展。在本工作中,我们提出了 Linear Alignment,这是一种新颖的算法,可在单个推理步骤中使语言模型与人类偏好对齐,消除了对数据标注和模型训练的依赖。Linear Alignment 结合了一种用于散度约束下策略优化的新参数化方法,能够以闭式方式提取最优策略,并有助于直接估计对齐后的响应。在通用和个性化偏好数据集上的大量实验表明,Linear Alignment 在多种场景下显著提升了 LLM 对齐的性能和效率。我们的代码和数据集发布于 https://github.com/Wizardcoast/Linear_Alignment.git。

英文摘要:

The success of AI assistants based on Language Models (LLMs) hinges on Reinforcement Learning from Human Feedback (RLHF) to comprehend and align with user intentions. However, traditional alignment algorithms, such as PPO, are hampered by complex annotation and training requirements. This reliance limits the applicability of RLHF and hinders the development of professional assistants tailored to diverse human preferences. In this work, we introduce \textit{Linear Alignment}, a novel algorithm that aligns language models with human preferences in one single inference step, eliminating the reliance on data annotation and model training. Linear alignment incorporates a new parameterization for policy optimization under divergence constraints, which enables the extraction of optimal policy in a closed-form manner and facilitates the direct estimation of the aligned response. Extensive experiments on both general and personalized preference datasets demonstrate that linear alignment significantly enhances the performance and efficiency of LLM alignment across diverse scenarios. Our code and dataset is published on \url{https://github.com/Wizardcoast/Linear_Alignment.git}.

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

↑