atlas

Research map / Reinforcement learning / RL from feedback

The Importance of Online Data: Understanding Preference Fine-tuning via Coverage

NeurIPS 2024 · Reinforcement learning › RL from feedback · approach: Reward modeling versus direct optimization · 1 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.

Theoretically separates offline contrastive preference methods (DPO/IPO, needing global coverage) from online RLHF (needing only partial KL-ball coverage), and proposes HyPO, which adds online unlabeled samples for a reverse-KL regularizer to DPO.

Problem

Understanding the statistical separation between online RLHF (PPO with a learned reward model) and offline contrastive preference optimization (DPO, IPO) for LLM fine-tuning, and why online methods often outperform offline ones when preference data lacks diversity.

Approach

Uses dataset coverage to show global coverage is necessary and sufficient for DPO/IPO, while a weaker local KL-ball coverage suffices for online RLHF because reverse KL is controlled via online samples; this motivates HyPO, which applies the DPO loss to offline data and a reverse-KL penalty estimated on unlabeled online samples.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add online reverse-KL regularization to DPODPO on TL;DR with Pythia 1.4B · TL;DRGPT4 winrate: 42.17% (2.5%) -> 46.44% (2.39%)Helped
Add online reverse-KL regularization to DPODPO on TL;DR with Pythia 1.4B · TL;DRRM score: 0.16 (0.05) -> 0.37 (0.05)Helped
Add online reverse-KL regularization to DPODPO on TL;DR with Pythia 1.4B · TL;DRreverse KL: 44.90 (1.29) -> 27.07 (2.34)Helped
Add online reverse-KL regularization to DPODPO on TL;DR with Pythia 2.8B · TL;DRGPT4 winrate: 44.39% (0.4%) -> 50.50% (1.89%)Helped
Add online reverse-KL regularization to DPODPO on TL;DR with Pythia 2.8B · TL;DRRM score: 2.43 (0.10) -> 2.51 (0.13)Helped
Add online reverse-KL regularization to DPODPO on TL;DR with Pythia 2.8B · TL;DRreverse KL: 68.95 (3.08) -> 48.98 (4.23)Helped
Add online reverse-KL regularization to DPODPO-Llama-3 · MT-BenchAverage: 7.75 -> 8.09Helped
Add online reverse-KL regularization to DPODPO-Llama-3 · AlpacaEval 2.0LC Win Rate: 28.4 -> 30.7Helped
Add online reverse-KL regularization to DPODPO-Llama-3 · AlpacaEval 2.0Win Rate: 30.9 -> 32.2Helped
Add online reverse-KL regularization to DPODPO-Llama-3 · Open LLM leaderboardAverage: 65.34 -> 65.51Helped
Add online reverse-KL regularization to DPODPO-Llama-3 · MMLU (5-shot)score: 65.82 -> 65.74Hurt
Add online reverse-KL regularization to DPODPO-Llama-3 · Arc (25-shot)score: 63.14 -> 62.71Hurt

Surprising findings

Limitations

Experimental setup

Datasets: TL;DR summarization preference dataset UltraFeedback (HuggingFaceH4/ultrafeedback_binarized)

Benchmarks: TL;DR summarization AlpacaEval 2.0 MT-Bench MMLU GSM8K ARC TruthfulQA HellaSwag

Metrics: GPT-4 win rate reward model score reverse KL to reference length-controlled win rate MT-Bench score accuracy

Baselines: DPO IPO Meta-Llama-3-8B-Instruct

Builds on: DPO PPO IPO RLOO

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