机器学习 | 简单偏好优化 (SimPO)
简单偏好优化 (Simple Preference optimization, SimPO):大语言模型强化学习中的一种偏好优化方法,其对齐了偏好优化目标中的奖励函数与生成指标,同时解放了参考模型,相比 DPO 更简单、稳定与高效。
论文:https://arxiv.org/abs/2405.14734
1 DPO 的问题
回顾 DPO,它的目标函数是:
它的奖励函数是:
可以发现 DPO 有以下的问题:
- DPO 在训练时需要参考模型
,这带来了额外的内存和计算开销。 - 由于推理时没有
,DPO 偏好优化时的奖励函数和推理时的生成指标不一致。
第一点是显而易见的,下面展开分析一下第二点。
在生成阶段,策略模型
找到这个平均对数似然序列的最优解是非常昂贵的,因此在实际中是通过各种解码策略找到这个序列的局部最优解。这些解码策略在“语言模型解码算法”这篇笔记中有详细列出,可以跳转查看。
可以看到,在生成阶段,是没有模型
SimPO 便是针对 DPO 的这两个问题进行优化,其对齐了偏好优化目标中的奖励函数与生成指标,同时无需参考模型。
2 SimPO 方法
2.1 长度归一化奖励公式
为了对齐训练和推理,同时消去参考模型,SimPO 的想法很简单,直接拿生成时的指标来做训练时的奖励函数不就行了:
同时,需要注意长度归一化项是关键的。若没有该归一化项,长的序列往往时会有更低的对数似然值的,这就会导致模型在处理较长序列时必须额外拉高概率来弥补长度带来的损失,可能导致模型效果的降低。
2.2 目标奖励差额
SimPO 还引入了目标奖励差额 (Target reward margin) 概念:
其中,
这个变动造成的影响比较好理解,就是获胜的奖励
2.3 目标函数
综上,结合 2.1, 2.2,SimPO 最终的目标函数便是:
本文采用 CC BY-SA 4.0 许可,本文 Markdown 源码:Haotian-BiJi