Direct Preference Optimization: Your Language Model is Secretly a Reward Model
Rafailov et al., 2023, Direct Preference Optimization (Advances in Neural Information Processing Systems 36 (NeurIPS 2023))
Subscriber deck — free during early access
2 cards in “1 Introduction” · whole deck
1 Introduction
- What is wrong with RLHF that DPO addresses?RLHF is complex and often unstable: it fits a separate reward model to human preferences, then fine-tunes the language model with reinforcement learning (typically PPO) to maximize that reward without drifting too far from the original model, training several models and sampling from the policy during training.
- Rafailov et al., 2023, Direct Preference Optimization (Advances in Neural Information Processing Systems 36 (NeurIPS 2023)), §1
- What does DPO do, in one sentence?It optimizes the same KL-constrained reward-maximization objective as RLHF exactly, but with a single stage of policy training: a simple binary cross-entropy (classification) loss on preference pairs, with no explicit reward model and no reinforcement learning.
DPO replaces reward modelling plus reinforcement learning with one supervised loss - Rafailov et al., 2023, Direct Preference Optimization (Advances in Neural Information Processing Systems 36 (NeurIPS 2023)), §1
Source
- Rafael Rafailov, Archit Sharma, Eric Mitchell, Stefano Ermon, Christopher D. Manning and Chelsea Finn. 2023. Direct Preference Optimization: Your Language Model is Secretly a Reward Model. In Advances in Neural Information Processing Systems 36 (NeurIPS 2023). https://papers.nips.cc/paper_files/paper/2023/hash/a85b405ed65c6477a4fe8302b5e06ce7-Abstract-Conference.html