Post

Preference learning without the reward model

Direct Preference Optimization teaches language models to follow human preferences without a trained reward model or reinforcement learning. A close reading of the Stanford paper that redrew the alignment recipe.

Making a language model follow human preferences used to mean building a second model first. The standard recipe, reinforcement learning from human feedback, trains a reward model on preference pairs and then uses reinforcement learning to push the policy toward higher scores. It works, but it is heavy: a reward model to fit, fresh samples from the policy at every training step, and enough stability tricks to fill a paper of their own. Direct Preference Optimization, posted to arXiv in May 2023 and presented at NeurIPS 2023, deletes the middle of that pipeline. The reward model disappears. The reinforcement loop disappears. What remains is a loss function that fits in a paragraph.

The authors, Rafael Rafailov, Archit Sharma, Eric Mitchell, Stefano Ermon, Christopher Manning and Chelsea Finn of Stanford, built the method on a change of variables. The usual RLHF objective couples a reward model with a KL penalty that keeps the tuned policy close to a reference model. The optimal policy for that objective has a closed form: the reference distribution reweighted by the exponentiated reward. Turn the equation around, solve for the reward in terms of the policy, and substitute back into the objective. The reward model is gone, and in its place sits an objective defined only on the policy and the preference data.

Training reduces to a classification problem over pairs. For each pair of chosen and rejected responses the loss widens the gap between the model’s log-probability of the chosen answer and its log-probability of the rejected one, with a temperature parameter standing in for the KL coefficient. No reward fitting, no policy sampling, no PPO. The authors describe the resulting algorithm as stable, performant and computationally lightweight, and the experiments back that up. Across sentiment control, summarization and dialogue, fine-tuning with DPO matches or beats the PPO-based baseline on evaluator-judged win rates, while using a small fraction of the compute.

What makes the paper important is not only the result but the shape of the idea. Alignment had been treated as an engineering problem: build a reward, stabilize an RL loop, tune. DPO shows that a large part of that machinery was scaffolding. Once the objective is rewritten, alignment becomes a data problem. The quality of the policy is bounded by the quality of the preference pairs, and the algorithm itself is simple enough to run on a single GPU in an afternoon. That simplicity is the paper’s quiet argument. In a field where methods compete on benchmarks, a formula that fits on a whiteboard spreads faster than any system that needs a cluster to train and a team to debug.

The approach does not answer every question about preference learning, and it was not meant to. The reference model still constrains how far the policy can move, so the choice of that anchor matters. Biases in the preference data pass through unchanged, because the method faithfully amplifies whatever signal the pairs contain. And a classification loss treats every pair as equally informative, which is not obviously right. These are open problems, and the fact that they can be stated this cleanly is itself a mark of how far the field has come.

A year ago the standard advice for aligning a model was to reach for a reward model and a PPO loop. The winter’s NeurIPS proceedings now contain the counter-argument, and it is hard to look at the two side by side and think the heavier machinery is the better default. Preference learning without the reward model turned out not to be a compromise. It is the cleaner statement of the same problem, and the field is still catching up with the consequences.

This post is licensed under CC BY 4.0 by the author.