写点什么

DPO 直接偏好优化:跳过复杂的对抗学习,语言模型本来就会奖励算法

作者:Zilliz
  • 2023-07-13
    北京
  • 本文字数:1248 字

    阅读完需:约 4 分钟

大语言模型(LLM)在掀起“全民”热潮的同时,也将人类反馈强化学习(RLHF)和奖励模型(Reward Model)带进了大家的视线。

大语言模型目前的调优策略一般是在大规模的无监督训练之后,通过人类偏好的策略将期望的行为融入到语言模型中。虽然最直接的偏好学习方法是基于高质量的示范进行监督微调,但最成功的方法类别是通过人类(或 AI)反馈进行强化学习,即 RLHF。

RLHF 方法将奖励模型适应到人类偏好的数据集上,然后使用强化学习优化语言模型策略,以产生被分配高奖励的回应,同时不过度偏离原始模型。RLHF 和奖励模型的迭代训练,语言模型可以不断改进自己的文本生成能力。然而,RLHF 的流程远比监督学习复杂得多,其中涉及到训练多个语言模型。同时,该方法还需要相当大的计算成本,用来在训练过程中从语言模型策略中抽样。此外,RLHF 主要受到了数据的限制。它需要大量的人工反馈和偏好数据,不仅会消耗大量的人力资源,还可能会引入人为的偏见。


|RLHF vs DPO optimizing for human preferences

为了寻求更简单有效的大语言模型优化策略,斯坦福大学的团队提出了一种新的算法 Direct Preference Optimization(DPO)。该方法可以通过直接优化语言模型来实现对其行为的精确控制,而无需使用复杂的强化学习。DPO 将奖励函数和最优策略之间的映射联系起来,从而把约束奖励最大化问题转化为一个单阶段的策略训练问题。这种算法不仅不用拟合奖励模型,还避免了在微调过程中从语言模型中采样或调整重要超参数的需要。实验结果表明,DPO 算法可以与现有 RLHF 方法一样有效地从人类偏好中学习,甚至在某些任务中表现更好,比如情感调节、摘要和单轮对话。

DPO 是一种隐式优化策略,与现有的 RLHF 方法具有相同的目标,但更容易实现且易于训练。DPO 虽然增加了首选回复与其他回复之间的相对对数概率,但单纯的相对概率作为目标会引发模型退化。为了解决这个问题,DPO 使用了一个动态的权重表示每个示例回复的重要性。与现有的算法一样,DPO 同样依赖一个理论上的偏好模型,用于衡量给定的奖励函数和实际的偏好数据之间的对齐程度。然而与其他算法不同的是,DPO 使用变量的变化将偏好损失直接定义为一个策略函数。因此,DPO 能根据给定的偏好数据和模型回复,用简单的二进制交叉熵作为目标进行策略优化。

DPO 针对人类偏好进行了优化,同时避免了强化学习。在大语言模型微调中,现有的基于人类反馈的方法都会首先将奖励模型拟合到一个包含提示和人类偏好的数据集上,然后使用对比学习来找到一个策略最大化学习到的奖励。相比之下,DPO 只通过简单的分类目标,就能直接针对最满足人类偏好的策略进行优化,无需明确的奖励函数或者强化学习。

相关资料

🌟探索国内 Zilliz Cloud 云服务可点击访问 https://zilliz.com.cn,海外官网和云服务入口为 https://zilliz.com


  • 如果在使用 Milvus 或 Zilliz 产品有任何问题,可添加小助手微信 “zilliz-tech” 加入交流群。

  • 欢迎关注微信公众号“Zilliz”,了解最新资讯。

用户头像

Zilliz

关注

Data Infrastructure for AI Made Easy 2021-10-09 加入

还未添加个人简介

评论

发布
暂无评论
DPO 直接偏好优化:跳过复杂的对抗学习,语言模型本来就会奖励算法_AIGC_Zilliz_InfoQ写作社区