arXiv:2510.05342cs.LGcs.AI2025-10被引 1

通过自适应调整学习信号,让模型更专注难样本,提升对齐效果。

Margin Adaptive DPO: Leveraging Reward Model for Granular Control in Preference Optimization

  • 用奖励模型估计偏好差距,为每条数据动态调整训练权重。
  • 在摘要任务上,不同解码温度下均优于现有方法,最高提升12.3%。
  • 适合需要精细控制训练过程的模型对齐研究者使用。

直接偏好优化(DPO)是大语言模型对齐的有效方法,但其固定温度参数导致在多样偏好数据上训练效果不佳,易过拟合简单样本而忽略信息丰富的样本。现有方法如IPO虽缓解过拟合,但正则化过于保守;β-DPO虽更精准,却存在批级适应导致的折中温度、线性更新引发负β值及过滤机制丢弃有效信号等问题。本文提出边际自适应直接偏好优化(MADPO),采用两步法:先训练奖励模型估算偏好边际,再基于该边际对每条样本的DPO损失施加连续自适应权重。该重加权机制使难样本的学习信号增强、易样本减弱,实现粒度化的学习调控。我们提供理论分析,证明MADPO具有稳定优化景观且对奖励模型误差鲁棒。在基于人工偏好数据的摘要任务实验中,MADPO在多种解码温度下持续超越强基线,性能最优提升达12.3%。

原文摘要 · Abstract (English)

Direct Preference Optimization (DPO) has emerged as a simple and effective method for aligning large language models. However, its reliance on a fixed temperature parameter leads to suboptimal training on diverse preference data, causing overfitting on easy examples and under-learning from informative ones. Recent methods have emerged to counter this. While IPO addresses general overfitting, its uniform regularization can be overly conservative. The more targeted approach of $β$-DPO suffers from its own limitations: its batch-level adaptation applies a single, compromised temperature to mixed-margin pairs, its linear update rule can produce unstable negative $β$ values, and its filtering mechanism discards potentially useful training signals. In this work, we introduce Margin-Adaptive Direct Preference Optimization (MADPO), a method that provides a stable, data-preserving, and instance-level solution. MADPO employs a practical two-step approach: it first trains a reward model to estimate preference margins and then uses these margins to apply a continuous, adaptive weight to the DPO loss for each individual training sample. This re-weighting scheme creates an effective target margin that is amplified for hard pairs and dampened for easy pairs, allowing for granular control over the learning signal. We provide a comprehensive theoretical analysis, proving that MADPO has a well-behaved optimization landscape and is robust to reward model estimation errors. We validate our theory with experiments on a summarization task using human preference data. MADPO consistently outperforms strong baselines across a comprehensive sweep of decoding temperatures.

偏好优化奖励模型模型对齐

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