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

Abstract

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.

Cite

Text

Burns et al. "Weak-to-Strong Generalization: Eliciting Strong Capabilities with Weak Supervision." International Conference on Machine Learning, 2024.

Markdown

[Burns et al. "Weak-to-Strong Generalization: Eliciting Strong Capabilities with Weak Supervision." International Conference on Machine Learning, 2024.](https://mlanthology.org/icml/2024/burns2024icml-weaktostrong/)

BibTeX

@inproceedings{burns2024icml-weaktostrong,
  title     = {{Weak-to-Strong Generalization: Eliciting Strong Capabilities with Weak Supervision}},
  author    = {Burns, Collin and Izmailov, Pavel and Kirchner, Jan Hendrik and Baker, Bowen and Gao, Leo and Aschenbrenner, Leopold and Chen, Yining and Ecoffet, Adrien and Joglekar, Manas and Leike, Jan and Sutskever, Ilya and Wu, Jeffrey},
  booktitle = {International Conference on Machine Learning},
  year      = {2024},
  pages     = {4971-5012},
  volume    = {235},
  url       = {https://mlanthology.org/icml/2024/burns2024icml-weaktostrong/}
}