arXiv:2410.12832cs.LG2024-10被引 119

用AI自动生成更符合人类偏好的奖励标签,提升大模型训练效果。

Generative Reward Models

  • 让大模型自我生成推理过程,生成与人类偏好一致的合成标签。
  • 在分布内任务上达到与人工标注相当的准确率,在分布外任务上提升10%-45%。
  • 适合需要高质量合成数据的大模型训练场景,尤其关注泛化能力的开发者。

基于人类反馈的强化学习(RLHF)显著提升了现代大语言模型(LLMs)的性能,但其依赖大量人工偏好标注,成本高昂且技术复杂。基于AI反馈的强化学习(RLAIF)通过大模型生成合成偏好来缓解此问题,但现有研究表明合成标签与人类判断存在偏差。为此,我们提出一种融合RLHF与RLAIF的方法——GenRM,一种迭代算法,使大模型在自生成推理轨迹上进行训练,从而生成更符合人类偏好的合成偏好标签。实验证明,零样本大模型判断在分布内任务上的表现低于布拉德利-特里(Bradley-Terry)奖励模型9%-36%;而GenRM在分布内任务上达到与布拉德利-特里模型相当的准确率,并在分布外任务上显著超越其10%-45%。此外,GenRM在分布内任务上优于直接使用大模型作为评判者9%-31%,在分布外任务上也高出2%-6%。结果表明,结合两种方法的优势可有效提升合成偏好标签质量。

原文摘要 · Abstract (English)

Reinforcement Learning from Human Feedback (RLHF) has greatly improved the performance of modern Large Language Models (LLMs). The RLHF process is resource-intensive and technically challenging, generally requiring a large collection of human preference labels over model-generated outputs. Reinforcement Learning from AI Feedback (RLAIF) addresses this data collection challenge by leveraging synthetic preferences generated by an LLM. However, recent work has shown that synthetic preferences labels may not align well with human preference judgments. To address this, we propose a hybrid approach that unifies RLHF and RLAIF methodologies. We introduce GenRM, an iterative algorithm that trains an LLM on self-generated reasoning traces, leading to synthetic preference labels matching human preference judgments. Empirically, we show that zero-shot LLM-based judgments under-perform compared to Bradley-Terry reward models on in-distribution tasks (between 9-36%). In contrast, GenRM achieves in-distribution accuracy comparable to Bradley-Terry models, while significantly outperforming them on out-of-distribution tasks (between 10-45%). Moreover, GenRM surpasses the performance of using LLMs as judges on both in-distribution (by 9-31%) and out-of-distribution tasks (by 2- 6%). Our results show that combining the strengths of RLHF and RLAIF offers a promising approach for improving the quality of synthetic preference labels.

强化学习大模型奖励建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。