arXiv:2410.08847cs.LGcs.AI2024-10ICLR被引 67

DPO训练时偏好响应概率反而下降,可能引发模型误判,该研究揭示了原因并提出解决方法。

Unintentional Unalignment: Likelihood Displacement in Direct Preference Optimization

  • 发现偏好相似的响应会导致概率错位,使模型更倾向错误答案
  • 实验显示拒绝危险指令的能力从74.4%降至33.4%,严重偏离对齐目标
  • 用嵌入相似度指标筛选数据,可有效缓解意外失准问题

直接偏好优化(DPO)及其变体被广泛用于将语言模型对齐人类偏好。尽管这些方法旨在提高偏好响应的生成频率,但已有研究表明训练过程中偏好响应的似然性常下降。本文揭示了这一反直觉现象的成因与影响,称之为似然位移。我们证明似然位移可能造成灾难性后果,导致概率质量从偏好响应转移到语义相反的响应。例如,训练模型偏好'No'而非'Never'会显著提升'Yes'的概率。在对模型进行安全拒答对齐时,这种位移可能导致意外失准,使模型将概率质量从预期的拒答响应转移到有害响应(如将Llama-3-8B-Instruct的拒答率从74.4%降至33.4%)。理论分析表明,似然位移由诱导相似隐藏嵌入的偏好驱动,通过中心化隐藏嵌入相似度(CHES)评分进行量化。实证上,该评分能识别数据集中导致位移的主要样本。剔除这些样本可有效缓解训练中的意外失准。更广泛地,研究强调需选择语义差异明显的偏好数据,而CHES评分或为此提供重要支持。

原文摘要 · Abstract (English)

Direct Preference Optimization (DPO) and its variants are increasingly used for aligning language models with human preferences. Although these methods are designed to teach a model to generate preferred responses more frequently relative to dispreferred responses, prior work has observed that the likelihood of preferred responses often decreases during training. The current work sheds light on the causes and implications of this counter-intuitive phenomenon, which we term likelihood displacement. We demonstrate that likelihood displacement can be catastrophic, shifting probability mass from preferred responses to responses with an opposite meaning. As a simple example, training a model to prefer $\texttt{No}$ over $\texttt{Never}$ can sharply increase the probability of $\texttt{Yes}$. Moreover, when aligning the model to refuse unsafe prompts, we show that such displacement can unintentionally lead to unalignment, by shifting probability mass from preferred refusal responses to harmful responses (e.g., reducing the refusal rate of Llama-3-8B-Instruct from 74.4% to 33.4%). We theoretically characterize that likelihood displacement is driven by preferences that induce similar embeddings, as measured by a centered hidden embedding similarity (CHES) score. Empirically, the CHES score enables identifying which training samples contribute most to likelihood displacement in a given dataset. Filtering out these samples effectively mitigated unintentional unalignment in our experiments. More broadly, our results highlight the importance of curating data with sufficiently distinct preferences, for which we believe the CHES score may prove valuable.

模型对齐偏好优化安全对齐嵌入相似度

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