five

allenai/reward-bench

收藏
Hugging Face2024-06-10 更新2024-03-29 收录
下载链接:
https://hf-mirror.com/datasets/allenai/reward-bench
下载链接
链接失效反馈
官方服务:
资源简介:
RewardBench评估数据集用于评估奖励模型在多个类别中的能力,包括聊天、聊天难度、安全性和推理等。数据集包含多个子集,每个子集都有特定的用途和描述。数据集的结构包括提示、选择的响应、拒绝的响应、模型信息、子集信息和唯一ID。数据集还提供了过滤和子集选择的详细说明,以及数据集的许可证信息。

RewardBench is an evaluation dataset designed to assess the capabilities of reward models across multiple categories, including chat, chat difficulty, safety, and reasoning. The dataset comprises multiple subsets, each with specific use cases and descriptions. Its structure includes prompts, selected responses, rejected responses, model information, subset information, and unique IDs. The dataset also provides detailed guidelines for filtering and subset selection, along with its license information.
提供机构:
allenai
原始信息汇总

数据集概述

数据集基本信息

  • 名称: RM Bench
  • 语言: 英语 (en)
  • 许可证: ODC-BY
  • 大小: 1K<n<10K
  • 任务类别: 问答 (question-answering)

数据集结构

  • 特征:
    • prompt (字符串): 各种测试集中的指令。
    • chosen (字符串): 来自更好模型的响应或更高评分的提示。
    • chosen_model (字符串): 适用时。
    • rejected (字符串): 得分较低的响应或来自较差模型的响应。
    • rejected_model (字符串): 适用时。
    • subset (字符串): 关联提示的子集(例如 alpacaeval-easy)。
    • id (整数): 基准中每个提示的递增ID。

数据集拆分

  • 训练集: 5123个样本,10853788字节
  • 过滤后: 2985个样本,4861303字节
  • 下载大小: 7957019字节
  • 数据集大小: 15715091字节

数据集内容

  • 评估类别:
    1. 聊天: 包括简单聊天子集(alpacaeval-easy, alpacaeval-length, alpacaeval-hard, mt-bench-easy, mt-bench-medium)
    2. 聊天困难: 包括困难聊天子集(mt-bench-hard, llmbar-natural, llmbar-adver-neighbor, llmbar-adver-GPTInst, llmbar-adver-GPTOut, llmbar-adver-manual)
    3. 安全性: 包括安全性子集(refusals-dangerous, refusals-offensive, xstest-should-refuse, xstest-should-respond, do not answer)
    4. 推理: 包括代码和数学子集(math-prm, hep-cpp, hep-go, hep-java, hep-js, hep-python, hep-rust)

许可证信息

  • 数据集许可证: ODC-BY
  • 子集许可证:
    • AlpacaEval: CC By NC 4.0
    • MT Bench: Apache 2.0
    • LLMBar: MIT License
    • Do Not Answer: CC BY NC SA 4.0
    • XSTest: CC By 4.0
    • HumanEvalPack: MIT License
    • PRM Math: MIT License

数据集开发

  • 构建要求: datasets, notebook, nbconvert
  • 构建命令: python build_dataset.py

此概述提供了RM Bench数据集的关键信息,包括其结构、内容、许可证和开发要求。

搜集汇总
数据集介绍
main_image_url
构建方式
RewardBench数据集由Allen Institute for AI构建,旨在系统评估奖励模型在对话、安全与推理等多维度的表现。其构建过程融合了AlpacaEval、MT-Bench、LLMBar、XSTest、DoNotAnswer、HumanEvalPack及PRM Math等现有基准测试的提示与响应对,通过人工筛选与自动过滤相结合的方式,从5123条原始提示中精炼出2985条高质量样本。每条样本包含提示文本、优选与拒选响应及其来源模型标识、所属子集与唯一编号,形成了覆盖易难对话、安全拒绝与代码数学推理等23个细分场景的结构化评估集。
特点
该数据集的核心特点在于其多维度的分层架构与精细化的质量管控。它横跨聊天、硬核聊天、安全与推理四大领域,每个领域内又细分为多个子集,如聊天类涵盖AlpacaEval的简单、长度与困难变体,安全类则聚焦危险与冒犯性拒绝场景。数据经过严格的人工验证,确保优选与拒选响应的排序可靠性,同时提供过滤前后版本以支持不同研究需求。此外,数据集记录了响应长度、模型来源等元信息,为分析奖励模型的行为偏差提供了丰富的统计基础。
使用方法
研究人员可通过HuggingFace Datasets库便捷加载RewardBench数据集,默认使用'filtered'分割以获取高质量样本。若要聚焦特定子集,可使用.filter()方法按'subset'字段筛选,例如仅分析AlpacaEval简单子集。评估时需计算奖励模型对每条提示的优选与拒选响应得分,并比较两者高低以判定模型偏好。最终评分采用加权平均策略,对各子集得分进行均衡聚合,其中推理领域通过提升PRM-Math子集权重来平衡代码与数学能力的影响,从而生成一个综合性的奖励模型性能基准分数。
背景与挑战
背景概述
在大型语言模型(LLM)飞速演进的浪潮中,奖励模型(Reward Model)作为基于人类反馈的强化学习(RLHF)框架的核心组件,其评估能力直接决定了模型对齐的质量与安全性。2024年,艾伦人工智能研究所(AI2)的研究团队推出了RewardBench数据集,旨在系统性地评测奖励模型在对话、安全性、推理及复杂聊天等多元场景下的表现。该数据集整合了AlpacaEval、MT-Bench、LLMBar、XSTest等十余个权威子集,覆盖从简单指令遵循到对抗性攻击防御的广泛维度,共计2985条经过人工与自动双重校验的提示-响应对。作为首个全面聚焦奖励模型能力的标准化基准,RewardBench不仅为模型开发者提供了统一的性能标尺,更深刻揭示了当前奖励模型在长尾场景与细微语义判别上的局限性,对推动RLHF技术的透明化与鲁棒性具有里程碑式的影响。
当前挑战
RewardBench所面临的挑战根植于奖励模型评估的深层复杂性。首先,在领域问题层面,如何精准量化奖励模型在‘聊天硬’(Chat Hard)与‘推理’(Reasoning)等高端任务中的泛化能力是一大难题,尤其是当模型需要区分语法正确但逻辑谬误的代码或数学推导时,现有的胜率指标往往掩盖了细粒度的失败模式。其次,在构建过程中,数据集的制作遭遇了多重障碍:不同来源子集(如AlpacaEval与LLMBar)的标签标准与噪声分布迥异,人工筛选5123条原始提示至2985条有效样本的过程需耗费大量人力以纠正自动标注的偏差;此外,对抗性样本(如LLMBar-adver-GPTInst)的设计需巧妙平衡攻击强度与语义自然度,避免引入人为伪影。更棘手的是,安全子集(如refusals-dangerous)中‘拒绝’与‘回应’的边界模糊,模型对微妙语境的误判可能导致灾难性后果,而现有基准尚无法完全覆盖这类高风险边缘案例。
常用场景
经典使用场景
在奖励模型评估领域,RewardBench数据集扮演着标准测试基准的角色。该数据集精心设计了涵盖对话、安全与推理四大维度的评测子集,如Chat Hard与Safety类别,旨在系统性地衡量奖励模型在不同难度与风险场景下的判别能力。研究者通过对比模型对chosen与rejected响应的评分正确率,获得一个综合性的RewardBench分数,从而精准定位模型在偏好对齐方面的优势与短板。这种结构化的评估范式为奖励模型的横向对比提供了可靠依据,已成为该领域广泛采纳的验证协议。
实际应用
在实际应用中,RewardBench数据集为大规模语言模型的训练与部署提供了关键的质检工具。开发团队可利用该基准在模型发布前评估其奖励模型的安全性,例如通过Safety子集检测模型是否对危险或冒犯性内容做出正确拒绝。在对话系统优化中,Chat Hard子集帮助识别模型在复杂指令或对抗性输入下的失效模式。此外,该数据集还可用于监控模型迭代过程中的性能回归,确保奖励模型在提升某一维度能力时不牺牲其他维度的表现,从而在学术研究与工业落地之间架起了一座桥梁。
衍生相关工作
RewardBench数据集的发布催生了一系列重要的衍生工作。其开源的评测代码与排行榜激励了社区开发更先进的奖励模型,如基于该基准改进的SteerLM与Direct Preference Optimization变体。研究者利用其子集分析奖励模型的长度偏好与虚假相关性,催生了关于奖励黑客现象的深入探讨。此外,该数据集还启发了跨领域评测集的构建,如将类似方法论应用于多模态奖励模型的评估。这些衍生工作不仅丰富了偏好对齐的理论工具箱,也推动了奖励模型从单一评分器向更可靠、更可解释的评估系统演进。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务