atlas

Research map / Reinforcement learning / RL from feedback

Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-constraint

ICML 2024 · Reinforcement learning › RL from feedback · approach: Theory of RLHF

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.

Formulates RLHF as a reverse-KL regularized contextual bandit and gives offline (pessimism), online/hybrid (batch exploration) theory, then practical iterative DPO and multi-step RSO algorithms evaluated on Open-LLaMA-3B-V2 with HH-RLHF and UltraRM gold reward.

Problem

Existing RLHF methods (PPO, DPO) lack rigorous theory under the reverse-KL regularized objective; offline methods need broad data coverage and suffer reward over-optimization.

Approach

Treat RLHF as a KL-regularized contextual bandit with Gibbs policy oracle; derive pessimistic offline learning and enhancer-based online exploration with finite-sample guarantees; implement via DPO with pessimistic margins and multi-step rejection sampling (RSO generalization).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Hybrid-GSHF-DPO instead of DPOOpen-LLaMA-3B-V2 alignment · HH-RLHF testGold Reward: 2.15 -> 2.67Helped
Use Hybrid-GSHF-DPO instead of DPOOpen-LLaMA-3B-V2 alignment · HH-RLHF testGPT4 Eval: 0.5 -> 0.65Helped
Use Hybrid-GSHF-DPO instead of DPOOpen-LLaMA-3B-V2 alignment · UltraFeedback OODOOD Gold Reward: 1.71 -> 2.46Helped
Use Multi-step RSO instead of RSOOpen-LLaMA-3B-V2 alignment · HH-RLHF testGold Reward: 2.25 -> 2.59Helped
Use Multi-step RSO instead of RSOOpen-LLaMA-3B-V2 alignment · UltraFeedback OODOOD Gold Reward: 1.89 -> 2.41Helped
Use Multi-step RSO instead of RSOOpen-LLaMA-3B-V2 alignment · UltraFeedback OODDifference Δ: 0.36 -> 0.18Helped
Use Hybrid-GSHF-DPO instead of Multi-step RSOOpen-LLaMA-3B-V2 alignment · HH-RLHF testGold Reward: 2.59 -> 2.67Helped
Use Hybrid-GSHF-DPO instead of Multi-step RSOOpen-LLaMA-3B-V2 alignment · UltraFeedback OODOOD GPT4 Eval: 0.60 -> 0.59No clear effect
Iterate Hybrid-GSHF-DPO with newly collected online dataHybrid-GSHF-DPO · HH-RLHF and UltraFeedbackGold reward-KL trade-offHelped
Use pseudo-labeled rejection-sampling data with RSO instead of training directly with DPOOpen-LLaMA-3B-V2 alignment · HH-RLHF testGold Reward: 2.15 -> 2.25Helped

Surprising findings

Limitations

Experimental setup

Datasets: Anthropic HH-RLHF UltraFeedback

Benchmarks: HH-RLHF test split UltraFeedback OOD prompts

Metrics: gold reward gold win rate GPT-4 win/tie/lose KL divergence

Baselines: DPO RSO SFT RAFT

Builds on: DPO RSO PPO InfoNCA

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