arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04792cs.AIcs.CLcs.HC
收藏

基于在线AI反馈的直接语言模型对齐

Direct Language Model Alignment from Online AI Feedback

University of Edinburgh(爱丁堡大学);Google DeepMind(谷歌DeepMind);University of Basel(巴塞尔大学)

Shangmin Guo, Biao Zhang, Tianlin Liu, Tianqi Liu, Misha Khalman, Felipe Llinares, Alexandre Rame, Thomas Mesnard, Yao Zhao, Bilal Piot, Johan Ferret, Mathieu Blondel

AI总结:

针对直接偏好对齐中离线、离策略反馈的问题,本文提出OAIF,在每轮训练中由LLM标注当前模型生成的回复对,实现可控在线反馈,并在多任务人工评估中优于离线DAP与RLHF。

摘要:

直接偏好对齐(DAP)方法(如DPO)近期已成为人类反馈强化学习(RLHF)的高效替代方案,这类方法不需要单独的奖励模型。然而,DAP方法使用的偏好数据集通常在训练前收集且从不更新,因此反馈完全是离线的。此外,这些数据集中的回复往往从与正在对齐的语言模型不同的模型中采样,而由于模型在训练过程中不断演进,对齐阶段不可避免地是离策略的。本研究假设在线反馈是关键,并可改进DAP方法。我们的方法在线AI反馈(OAIF)使用LLM作为标注器:在每次训练迭代中,从当前模型采样两个回复,并提示LLM标注器选择更偏好的回复,从而提供在线反馈。尽管方法简单,我们通过在多个任务上的人工评估证明,OAIF优于离线DAP和RLHF方法。我们进一步表明,通过向LLM标注器提供指令提示,OAIF中利用的反馈易于控制。

英文摘要:

Direct alignment from preferences (DAP) methods, such as DPO, have recently emerged as efficient alternatives to reinforcement learning from human feedback (RLHF), that do not require a separate reward model. However, the preference datasets used in DAP methods are usually collected ahead of training and never updated, thus the feedback is purely offline. Moreover, responses in these datasets are often sampled from a language model distinct from the one being aligned, and since the model evolves over training, the alignment phase is inevitably off-policy. In this study, we posit that online feedback is key and improves DAP methods. Our method, online AI feedback (OAIF), uses an LLM as annotator: on each training iteration, we sample two responses from the current model and prompt the LLM annotator to choose which one is preferred, thus providing online feedback. Despite its simplicity, we demonstrate via human evaluation in several tasks that OAIF outperforms both offline DAP and RLHF methods. We further show that the feedback leveraged in OAIF is easily controllable, via instruction prompts to the LLM annotator.

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

↑