arXiv:2410.01257cs.LGcs.AI2024-10ICLR被引 145

对比两种奖励模型训练方法,提出融合新方案并实现顶尖性能。

HelpSteer2-Preference: Complementing Ratings with Preferences

  • 用偏好标注补全原评分数据,支持公平对比两种训练范式。
  • 融合方法使70B模型在RewardBench达94.1分,排名第一。
  • 开源数据与模型,适合对齐训练和强化学习研究者使用。

奖励模型对模型指令遵循至关重要,主流训练范式分为布拉德利-特瑞风格(Bradley-Terry)与回归风格(Regression)。但两者因数据格式不兼容,缺乏充分可比性。为此,我们为HelpSteer2数据集新增偏好标注(适配布拉德利-特瑞训练),并附人类撰写的理由以提升可解释性。基于此,首次在数据匹配条件下进行两类模型的直接对比。据此提出一种结合二者的新方法:使用该方法训练的Llama-3.1-70B-Instruct模型在RewardBench得分94.1,超越超过140个模型,位列第一(截至2024年10月1日)。再经REINFORCE算法(RLHF)微调后,在Arena Hard测试中达到85.0分,亦为当前最高分。数据集(CC-BY-4.0)已开源至HuggingFace,模型亦公开可用。

原文摘要 · Abstract (English)

Reward models are critical for aligning models to follow instructions, and are typically trained following one of two popular paradigms: Bradley-Terry style or Regression style. However, there is a lack of evidence that either approach is better than the other, when adequately matched for data. This is primarily because these approaches require data collected in different (but incompatible) formats, meaning that adequately matched data is not available in existing public datasets. To tackle this problem, we release preference annotations (designed for Bradley-Terry training) to complement existing ratings (designed for Regression style training) in the HelpSteer2 dataset. To improve data interpretability, preference annotations are accompanied with human-written justifications. Using this data, we conduct the first head-to-head comparison of Bradley-Terry and Regression models when adequately matched for data. Based on insights derived from such a comparison, we propose a novel approach to combine Bradley-Terry and Regression reward modeling. A Llama-3.1-70B-Instruct model tuned with this approach scores 94.1 on RewardBench, emerging top of more than 140 reward models as of 1 Oct 2024. This reward model can then be used with REINFORCE algorithm (RLHF) to align an Instruct model to reach 85.0 on Arena Hard, which is No. 1 as of 1 Oct 2024. We open-source this dataset (CC-BY-4.0 license) at https://huggingface.co/datasets/nvidia/HelpSteer2#preferences-new -- 1-oct-2024 and openly release the trained Reward and Instruct models at https://huggingface.co/nvidia/Llama-3.1-Nemotron-70B-Reward and https://huggingface.co/nvidia/Llama-3.1-Nemotron-70B-Instruct

奖励建模指令对齐强化学习数据开源

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