Table of Contents
Fetching ...

ADPO: Anchored Direct Preference Optimization

Wang Zixian

TL;DR

ADPO introduces Anchored Direct Preference Optimization, which minimizes $\mathrm{KL}(q\|\tilde{p}_\theta)$ where $\tilde{p}_\theta=\mathrm{softmax}\big(\frac{\boldsymbol{\ell}-\boldsymbol{\ell}^{\text{ref}}}{\tau_{\text{anc}}}\big)$. By centering logits around a reference policy and using forward KL, ADPO unifies supervised fine-tuning, knowledge distillation, maximum-entropy RL, and DPO as special cases, and reveals an implicit trust region governed by the softmax Fisher metric with curvature $\propto 1/\tau_{\text{anc}}^2$. Empirically, ADPO achieves 12–93% improvements over standard DPO across 12 noisy scenarios, with listwise variants often topping results; offline distillation sees up to 4–49× reductions in student–teacher KL and substantial returns. The framework also identifies task-dependent anchor strategies: dynamic anchors boost online exploration in noisy environments, while fixed anchors stabilize offline distillation. Overall, anchoring emerges as a general principle for robust policy optimization across diverse learning contexts, guiding anchor selection for online RLHF and offline distillation.

Abstract

Direct Preference Optimization (DPO) has emerged as a simple alternative to reinforcement learning from human feedback (RLHF) for aligning language models, but its reliance on hard pairwise labels makes it brittle under noise; our experiments show performance degrading by up to 93 percent in noisy settings. We introduce Anchored Direct Preference Optimization (ADPO), a unified framework that addresses this fragility through reference anchoring. By minimizing KL(q || softmax((l - l_ref) / tau_anc)), where l_ref are reference policy log probabilities, ADPO provides three key advantages: (1) it unifies major learning paradigms, including supervised fine-tuning, knowledge distillation, maximum-entropy reinforcement learning, and DPO, as special cases through different choices of target distribution q, anchor policy pi_ref, and temperature tau_anc; (2) it induces an implicit trust region governed by the softmax Fisher metric with curvature scaling as 1 / tau_anc^2, providing geometric regularization absent in standard methods; and (3) it enables flexible anchor strategies tailored to different learning contexts. Empirically, ADPO consistently outperforms standard DPO by 12 to 93 percent across twelve noisy scenarios, with listwise variants achieving top performance in eleven of twelve cases. In offline distillation, ADPO reduces student-teacher KL by 4 to 49 times while achieving superior returns (for example, 279.3 vs -309.0 for knowledge distillation on HalfCheetah). We further uncover a task-dependent tradeoff: dynamic anchors excel at online exploration in noisy environments (plus 5 to 11 percent), while fixed anchors enable stable offline distillation. Our work establishes anchoring as a general principle for robust policy optimization, with clear practical guidance for anchor selection across diverse learning scenarios.

ADPO: Anchored Direct Preference Optimization

TL;DR

ADPO introduces Anchored Direct Preference Optimization, which minimizes where . By centering logits around a reference policy and using forward KL, ADPO unifies supervised fine-tuning, knowledge distillation, maximum-entropy RL, and DPO as special cases, and reveals an implicit trust region governed by the softmax Fisher metric with curvature . Empirically, ADPO achieves 12–93% improvements over standard DPO across 12 noisy scenarios, with listwise variants often topping results; offline distillation sees up to 4–49× reductions in student–teacher KL and substantial returns. The framework also identifies task-dependent anchor strategies: dynamic anchors boost online exploration in noisy environments, while fixed anchors stabilize offline distillation. Overall, anchoring emerges as a general principle for robust policy optimization across diverse learning contexts, guiding anchor selection for online RLHF and offline distillation.

Abstract

Direct Preference Optimization (DPO) has emerged as a simple alternative to reinforcement learning from human feedback (RLHF) for aligning language models, but its reliance on hard pairwise labels makes it brittle under noise; our experiments show performance degrading by up to 93 percent in noisy settings. We introduce Anchored Direct Preference Optimization (ADPO), a unified framework that addresses this fragility through reference anchoring. By minimizing KL(q || softmax((l - l_ref) / tau_anc)), where l_ref are reference policy log probabilities, ADPO provides three key advantages: (1) it unifies major learning paradigms, including supervised fine-tuning, knowledge distillation, maximum-entropy reinforcement learning, and DPO, as special cases through different choices of target distribution q, anchor policy pi_ref, and temperature tau_anc; (2) it induces an implicit trust region governed by the softmax Fisher metric with curvature scaling as 1 / tau_anc^2, providing geometric regularization absent in standard methods; and (3) it enables flexible anchor strategies tailored to different learning contexts. Empirically, ADPO consistently outperforms standard DPO by 12 to 93 percent across twelve noisy scenarios, with listwise variants achieving top performance in eleven of twelve cases. In offline distillation, ADPO reduces student-teacher KL by 4 to 49 times while achieving superior returns (for example, 279.3 vs -309.0 for knowledge distillation on HalfCheetah). We further uncover a task-dependent tradeoff: dynamic anchors excel at online exploration in noisy environments (plus 5 to 11 percent), while fixed anchors enable stable offline distillation. Our work establishes anchoring as a general principle for robust policy optimization, with clear practical guidance for anchor selection across diverse learning scenarios.
Paper Structure (101 sections, 9 theorems, 49 equations, 3 figures, 8 tables, 1 algorithm)

This paper contains 101 sections, 9 theorems, 49 equations, 3 figures, 8 tables, 1 algorithm.

Key Result

Lemma 2.1

For any context-independent shift $h(s)$, if $r'(s,a)=r(s,a)-h(s)$ then $\pi^*$ in eq:boltzmann_opt is unchanged; hence all pairwise/listwise probabilities (BT/PL) are identical, and anchored ADPO---which operates on $(\ell-\ell^{\mathrm{ref}})$---inherits groupwise shift invariance.

Figures (3)

  • Figure 1: Fisher geometry as implicit trust region. The Fisher metric $\mathop{\mathrm{Diag}}\nolimits(q)-qq^\top$ defines a quadratic bowl in anchored coordinates $u=(\ell-\ell^{\mathrm{ref}})/\tau_{\mathrm{anc}}$, centered at target $q$. Smaller $\tau_{\mathrm{anc}}$ yields tighter curvature ($\propto 1/\tau_{\mathrm{anc}}^2$).
  • Figure 2: Comprehensive $2\times 2$ comparison across 12 scenarios (10 seeds each). Each subplot shows convergence curves for 6 methods. Key findings: (i) ADPO methods (solid orange/red/green lines) consistently outperform Standard DPO (solid/dashed blue lines) across all scenarios; (ii) listwise methods achieve highest final performance in 11/12 scenarios; (iii) performance gap widens as difficulty increases (left to right within each row), reaching 93% improvement under Heavy Noise-Heavy; (iv) anchored methods show faster convergence and higher stability. Error bands: mean $\pm$ s.e.
  • Figure 3: Model scale comparison (Heavy Noise-Medium, 10 seeds). ADPO's advantage grows with model capacity. Small model: +23% (0.516 vs. 0.420). Medium: +62% (0.716 vs. 0.440). Large: +73% (0.718 vs. 0.416). Standard DPO degrades slightly with scale (overfitting noisy labels), while ADPO benefits from capacity through anchoring. Error bands: mean $\pm$ s.e.

Theorems & Definitions (19)

  • Lemma 2.1: Groupwise shift invariance
  • proof : Proof sketch
  • Remark 3.1: Sequence tasks and length normalization
  • Proposition 3.2: Closed-form optimum
  • proof : Proof sketch
  • Remark 3.3: Optional: Fisher variance scaling
  • Proposition 3.4: ADPO unifies common paradigms
  • proof : Verification
  • Remark 3.5: Theory vs practice
  • Lemma 3.6: Implicit trust region
  • ...and 9 more