atlas

Research map / Large language models / Alignment and preferences

Self-Rewarding Language Models

ICML 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 10 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.

Proposes Self-Rewarding Language Models that use the model's own LLM-as-a-Judge scores within an iterative DPO loop, improving both instruction following and reward modeling from a Llama 2 70B seed model.

Problem

Preference-based alignment is bottlenecked by human preference data and by frozen reward models that cannot improve during LLM training.

Approach

A single model both generates instruction-following responses and judges them via LLM-as-a-Judge prompting; the self-generated preference pairs are used for iterative DPO, so the reward-modeling ability improves alongside instruction following.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
iterate self-rewarding DPO training from M1 to M3Llama 2 70B Self-Rewarding · AlpacaEval 2.0win rate: 9.94% -> 20.44%Helped
iterate self-rewarding DPO training from M1 to M3Llama 2 70B Self-Rewarding · MT-Benchoverall score: 6.78 ... 7.25Helped
train with AI Feedback Training preference pairs from the previous iterationSelf-Rewarding M1 · Open Assistant reward evaluationpairwise accuracy: 78.7% -> 80.4%Helped
continue iterative self-reward training from M2 to M3Self-Rewarding M2 · Open Assistant reward evaluationpairwise accuracy: 80.4% -> 81.7%Helped
add Evaluation Fine-Tuning (EFT) data (ablation)IFT-only SFT Baseline · Open Assistant reward evaluationpairwise accuracy: 65.1% -> 78.7%Helped
add EFT data to seed training (ablation)IFT-only instruction-following model · IFT test sethead-to-head win rate: 30.9% vs. 30.5%No clear effect
replace the additive score-counting judge prompt with a multiple-choice prompt (ablation)Llama 2 70B SFT Baseline · Open Assistant reward evaluationpairwise accuracy: 65.1% -> 26.6%Hurt
add only self-scored perfect-score examples instead of preference pairs (ablation)SFT Baseline · IFT test sethead-to-head win rate: 29% wins vs 30% winsNo clear effect
use IFT+EFT seed training rather than IFT-only seed training (ablation)Self-Rewarding training initialization · IFT test sethead-to-head win rate: 30.5% wins vs. 30.9% winsNo clear effect
fine-tune with successive self-generated preference dataLlama 2 70B · GSM8Kaccuracy (em): 56.80 -> 60.27 -> 59.29 -> 57.70Mixed

Surprising findings

Limitations

Experimental setup

Datasets: Open Assistant AlpacaEval MT-Bench ARC HellaSwag GSM8K MMLU NQ

Benchmarks: AlpacaEval 2.0 MT-Bench ARC HellaSwag GSM8K MMLU NQ Open Assistant held-out ranking set

Metrics: win rate pairwise accuracy exact match Spearman correlation Kendall tau MT-Bench score

Baselines: SFT Baseline Claude 2 Gemini Pro GPT-4 0613 GPT-4 0314 Mistral Medium PCO SPIN

Builds on: DPO Iterative DPO Self-Instruct Pairwise Cringe Optimization Constitutional AI

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