atlas

Research map / Large language models / Alignment and preferences

RLAIF vs. RLHF: Scaling Reinforcement Learning from Human Feedback with AI Feedback

ICML 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 69 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 RLAIF (LLM-labeled preferences distilled into a reward model) with RLHF on summarization and dialogue tasks, and introduces d-RLAIF, which uses an off-the-shelf LLM's scores directly as RL reward.

Problem

Human preference labels for RLHF are expensive and hard to scale; it was unknown whether LLM-generated preference labels (RLAIF) can match human feedback for aligning LLM policies.

Approach

Label preferences with an off-the-shelf PaLM 2 LLM using scored log-probabilities of "1"/"2" with position debiasing and optional chain-of-thought, train a reward model on those soft labels (canonical RLAIF), or skip RM training and let the LLM score each generation 1-10 directly as the RL reward (d-RLAIF).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace human preference labels with AI preference labels for reward-model trainingRLAIF vs. RLHF · Reddit TL;DR and helpful dialoguehuman win rate vs. SFT: 71% vs 73%; 63% vs 64%No clear effect
Replace human preference labels with AI preference labels for reward-model trainingHarmless dialogue policy · Harmless dialogue testharmless rate: 76% -> 88%Helped
Use PaLM 2 XS as the AI labeler instead of a larger modelSame-size RLAIF · Reddit TL;DR testwin rate vs. SFT: 68%Helped
Use direct LLM rewards during RL instead of distilling AI preferences into a reward modeld-RLAIF · Reddit TL;DR testwin rate vs. Same-size RLAIF: 60%Helped
Use direct LLM rewards during RLd-RLAIF · Helpful dialogue testwin rate vs. SFT: 66%Helped
Add chain-of-thought reasoning to the base zero-shot prompt (ablation)AI preference labeler · Summarization, helpful dialogue, and harmless dialogueAI labeler alignment: 76.1% -> 77.5%; 67.8% -> 69.1%; 69.4% -> 70.6%Helped
Use a detailed preamble instead of a base preamble (ablation)AI preference labeler · SummarizationAI labeler alignment: 76.1% -> 77.4%Helped
Use a detailed preamble instead of a base preamble (ablation)AI preference labeler · Helpful and harmless dialogueAI labeler alignment: 67.8% -> 67.6%; 69.4% -> 70.1%Mixed
Add in-context exemplars (ablation)AI preference labeler · Summarization and helpful dialogueAI labeler alignment: 76.1% -> 76.0% -> 75.7%; 67.8% -> 67.1% -> 66.8%Hurt
Add in-context exemplars (ablation)AI preference labeler · Harmless dialogueAI labeler alignment: 69.4% -> 71.7% -> 72.1%Helped
Increase the AI labeler size (ablation)AI preference labeler · Summarization preference holdoutAI labeler alignment: 62.7% -> 73.8% -> 78.0%Helped
Combine human and AI feedback for reward-model training (ablation)RLHF + RLAIF · Reddit TL;DR testwin rate vs. SFT: 74% -> 71%Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Reddit TL;DR OpenAI Human Preferences Anthropic Helpful and Harmless human preferences

Benchmarks: Reddit TL;DR Anthropic HH helpful test Anthropic HH harmless test

Metrics: win rate harmless rate AI labeler alignment pairwise reward model accuracy length-controlled win rate

Baselines: RLHF SFT same-size RLAIF canonical RLAIF

Builds on: RLHF Constitutional AI REINFORCE Reddit TL;DR

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