arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02416cs.CLcs.AIcs.LG

Aligner:通过学习纠错实现高效对齐

Aligner: Efficient Alignment by Learning to Correct

Center for AI Safety and Governance, Institute for AI, Peking University(北京大学人工智能研究院、人工智能安全与治理中心)

Jiaming Ji, Boyuan Chen, Hantao Lou, Donghai Hong, Borong Zhang, Xuehai Pan, Juntao Dai, Tianyi Qiu, Yaodong Yang

AI总结:

针对大语言模型对齐方法复杂、难以快速迭代的问题,提出即插即用的模型无关对齐范式Aligner,用小模型学习校正残差,可提升多类LLM的3H性能,还能通过合成数据自举突破性能上限。

摘要:

随着大语言模型(LLM)的快速发展以及实际需求的不断演进,寻找一种高效且有效的对齐方法变得前所未有的重要。然而,当前对齐方法的复杂性与部署场景中快速迭代的需求之间存在矛盾,这就要求开发一种能够在这些约束条件下运行的模型无关对齐方法。本文提出了Aligner,这是一种新颖且简洁的对齐范式,它使用小型模型学习偏好答案与非偏好答案之间的校正残差。Aligner被设计为一个模型无关的即插即用模块,仅需一次性训练即可直接应用于各种开源模型和基于API的模型,适用于快速迭代场景。值得注意的是,Aligner可应用于任何强大的大规模上游模型。此外,它甚至可以将校正后的响应作为合成人类偏好数据,对上游模型进行迭代式自举提升,突破模型的性能上限。我们的实验表明,在3H维度(有用性、无害性、诚实性)上进行评估,将同一个Aligner模型部署在11种不同的LLM上均能实现性能提升。具体而言,Aligner-7B在测试的所有LLM中实现了有用性平均提升68.9%、无害性平均提升23.8%的效果,同时有效减少了幻觉。在Alpaca-Eval排行榜上,将Aligner-2B堆叠在GPT-4 Turbo上,使其LC胜率从55.0%提升至58.3%,超过了GPT-4 Omni 57.5%的胜率(社区报告数据)。

英文摘要:

With the rapid development of large language models (LLMs) and ever-evolving practical requirements, finding an efficient and effective alignment method has never been more critical. However, the tension between the complexity of current alignment methods and the need for rapid iteration in deployment scenarios necessitates the development of a model-agnostic alignment approach that can operate under these constraints. In this paper, we introduce Aligner, a novel and simple alignment paradigm that learns the correctional residuals between preferred and dispreferred answers using a small model. Designed as a model-agnostic, plug-and-play module, Aligner can be directly applied to various open-source and API-based models with only one-off training, making it suitable for rapid iteration. Notably, Aligner can be applied to any powerful, large-scale upstream models. Moreover, it can even iteratively bootstrap the upstream models using corrected responses as synthetic human preference data, breaking through the model's performance ceiling. Our experiments demonstrate performance improvements by deploying the same Aligner model across 11 different LLMs, evaluated on the 3H dimensions (helpfulness, harmlessness, and honesty). Specifically, Aligner-7B has achieved an average improvement of 68.9% in helpfulness and 23.8% in harmlessness across the tested LLMs while also effectively reducing hallucination. In the Alpaca-Eval leaderboard, stacking Aligner-2B on GPT-4 Turbo improved its LC Win Rate from 55.0% to 58.3%, surpassing GPT-4 Omni's 57.5% Win Rate (community report).

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

↑