arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.08005cs.CLcs.LG
收藏

用于LLM行为对齐的基于合成数据的改进直接偏好优化

Refined Direct Preference Optimization with Synthetic Data for Behavioral Alignment of LLMs

Komorebi AI

Víctor Gallego

AI总结:

本文提出rDPO方法,通过教师LLM自我批评生成合成数据,结合外部奖励模型的广义DPO损失函数蒸馏至学生LLM,无需人工标注即可有效提升LLM在安全性、鲁棒性和减少谄媚性等行为对齐任务上的表现。

摘要:

本文介绍了改进的直接偏好优化(rDPO),一种无需人工标注数据即可提升大型语言模型(LLMs)行为对齐的方法。该方法通过教师LLM的自我批评提示创建合成数据,随后利用广义DPO损失函数蒸馏至学生LLM。该损失函数引入额外的外部奖励模型以提升合成数据质量,使rDPO对合成数据集中的潜在噪声具有鲁棒性。实验表明,rDPO在多种行为对齐任务中有效,如提升安全性、增强对角色扮演的鲁棒性以及减少谄媚性。代码发布于https://github.com/vicgalle/refined-dpo。

英文摘要:

In this paper, we introduce \emph{refined Direct Preference Optimization} (rDPO), a method for improving the behavioral alignment of Large Language Models (LLMs) without the need for human-annotated data. The method involves creating synthetic data using self-critique prompting by a teacher LLM and then utilising a generalized DPO loss function to distil to a student LLM. The loss function incorporates an additional external reward model to improve the quality of synthetic data, making rDPO robust to potential noise in the synthetic dataset. rDPO is shown to be effective in a diverse set of behavioural alignment tasks, such as improved safety, robustness against role-playing, and reduced sycophancy. Code to be released at https://github.com/vicgalle/refined-dpo.

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

↑