通过人类反馈的强化学习训练有益且无害的助手

Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback

白云涛 Yuntao Bai · Anthropic · 2022-04-12 · arXiv:2204.05862 ↗ · 被引 4175

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

摘要 · Abstract

我们应用偏好建模和基于人类反馈的强化学习(RLHF)来微调语言模型,使其成为有益且无害的助手。我们发现这种对齐训练几乎在所有 NLP 评估中都能提升性能,并且与 Python 编程和摘要等专业技能的训练完全兼容。我们探索了一种迭代在线训练模式,其中偏好模型和 RL 策略每周根据新的人类反馈数据进行更新,从而高效地改进我们的数据集和模型。最后,我们研究了 RLHF 训练的鲁棒性,并发现 RL 奖励与策略及其初始化之间的 KL 散度的平方根大致呈线性关系。除了主要结果外,我们还进行了关于校准、竞争目标和 OOD 检测使用的辅助分析,将我们的模型与人类写手进行比较,并提供了使用近期相关工作提示的模型样本。

We apply preference modeling and reinforcement learning from human feedback (RLHF) to finetune language models to act as helpful and harmless assistants. We find this alignment training improves performance on almost all NLP evaluations, and is fully compatible with training for specialized skills such as python coding and summarization. We explore an iterated online mode of training, where preference models and RL policies are updated on a weekly cadence with fresh human feedback data, efficiently improving our datasets and models. Finally, we investigate the robustness of RLHF training, and identify a roughly linear relation between the RL reward and the square root of the KL divergence between the policy and its initialization. Alongside our main results, we perform peripheral analyses on calibration, competing objectives, and the use of OOD detection, compare our models with human writers, and provide samples from our models using prompts appearing in recent related work.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 55)

阅读逐段中英对照全文 →