Master-RM数据集
收藏资源简介:
Master-RM数据集是腾讯AI实验室、普林斯顿大学和弗吉尼亚大学的研究人员为了提高生成性奖励模型在强化学习中的鲁棒性而创建的。该数据集包含180,000条训练数据,旨在训练一个对“master keys”攻击具有高度鲁棒性的生成性奖励模型。数据集由两部分组成:原始的160,000条有效训练数据和一个20,000条的合成负样本集。合成负样本集是通过截断模型输出来构造的,保留了每条回答的第一句话,这些话通常是通用的框架或高级推理引导语,类似于推理开放者。该数据集可用于训练新的生成性奖励模型,以提高其在不同数据集上的鲁棒性。
The Master-RM dataset was developed by researchers affiliated with Tencent AI Lab, Princeton University, and the University of Virginia, with the goal of enhancing the robustness of generative reward models in reinforcement learning. It contains 180,000 training samples, and is designed to train a generative reward model that demonstrates high robustness against "master keys" attacks. The dataset consists of two parts: 160,000 original valid training samples and a 20,000-sample synthetic negative set. The synthetic negative set is constructed by truncating model outputs, retaining only the first sentence of each response. These retained sentences are usually general frameworks or high-level reasoning guides, similar to reasoning openers. This dataset can be used to train new generative reward models to improve their robustness across different datasets.
数据集概述
基本信息
- 语言: 英文 (en)
- 许可证: Apache 2.0 (apache-2.0)
- 数据规模: 100K<n<1M
- 任务类别: 文本分类 (text-classification)
- 标签: RLVR, reward-modeling, preference-alignment
- 格式: JSON列表,使用chat-template格式,包含字段: query, output
数据集描述
该数据集包含用于训练鲁棒生成奖励模型的合成数据,旨在解决生成奖励模型中的漏洞问题。这些漏洞包括非单词符号或推理开头等表面操作可能导致错误的正奖励。
数据构建
- 基础数据: 基于160k实例的数据集,包含(q, a*, r, y)元组。
- 生成过程:
- 使用Qwen2.5-7B-base模型生成响应r。
- 使用Qwen2.5-72B-Instruct提供二进制信号y("YES"或"NO")判断r是否与a*对齐。
- 增强数据: 增加20k负例,通过GPT-4o-mini生成响应并截断为第一句(通常为通用、无解决方案的推理开头),标记为"NO"。
- 数据分布:
- 条目1–159,740: 原始160k数据点。
- 条目159,741–179,733: 增强示例。
快速开始
python from datasets import load_dataset ds = load_dataset("sarosavo/Master-RM") print(ds) print("lenth of reward training data:", len(ds[train]))
引用
bibtex @article{zhao2025one, title={One Token to Fool LLM-as-a-Judge}, author={Zhao, Yulai and Liu, Haolin and Yu, Dian and Kung, S.Y. and Mi, Haitao and Yu, Dong}, journal={arXiv preprint arXiv:2507.08794}, year={2025} }




