allenai/reward-bench

RewardBench: the first evaluation tool for reward models.

它解决了什么问题

RewardBench 提供了一套标准化的方式,用来评估奖励模型(RMs)的能力与安全性,这些模型用于让 LLM 与人类偏好对齐。它解决了这些模型缺乏一致基准测试的问题,包括通过直接偏好优化(DPO)训练的模型,以及生成式“LLM-as-a-judge”模型。

它如何运作

该项目提供了一套工具与数据集,用来测试奖励模型区分“chosen”与“rejected”响应的能力。它包括:

  • 推理代码:用于运行各种奖励模型的统一代码(例如 Starling、PairRM、OpenAssistant)。
  • 评估脚本:针对标准奖励模型(run_rm.py)、DPO 模型(run_dpo.py)以及更新版 RewardBench 2(run_v2.py)的专用脚本。
  • 生成式支持:通过本地 vLLM 推理或 API(OpenAI、Anthropic、Google Gemini、Together)将 LLM 用作评审的工具。
  • 分析工具:用于可视化结果、计算各分区分数(Chat、Safety、Reasoning),以及测试奖励模型集成(ensembles)的实用工具。

适合谁使用

  • AI 研究人员:开发与测试新的奖励模型或偏好优化技术。
  • ML 工程师:在将奖励模型部署到 RLHF 流程之前,评估其安全性与质量。
  • 模型开发者:将自己的模型与公开排行榜进行比较,以衡量性能。

亮点

  • 广泛的模型支持:兼容序列分类器、DPO/KTO 模型,以及生成式评审。
  • RewardBench 2:包含针对“best-of-4”与“Ties”数据集的高级数据处理。
  • 灵活部署:支持通过 uv 或 pip 安装,并提供 Docker 镜像以实现可复现的研究。
  • 集成:允许将评估结果与模型元数据直接推送至 Hugging Face Hub。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目