atlas

Research map / Reinforcement learning / RL from feedback

Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithms

NeurIPS 2024 · Reinforcement learning › RL from feedback · approach: Theory of RLHF · 7 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.

Shows that direct alignment algorithms (DPO, IPO, SLiC) exhibit reward over-optimization-style degradation at high KL budgets, fits a scaling law from RLHF to GPT-4 win rates, and explains it via under-constrained objectives and OOD probability mass in a toy MDP.

Problem

Reward over-optimization (hacking) is well characterized for RLHF with a learned proxy reward, but it is unclear how it manifests in direct alignment algorithms (DAAs) that have no explicit reward model.

Approach

Treats DAA win-rate versus KL budget as an over-optimization curve, fits the RLHF reward scaling law R(d)=d(alpha - beta log d) to GPT-4 win rates as a gold proxy, and attributes the degradation to under-constrained (non-strictly-convex) DAA objectives that push probability mass onto out-of-distribution responses.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
increase the KL budgetDPO, IPO, and SLiC · Reddit TL;DR; Pythia 1B, 2.8B, and 6.9BGPT-4 win rateMixed
continue training beyond the first 25% of an epochDPO, IPO, and SLiC · Reddit TL;DR; Pythia 6.9BGPT-4 win rateHurt
use IPO instead of DPO or SLiCDirect Alignment Algorithms · Reddit TL;DR; Pythia modelsover-optimization susceptibilityHelped
increase model size from Pythia 1B to Pythia 6.9BDPO, IPO, and SLiC · Reddit TL;DRGPT-4 win rateHelped
use length regularizationDPO · Reddit TL;DR; Pythia 2.8Blength-corrected GPT-4 win rateHelped
use length regularizationDPO · Reddit TL;DR; Pythia 2.8B, lower KL constraint regionGPT-4 win rateHurt
use length regularizationDPO · Reddit TL;DR; Pythia 2.8Breward over-optimizationNo clear effect
fit the reward-versus-KL scaling law against square-root KL instead of using a quadratic fit against KLDirect Alignment Algorithms · Reddit TL;DR; GPT-4 win ratesRMSE: halves the RMSEHelped
increase model capacityDPO · Reddit TL;DR; Pythia modelslength extrapolation (R²)Helped
use a smaller KL budgetDPO · Reddit TL;DR; Pythia modelslength extrapolation (R²)Hurt
increase the DAA implicit reward classification accuracyDPO and SLiC · Reddit TL;DR; Pythia modelsGPT-4 win rateNo clear effect
increase the DAA implicit reward classification accuracyIPO · Reddit TL;DR; within each Pythia model sizeGPT-4 win rateNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: OpenAI TL;DR summarization preferences (92K) Anthropic Helpfulness-Harmlessness

Benchmarks: OpenAI TL;DR held-out prompts Anthropic-HH

Metrics: GPT-4 win rate KL divergence implicit reward accuracy RMSE of scaling law fit R^2 of length regression

Baselines: DPO IPO SLiC-HF

Builds on: DPO IPO SLiC-HF Scaling Laws for Reward Model Overoptimization

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