atlas

Research map / Large language models / Alignment and preferences

Language Model Self-improvement by Reinforcement Learning Contemplation

ICLR 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 2 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 SIRLC, an unsupervised method where an LLM acts as both student (answers unlabeled questions) and teacher (self-evaluates answers), with PPO optimizing the self-evaluation reward; applied to FLAN-T5 on BigBench-hard, IWSLT 2017 and CNN/DailyMail.

Problem

Fine-tuning LLMs usually requires labeled data or human feedback; the paper asks whether an LLM can improve itself using only its own evaluations of generated text, without external labels or reward models.

Approach

Exploit the observation that LLMs are better at evaluating text than generating it: the same frozen initial LLM scores its own answers (correctness prompt for reasoning, 1-10 quality prompt for translation/summarization), and the score is used as an RL reward via PPO to update the policy.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
RL training with self-evaluation reward (SIRLC) versus direct generationFLAN-T5-Large on BigBench-hard · BigBench-hard 12 tasksanswer accuracy: +5.6% average accuracy over DGHelped
RL training with self-evaluation reward (SIRLC) versus direct generationFLAN-T5-Large on translation · IWSLT 2017 zh-enBERTScore: 0.818 to 0.86Helped
RL training with self-evaluation reward (SIRLC) versus direct generationFLAN-T5-Large on summarization · CNN/Daily MailBERTScore: 0.886 to 0.899Helped
Training with self-evaluation reward vs. oracle-label RL (RLFT)FLAN-T5-Large on BigBench-hard · Web of Lies, Logical Deduction (7), Navigate, Penguinsanswer accuracy: e.g. Web of Lies 52.9% vs RLFT 72.2%Hurt
Self-evaluation (check answer and regenerate) at inference, no training (ablation)FLAN-T5-Large on BigBench-hard · 27 BigBench tasks averageanswer accuracy: 31.6% to 32.9%Helped
Training on five BigBench tasks then testing on unseen tasks (ablation)FLAN-T5-Large SIRLC-trained model · unseen BigBench tasksaverage answer accuracy: +0.8 average; +1.5, +0.7, +2.2 on three tasks; -0.4 and -0.1 on twoMixed
Self-evaluation reward size scaling (FLAN-T5 Small/Base/Large) (ablation)FLAN-T5 80M to 780M · three BigBench tasksanswer accuracy: qualitative, largest gain at 80MHelped

Surprising findings

Limitations

Experimental setup

Datasets: BigBench-hard IWSLT 2017 CNN/Daily Mail BBC CommonGen

Benchmarks: BigBench-hard (12 tasks) IWSLT 2017 (zh-en) CNN/Daily Mail BBC

Metrics: answer accuracy BERTScore BLEU ROUGE

Baselines: DG (direct generation) Self-consistency (SC) LMSI RLFT (oracle-metric RL)

Builds on: PPO self-consistency LMSI chain-of-thought prompting trlx

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