Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

Widely used alignment techniques, such as reinforcement learning from human feedback (RLHF), rely on the ability of humans to supervise model behavior - for example, to evaluate whether a model faithfully followed instructions or generated safe outputs. However, future superhuman models will behave in complex ways too difficult for humans to reliably evaluate; humans will only be able to weakly supervise superhuman models. We study an analogy to this problem: can weak model supervision elicit the full capabilities of a much stronger model? We test this using a range of pretrained language models in the GPT-4 family on natural language processing (NLP), chess, and reward modeling tasks. We find that when we naively finetune strong pretrained models on labels generated by a weak model, they consistently perform better than their weak supervisors, a phenomenon we call weak-to-strong generalization. However, we are still far from recovering the full capabilities of strong models with naive finetuning alone, suggesting that techniques like RLHF may scale poorly to superhuman models without further work. We find that simple methods can often significantly improve weak-to-strong generalization: for example, when finetuning GPT-4 with a GPT-2-level supervisor and an auxiliary confidence loss, we can recover close to GPT-3.5-level performance on NLP tasks. Our results suggest that it is feasible to make empirical progress today on a fundamental challenge of aligning superhuman models.

Collin Burns, Pavel Izmailov, Jan Hendrik Kirchner, Bowen Baker, Leo Gao, Leopold Aschenbrenner, Yining Chen, Adrien Ecoffet, Manas Joglekar, Jan Leike, Ilya Sutskever, Jeff Wu• 2023

Related benchmarks

TaskDatasetResultRank
Reward Model TransferHelpSteer3 (H3) v1 (test)
AOG6.78
16
Reward Model TransferUltraFeedback (UF)
AOG7.64
16
Reward Model TransferAnthropic Helpful (AH)
AOG8.06
16
Reward Model TransferRAIL
AOG2.26
8
Reward Model TransferAnthropic Harmless (AHar)
AOG2.44
8
Reward Model TransferPKU-SafeRLHF
AOG2.05
8
Dialogue SummarizationSAMSum Multiple Client (test)
ROUGE-1 (Client 1)49.99
6
Mathematical ReasoningGSM8K Single Client
Accuracy74.3
6
Commonsense ReasoningCommonsenseQA Single Client
Accuracy83.46
6
Commonsense ReasoningCommonsenseQA Multiple Client (test)
Client 1 Accuracy78.57
6
Showing 10 of 16 rows

Other info

Follow for update