arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.08673cs.CLcs.AIcs.LG

HelpSteer2:用于训练顶尖性能奖励模型的开源数据集

HelpSteer2: Open-source dataset for training top-performing reward models

NVIDIA(英伟达)

Zhilin Wang, Yi Dong, Olivier Delalleau, Jiaqi Zeng, Gerald Shen, Daniel Egert, Jimmy J. Zhang, Makesh Narsimhan Sreedhar, Oleksii Kuchaiev

AI总结:

本文发布CC-BY-4.0许可的HelpSteer2偏好数据集,仅用一万个回答对训练出在Reward-Bench达到92.0% SOTA的奖励模型,并提出利用多属性分数的SteerLM 2.0对齐方法。

摘要:

高质量偏好数据集对于训练奖励模型至关重要,这些模型能够有效引导大型语言模型(LLM)生成符合人类偏好的高质量回答。随着LLM变得更强大且对齐更完善,Open Assistant、HH-RLHF和HelpSteer等采用宽松许可的偏好数据集需要更新,才能在奖励建模中继续保持有效性。从GPT-4等专有LLM蒸馏偏好数据的方法会受到模型提供商施加的商业使用限制。为同时改进生成回答和属性标注质量,我们发布HelpSteer2,这是一个采用CC-BY-4.0宽松许可的偏好数据集。使用在HelpSteer2上训练的强大内部基础模型,我们得以在Reward-Bench的主要数据集上取得SOTA分数(92.0%),截至2024年6月12日优于当前列出的开放模型和专有模型。值得注意的是,HelpSteer2仅包含一万个回答对,比HH-RLHF等现有偏好数据集少一个数量级,因此用于训练奖励模型时非常高效。大量实验表明,使用HelpSteer2训练的奖励模型可有效对齐LLM。特别是,我们提出SteerLM 2.0这一模型对齐方法,能够有效利用奖励模型预测的丰富多属性分数。HelpSteer2可在https://huggingface.co/datasets/nvidia/HelpSteer2获取,代码可在https://github.com/NVIDIA/NeMo-Aligner获取。

英文摘要:

High-quality preference datasets are essential for training reward models that can effectively guide large language models (LLMs) in generating high-quality responses aligned with human preferences. As LLMs become stronger and better aligned, permissively licensed preference datasets, such as Open Assistant, HH-RLHF, and HelpSteer need to be updated to remain effective for reward modeling. Methods that distil preference data from proprietary LLMs such as GPT-4 have restrictions on commercial usage imposed by model providers. To improve upon both generated responses and attribute labeling quality, we release HelpSteer2, a permissively licensed preference dataset (CC-BY-4.0). Using a powerful internal base model trained on HelpSteer2, we are able to achieve the SOTA score (92.0%) on Reward-Bench's primary dataset, outperforming currently listed open and proprietary models, as of June 12th, 2024. Notably, HelpSteer2 consists of only ten thousand response pairs, an order of magnitude fewer than existing preference datasets (e.g., HH-RLHF), which makes it highly efficient for training reward models. Our extensive experiments demonstrate that reward models trained with HelpSteer2 are effective in aligning LLMs. In particular, we propose SteerLM 2.0, a model alignment approach that can effectively make use of the rich multi-attribute score predicted by our reward models. HelpSteer2 is available at https://huggingface.co/datasets/nvidia/HelpSteer2 and code is available at https://github.com/NVIDIA/NeMo-Aligner

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑