atlas

Research map / Large language models / Alignment and preferences

Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervision

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

Shows that strong GPT-4-family models finetuned on labels from much weaker models often outperform their supervisors, and that simple methods (auxiliary confidence loss, bootstrapping, generative finetuning) recover more of the gap.

Problem

Superalignment: can weak supervisors (humans or small models) elicit the full capabilities of much stronger models, when naive finetuning on weak labels imitates the weak supervisor's errors?

Approach

Study weak-to-strong learning as an empirical analogy by finetuning large pretrained GPT-4-family base models on labels produced by small weak models, measure performance gap recovered (PGR), and propose an auxiliary confidence loss that lets the strong student confidently disagree with weak labels.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add auxiliary confidence loss to weak-label finetuningGPT-4 student trained with GPT-2-level supervision · NLP tasksperformance gap recovered: about 25% to nearly 80%Helped
Add auxiliary confidence loss to weak-label finetuningStrong students trained with weak supervision · NLP tasksaccuracyMixed
Bootstrap supervision through intermediate model sizesStrong students trained with weak supervision · Chess puzzlesperformance gap recoveredHelped
Bootstrap supervision through intermediate model sizesStrong students trained with weak supervision · NLP tasksperformance gap recoveredNo clear effect
Bootstrap supervision through intermediate model sizesStrong students trained with weak supervision · ChatGPT reward modelingperformance gap recoveredNo clear effect
Add unsupervised generative finetuning on reward-modeling dataStrong students trained with weak supervision · ChatGPT reward modelingperformance gap recovered: approximately 10-20%Helped
Combine generative finetuning with ground-truth early stoppingStrong students trained with weak supervision · ChatGPT reward modelingperformance gap recovered: approximately 30-40%Helped
Finetune on weak labels, then train a linear probe on ground-truth labelsGPT-4 family model · NLP tasks, subset averageaccuracy: 78%Helped
Use LP-FT: train a linear probe, then finetune all layersStrong model trained with weak supervision · Weak-to-strong learningperformanceNo clear effect
Apply weight regularization with stronger weight decay or dropoutStrong model trained with weak supervision · Weak-to-strong learningperformanceNo clear effect
Apply LoRA and sweep the rankStrong model trained with weak supervision · Weak-to-strong learningperformanceNo clear effect
Add consistency loss between original and rephrased inputsStrong model trained with weak supervision · Selected NLP datasetsperformanceNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: BoolQ CosmosQA DREAM ETHICS FLAN ANLI R2 GLUE CoLA GLUE SST-2 HellaSwag MCTACO OpenBookQA PAWS QuAIL

Benchmarks: 22 NLP classification datasets chess puzzles (lichess) ChatGPT reward modeling test set

Metrics: test accuracy performance gap recovered (PGR) student-supervisor agreement

Baselines: naive finetuning on weak labels 5-shot prompting zero-shot prompting linear probing ground truth finetuning

Builds on: RLHF self-training conditional entropy minimization knowledge distillation

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