Learning Where It Matters: Geometric Anchoring for Robust Preference Alignment
cs.LG
Submitted: 2026-02-04
Updated: 2026-09-25
Comments: Accepted at NeurIPS 2026
Code: https://github.com/tatsu-lab/alpaca_eval
License: http://creativecommons.org/licenses/by/4.0/
The gist: Direct Preference Optimization (DPO) and related methods align large language models from pairwise preferences by regularizing updates against a fixed reference policy.
Terminology
Abstract
Direct Preference Optimization (DPO) and related methods align large language models from pairwise preferences by regularizing updates against a fixed reference policy. As the policy drifts, a static reference, however, can become increasingly miscalibrated, leading to distributional mismatch and amplifying spurious preference signals under noisy supervision. Conversely, reference-free variants avoid mismatch but often suffer from unconstrained reward drift. We propose Geometric Anchor Preference Optimization (GAPO), which replaces the fixed reference with a dynamic, geometry-aware anchor: an adversarial local perturbation of the current policy within a small radius that serves as a pessimistic baseline. This anchor enables an adaptive reweighting mechanism, modulating the importance of each preference pair based on its local sensitivity. We further introduce the Anchor Gap, the reward discrepancy between the policy and its anchor, and show under smoothness conditions that it approximates worst-case local margin degradation. Optimizing a logistic objective weighted by this gap downweights geometrically brittle instances while emphasizing robust preference signals. Across diverse noise settings, GAPO consistently improves robustness while matching or improving performance on standard LLM alignment and reasoning benchmarks.
Sources
- Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback
- Provably Robust DPO: Aligning Language Models with Noisy Feedback
- KTO: Model Alignment as Prospect Theoretic Optimization
- Sharpness-Aware Minimization for Efficiently Improving Generalization
- ORPO: Monolithic Preference Optimization without Reference Model
- Disentangling Length from Quality in Direct Preference Optimization
- Proximal Policy Optimization Algorithms
- Gemma 2: Improving Open Language Models at a Practical Size
- Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimization
- Contrastive Preference Optimization: Pushing the Boundaries of LLM Performance in Machine Translation
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks