提出梯度门控机制,解决大模型对齐中的概率坍缩问题。
Gradient-Gated DPO: Stabilizing Preference Optimization in Language Models

- 根据概率几何调节拒答梯度,抑制有害更新
- 实验显示可降低挤压效应并提升优选响应概率
- 小模型经门控后性能超大无门控模型,凸显梯度控制重要性
偏好优化已成为对齐大语言模型与人类反馈的核心范式。直接偏好优化(DPO)通过直接优化成对偏好简化了强化学习中的反馈过程,无需奖励建模与策略优化。然而,近期研究发现DPO存在挤压效应:作用于被拒绝响应的负梯度会集中概率质量至高置信度预测,同时压制其他选项。这一现象在简单softmax模型中即出现,可能导致训练中系统性概率坍缩。本文提出梯度门控偏好优化(Gate-DPO),通过根据模型概率几何调节拒答梯度,稳定训练过程。当更新目标为极低概率响应时,门控机制削弱有害梯度,同时保持标准优化行为。Gate-DPO在不改变原有偏好目标的前提下解决该优化病态,且与扩展SFT、IPO、Cal-DPO等方法兼容。跨多种架构与偏好数据集的实验表明,Gate-DPO持续减少挤压效应并提升优选响应概率。质量动力学分析进一步揭示更健康的优化行为:优选响应增强,整体分布抑制减弱。值得注意的是,较小的有门控模型在优选响应提升上甚至超过更大的无门控模型,表明控制梯度动态而非单纯扩大规模,才是实现稳定高效对齐的关键。
原文摘要 · Abstract (English)
Preference optimization has become a central paradigm for aligning large language models with human feedback. Direct Preference Optimization (DPO) simplifies reinforcement learning from human feedback by directly optimizing pairwise preferences, removing the need for reward modeling and policy optimization. However, recent work shows that DPO exhibits a squeezing effect, where negative gradients applied to rejected responses concentrate probability mass on high-confidence predictions while suppressing alternative responses. This phenomenon arises even in simple softmax models and can lead to systematic probability collapse during training. We introduce Gradient-Gated Preference Optimization (Gate-DPO), a method that stabilizes training by modulating rejected gradients according to the model's probability geometry. When updates target extremely low-probability responses, the gate attenuates harmful gradients while preserving standard optimization behavior. Gate-DPO addresses this optimization pathology without modifying the underlying preference objective and is complementary to existing methods such as extended SFT, IPO, and Cal-DPO. Experiments across multiple architectures and preference datasets show that Gate-DPO consistently reduces squeezing and improves chosen-response likelihood. Mass-dynamics analysis further reveals healthier optimization behavior, with improved preferred responses and reduced suppression of the overall distribution. Notably, smaller gated models can exhibit stronger chosen-response improvements than larger ungated models, suggesting that controlling gradient dynamics, rather than scale alone, is key to stable and efficient alignment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。