使用人类反馈训练语言模型遵循指令

Training language models to follow instructions with human feedback

扬·莱克 Jan Leike · OpenAI · 2022-03-04 · arXiv:2203.02155 ↗ · 被引 21784

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

增大语言模型的规模并不天然使其更擅长遵循用户意图。例如,大型语言模型可能生成不真实、有害或对用户无帮助的输出。换句话说,这些模型与用户不一致。在本文中,我们展示了一种通过人类反馈微调来使语言模型在广泛任务上与用户意图对齐的途径。从一组标注者编写的提示和通过 OpenAI API 提交的提示开始,我们收集了标注者展示期望模型行为的数据集,用于通过监督学习微调 GPT-3。然后,我们收集了模型输出排名的数据集,用于通过来自人类反馈的强化学习进一步微调这个监督模型。我们将得到的模型称为 InstructGPT。在我们的提示分布上进行的人类评估中,1.3B 参数的 InstructGPT 模型的输出优于 175B 参数的 GPT-3,尽管参数数量少 100 倍。此外,InstructGPT 模型在真实性和减少有毒输出生成方面表现出改进,同时在公共 NLP 数据集上性能退化最小。尽管 InstructGPT 仍然会犯简单错误,但我们的结果表明,使用人类反馈进行微调是使语言模型与人类意图对齐的一个有前景的方向。

Making language models bigger does not inherently make them better at following a user's intent. For example, large language models can generate outputs that are untruthful, toxic, or simply not helpful to the user. In other words, these models are not aligned with their users. In this paper, we show an avenue for aligning language models with user intent on a wide range of tasks by fine-tuning with human feedback. Starting with a set of labeler-written prompts and prompts submitted through the OpenAI API, we collect a dataset of labeler demonstrations of the desired model behavior, which we use to fine-tune GPT-3 using supervised learning. We then collect a dataset of rankings of model outputs, which we use to further fine-tune this supervised model using reinforcement learning from human feedback. We call the resulting models InstructGPT. In human evaluations on our prompt distribution, outputs from the 1.3B parameter InstructGPT model are preferred to outputs from the 175B GPT-3, despite having 100x fewer parameters. Moreover, InstructGPT models show improvements in truthfulness and reductions in toxic output generation while having minimal performance regressions on public NLP datasets. Even though InstructGPT still makes simple mistakes, our results show that fine-tuning with human feedback is a promising direction for aligning language models with human intent.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 19)

阅读逐段中英对照全文 →