arXiv:2412.13862cs.LGcs.CL2024-12ICML被引 9

用能量模型改进偏好对齐,解决传统方法多解难题。

Energy-Based Preference Model Offers Better Offline Alignment than the Bradley-Terry Preference Model

  • 提出能量模型替代贝特朗偏好模型,确保最优解唯一。
  • 对比损失EPA在负样本充足时逼近误差几乎为零。
  • 实测优于DPO,在公开评测中表现更优,适合偏好对齐研究者。

自DPO问世以来,已证明通过KL约束的强化学习人类反馈(RLHF)损失对齐目标大模型与人类偏好,数学上等价于一种特殊的奖励建模任务:即使用目标大模型参数化奖励模型,并使其与真实奖励呈1:1线性关系。然而我们发现一个关键问题:DPO损失可能存在多个极小值点,其中仅一个满足线性条件。这源于底层贝特朗偏好模型的一个经典问题——最大似然估计量(MLE)不唯一。因此,理想的最小化点可能无法达到。为此,我们提出一种能量模型(EBM),其始终具有唯一MLE,天然满足线性要求。为近似该MLE,我们设计了一种对比损失——能量偏好对齐(EPA),每个正样本与一个或多个强负样本及大量弱自由负样本进行对比。理论表明,当负样本足够多时,EPA的近似误差几乎必然趋近于零。实验显示,EPA在多个开放基准上持续优于DPO,验证了所提能量模型的优势。

原文摘要 · Abstract (English)

Since the debut of DPO, it has been shown that aligning a target LLM with human preferences via the KL-constrained RLHF loss is mathematically equivalent to a special kind of reward modeling task. Concretely, the task requires: 1) using the target LLM to parameterize the reward model, and 2) tuning the reward model so that it has a 1:1 linear relationship with the true reward. However, we identify a significant issue: the DPO loss might have multiple minimizers, of which only one satisfies the required linearity condition. The problem arises from a well-known issue of the underlying Bradley-Terry preference model: it does not always have a unique maximum likelihood estimator (MLE). Consequently,the minimizer of the RLHF loss might be unattainable because it is merely one among many minimizers of the DPO loss. As a better alternative, we propose an energy-based model (EBM) that always has a unique MLE, inherently satisfying the linearity requirement. To approximate the MLE in practice, we propose a contrastive loss named Energy Preference Alignment (EPA), wherein each positive sample is contrasted against one or more strong negatives as well as many free weak negatives. Theoretical properties of our EBM enable the approximation error of EPA to almost surely vanish when a sufficient number of negatives are used. Empirically, we demonstrate that EPA consistently delivers better performance on open benchmarks compared to DPO, thereby showing the superiority of our EBM.

偏好对齐能量模型大模型训练强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。