atlas

Research map / Large language models / Alignment and preferences

Chain of Hindsight aligns Language Models with Feedback

ICLR 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 27 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 Chain of Hindsight (CoH), which fine-tunes language models on sequences of model outputs paired with templated natural-language feedback (good/bad) so they learn from all feedback without RL.

Problem

Aligning language models with human preferences via SFT wastes negatively rated data, while RLHF relies on imperfect reward models and difficult RL optimization.

Approach

Convert all human preference rankings into templated hindsight feedback sequences (e.g., "Bad: {worse} Good: {better}") and fine-tune with the standard causal LM objective, with loss only on non-feedback tokens, so the model learns from positive and negative examples and generates the good behavior when prompted with "Good".

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Chain of Hindsight instead of RLHFLanguage models on TL;DR summarization · TL;DR summarizationhuman evaluation average win rate (%): 30.8 -> 57.5Helped
Use Chain of Hindsight instead of RLHFLanguage models on HH dialogue · HH dialoguehuman evaluation average win rate (%): 23.4 -> 49.5Helped
Add natural language feedback (ablation)Chain of Hindsight · TL;DR summarizationhuman preference win rate (%): 10.6 -> 15.1Helped
Add natural language feedback (ablation)Chain of Hindsight · TL;DR summarizationhuman preference win rate (%) against RLHF: 42.4 -> 45.3Helped
Add unlikelihood loss on negative examplesSFT on HH dialogue · HH dialoguepreferred dialogue classification accuracyHurt
Increase model sizeChain of Hindsight on summarization · Summarization benchmarkperformanceMixed
Condition training on multiple feedback-example pairs rather than only binary feedback (ablation)Chain of Hindsight · TL;DR summarizationhuman evaluation preferenceHelped

Surprising findings

Limitations

Experimental setup

Datasets: WebGPT comparisons Anthropic HH-RLHF OpenAI summarize_from_feedback TL;DR Reddit ShareGPT The Pile

Benchmarks: TL;DR summarization (ROUGE and human evaluation) Anthropic HH dialogue preference classification few-shot NLP benchmarks (Language Model Evaluation Harness)

Metrics: ROUGE classification accuracy human pairwise win rate (%) few-shot accuracy

Baselines: SFT SFT-U C-SFT RLHF Koala GPT-J

Builds on: RLHF Decision Transformer unlikelihood training hindsight experience replay

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