Mind the Gap: Structure-Aware Consistency in Preference Learning
Abstract
Lay Summary
When AI language models are trained to match human preferences, surrogate loss functions are used in place of the original zero-one ranking objective, which is intractable to optimize directly. The most popular method, Direct Preference Optimization (DPO), and related approaches all rely on these surrogates as stand-ins for the true goal: correctly ranking responses. We prove that these surrogates are fundamentally unreliable for neural networks: a model can appear to perform perfectly on the surrogate while still ranking responses incorrectly. The root cause is that the model can "cheat" by making score differences vanishingly small, rather than actually learning which response is better. We fix this by requiring the model to maintain a minimum confidence gap between responses it ranks differently, and prove this gap is mathematically necessary. We further show that this gap should vary with how different the responses are: near-identical responses need only a small gap, while distinct ones require a large one. This leads to Structure-Aware DPO (SA-DPO), which adapts its margin to response similarity. We also establish a strict hierarchy among loss functions for capacity-bounded models. In experiments, SA-DPO consistently outperforms existing methods, achieving a 58.5% win-rate in head-to-head output quality comparisons.