Beyond Pairs: Your Language Model is Secretly Optimizing a Preference Graph

Generative AI & LLMs
Published: arXiv: 2605.08037v1
Authors

Ning Liu Chuanneng Sun Kristina Klinkner Shervin Malmasi

Abstract

Direct Preference Optimization (DPO) aligns language models using pairwise preference comparisons, offering a simple and effective alternative to Reinforcement Learning (RL) from human feedback. However, in many practical settings, training data consists of multiple rollouts per prompt, inducing rich preference structure that pairwise DPO fails to exploit. Collapsing such data into independent pairs discards transitivity, introduces redundant or conflicting supervision, and can lead to unstable optimization. We propose Graph Direct Preference Optimization (GraphDPO), a principled generalization of DPO that operates over directed acyclic preference graphs induced by rollout rankings. GraphDPO encodes dominance relations as edges and optimizes a graph-structured Plackett--Luce-inspired objective that aggregates supervision over graph neighborhoods, enforcing transitivity while recovering standard DPO as a special case. To handle discrete or sparse signals, we introduce an equivalence-class construction where responses with identical preferences form graph layers, and intra-layer edges contribute zero loss, preventing spurious gradients. Despite leveraging full graph structure, GraphDPO maintains linear per-prompt complexity via efficient log-sum-exp aggregation. We further incorporate optional ground-truth anchoring by inserting verified solutions as dominant nodes and applying an annealed schedule that stabilizes early training while gradually relaxing oracle supervision. Experiments on reasoning and program synthesis tasks demonstrate superior performance, suggesting that graph-structured preference modeling is a scalable and robust alternative to pairwise and listwise alignment objectives.

Paper Summary

Problem
Aligning large language models with human preferences is crucial for deploying safe, reliable, and useful AI systems. However, current methods, such as Reinforcement Learning from Human Feedback (RLHF), are often expensive and prone to reward misspecification and optimization instability. Direct Preference Optimization (DPO) offers a simpler alternative but still has limitations when dealing with complex preference data.
Key Innovation
The researchers propose Graph Direct Preference Optimization (GraphDPO), a principled generalization of DPO that operates over directed acyclic preference graphs induced by rollout rankings. GraphDPO encodes dominance relations as edges and optimizes a graph-structured objective that aggregates supervision over graph neighborhoods, enforcing transitivity while recovering standard DPO as a special case.
Practical Impact
GraphDPO has the potential to improve the alignment of language models with human preferences, leading to more accurate and reliable AI systems. By leveraging the full structure of ranked rollouts, GraphDPO can handle complex preference data more effectively than current methods. This can have significant practical impacts in applications such as reasoning and program synthesis tasks.
Analogy / Intuitive Explanation
Imagine you're trying to find the best restaurant in a city. You ask friends for their recommendations, and they give you a list of their top choices. Current methods would treat each comparison between two restaurants as an isolated pair, whereas GraphDPO would look at the entire list of recommendations and see how each restaurant is ranked relative to the others. This allows GraphDPO to capture more nuanced information and make better decisions.
Paper Information
Categories:
cs.LG cs.AI
Published Date:

arXiv ID:

2605.08037v1

Quick Actions