atlas

Research map / Large language models / Alignment and preferences

Is DPO Superior to PPO for LLM Alignment? A Comprehensive Study

ICML 2024 · Large language models › Alignment and preferences · approach: Active and token-level preference learning · 8 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Compares DPO and PPO for LLM alignment theoretically and empirically, finds DPO sensitive to distribution shift, identifies key PPO factors (advantage normalization, large batch, reference EMA), and shows PPO beats DPO on dialogue and code tasks.

Problem

Whether DPO, a reward-free RLHF method popular on academic benchmarks, is truly superior to reward-based PPO for LLM alignment, and why PPO underperforms on those benchmarks.

Approach

Theoretically shows PPO's reachable policy set is a proper subset of DPO's, yet DPO can produce policies that favor out-of-distribution responses; empirically shows DPO is hurt by distribution shift, and that tuned PPO (advantage normalization, large batch size, reference-model EMA) outperforms DPO on dialogue and code generation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use PPO rather than DPOLlama2-7B alignment · HH-RLHF test setOpenAssistant reward: 0.611 -> 0.718Helped
use PPO rather than DPO-IterLlama2-7B alignment · HH-RLHF test setOpenAssistant reward: 0.678 -> 0.718Helped
use PPO rather than DPO-IterCode Llama 34B · CodeContest Test Set10@1k: 3.2% -> 22.4%Helped
add advantage normalization (ablation)PPO · APPS Intro.pass@5: 18.0% -> 38.1%Helped
use a large batch (ablation)PPO · APPS Comp.pass@5: 1.1% -> 7.5%Helped
add exponential moving average reference-model updates (ablation)PPO · APPS Comp.pass@5: 7.5% -> 9.1%Helped
use a large batch (ablation)PPO · CodeContest pass@1kpass@1k: 7.7% -> 19.6%Helped
add exponential moving average reference-model updates (ablation)PPO · CodeContest pass@1kpass@1k: 19.6% -> 21.4%Helped
use SFT (Safe) instead of SFT (Alpaca) as reference model (ablation)DPO · SafeRLHFS.R.: 55.4% -> 71.8%Helped
filter dual-unsafe preference pairs (ablation)DPO · SafeRLHFΔHelp.: -1.62 -> 2.46Helped
filter dual-safe preference pairs (ablation)DPO · SafeRLHFΔHelp.: -1.62 -> -2.86Hurt
increase β from 0.1 to 0.2 (ablation)PPO · HH-RLHF, Llama-7BOpenAssistant reward: 0.718 -> 0.629Hurt

Surprising findings

Limitations

Experimental setup

Datasets: HH-RLHF SafeRLHF APPS CodeContest Alpaca

Benchmarks: HH-RLHF SafeRLHF APPS CodeContest

Metrics: OpenAssistant reward GPT-4 win rate helpfulness delta harmlessness score safety rate pass@k

Baselines: DPO DPO-Iter RRHF PRO SFT AlphaCode 41B CodeRL Beaver

Builds on: DPO PPO DeepSpeed-Chat generalized advantage estimation

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy