five

YYYYYYibo/ultrafeedback_binarized_dataset_offline_pairrm_part2_mini_1

收藏
Hugging Face2024-05-06 更新2024-06-12 收录
下载链接:
https://hf-mirror.com/datasets/YYYYYYibo/ultrafeedback_binarized_dataset_offline_pairrm_part2_mini_1
下载链接
链接失效反馈
官方服务:
资源简介:
--- dataset_info: features: - name: prompt dtype: string - name: prompt_id dtype: string - name: messages list: - name: content dtype: string - name: role dtype: string - name: score_chosen dtype: float64 - name: score_rejected dtype: float64 - name: reference_response dtype: string - name: chosen list: - name: content dtype: string - name: role dtype: string - name: rejected list: - name: content dtype: string - name: role dtype: string splits: - name: train_prefs num_bytes: 41380967 num_examples: 5183 download_size: 23152573 dataset_size: 41380967 configs: - config_name: default data_files: - split: train_prefs path: data/train_prefs-* --- # Dataset Card for "ultrafeedback_binarized_dataset_offline_pairrm_part2_mini_1" [More Information needed](https://github.com/huggingface/datasets/blob/main/CONTRIBUTING.md#how-to-contribute-to-the-dataset-cards)

--- # 「ultrafeedback_binarized_dataset_offline_pairrm_part2_mini_1」数据集卡片 [需补充更多信息](https://github.com/huggingface/datasets/blob/main/CONTRIBUTING.md#how-to-contribute-to-the-dataset-cards) ## 数据集信息 ### 特征字段 1. 提示词(prompt):数据类型为字符串(string) 2. 提示词ID(prompt_id):数据类型为字符串(string) 3. 对话消息(messages):列表类型,包含以下子字段: - 内容(content):数据类型为字符串(string) - 角色(role):数据类型为字符串(string) 4. 优选回复得分(score_chosen):数据类型为64位浮点型(float64) 5. 待拒回复得分(score_rejected):数据类型为64位浮点型(float64) 6. 参考回复(reference_response):数据类型为字符串(string) 7. 优选回复组(chosen):列表类型,包含以下子字段: - 内容(content):数据类型为字符串(string) - 角色(role):数据类型为字符串(string) 8. 待拒回复组(rejected):列表类型,包含以下子字段: - 内容(content):数据类型为字符串(string) - 角色(role):数据类型为字符串(string) ### 数据划分与存储信息 - 训练偏好划分(train_prefs): 占用字节数:41380967 样本数量:5183 - 下载总大小:23152573 - 数据集总存储大小:41380967 ### 配置项 - 配置名称:默认配置(default) 数据文件: - 对应划分:train_prefs - 文件路径:data/train_prefs-*
提供机构:
YYYYYYibo
原始信息汇总

数据集概述

数据集名称

  • 名称: ultrafeedback_binarized_dataset_offline_pairrm_part2_mini_1

数据集特征

  • prompt: 字符串类型
  • prompt_id: 字符串类型
  • messages: 列表类型,包含以下子特征:
    • content: 字符串类型
    • role: 字符串类型
  • score_chosen: 浮点数类型(float64)
  • score_rejected: 浮点数类型(float64)
  • reference_response: 字符串类型
  • chosen: 列表类型,包含以下子特征:
    • content: 字符串类型
    • role: 字符串类型
  • rejected: 列表类型,包含以下子特征:
    • content: 字符串类型
    • role: 字符串类型

数据集分割

  • train_prefs:
    • num_bytes: 41380967
    • num_examples: 5183

数据集大小

  • download_size: 23152573
  • dataset_size: 41380967

配置

  • config_name: default
  • data_files:
    • split: train_prefs
    • path: data/train_prefs-*
搜集汇总
数据集介绍
main_image_url
构建方式
该数据集基于UltraFeedback大规模反馈语料库,通过离线方式对模型生成的响应进行偏好对构建。具体而言,研究团队采用PairRM(一种基于成对排序的奖励模型)对原始响应进行评分,从中筛选出得分最高与最低的响应分别作为chosen与rejected样本,从而形成二元偏好对。数据集中每个样本包含原始提示(prompt)、对应消息序列(messages)、被选与拒绝响应的分数(score_chosen与score_rejected)、参考响应(reference_response)以及最终的偏好对内容。这一构建流程确保了偏好标签的可靠性与区分度,为后续强化学习中的偏好对齐提供了高质量训练基础。
特点
本数据集的核心特点在于其精炼的二元偏好结构与明确的评分体系。每个样本均包含由PairRM模型赋予的连续浮点分数,使得偏好强度可量化,而不仅仅是简单的正负标签。此外,数据集中保留了参考响应字段,便于研究者对比模型生成与标准答案之间的差异。数据集规模适中,训练集包含5183个样本,兼顾了训练效率与数据多样性。消息序列采用角色与内容分离的结构化存储方式,支持多轮对话场景的偏好学习,适用于从指令微调到人类反馈强化学习(RLHF)等多种对齐技术。
使用方法
该数据集可直接用于训练偏好对齐模型,如DPO(直接偏好优化)或PPO(近端策略优化)等强化学习算法。使用时,研究者需加载train_prefs分割,其中每个样本的prompt可作为输入,chosen与rejected字段分别代表正负偏好响应。建议将prompt_id作为唯一标识符以避免数据泄露,并利用score_chosen与score_rejected计算偏好强度或进行加权训练。数据集以HuggingFace Datasets格式存储,支持通过load_dataset函数直接调用,便于集成到现有训练流水线中,同时可结合transformers库进行模型微调与评估。
背景与挑战
背景概述
在大型语言模型(LLM)的对齐优化领域,基于人类反馈的强化学习(RLHF)已成为提升模型输出质量的关键范式。该数据集由研究者YYYibo创建,旨在为偏好学习提供结构化的二元反馈样本。其核心研究问题聚焦于如何通过离线偏好数据对(pairwise comparisons)高效训练奖励模型或直接优化策略,从而减少对在线人类标注的依赖。数据集中包含的'chosen'与'rejected'响应及其对应的评分(score_chosen/score_rejected),为研究奖励建模、直接偏好优化(DPO)等方法提供了标准化训练资源。该数据集作为UltraFeedback项目的一部分,推动了开源社区在模型对齐、安全性和可控性方面的探索,尤其在小样本场景下的偏好学习实验中具有重要参考价值。
当前挑战
该数据集所解决的领域挑战在于,传统RLHF依赖大量实时人类偏好标注,成本高昂且难以扩展。而离线二元偏好数据集的构建面临多重困难:首先,需要从海量模型生成响应中筛选出具有判别性的偏好对,避免选择偏差导致奖励模型过拟合于噪声信号。其次,评分机制的标度一致性难以保证,不同标注者的主观判断差异可能引入评分噪声。此外,数据集中仅包含5183个训练样本,小规模数据量对偏好模型的泛化能力构成严峻考验,尤其在处理分布外提示(out-of-distribution prompts)时易出现奖励黑客(reward hacking)现象。最后,离线数据中'chosen'与'rejected'响应的质量差异若不够显著,将削弱二元偏好信号的信息密度,影响下游对齐效果。
常用场景
经典使用场景
在大型语言模型(LLM)的对齐优化研究中,该数据集被广泛用于偏好学习与强化学习微调。其核心结构包含成对的“chosen”与“rejected”响应,配合评分与提示信息,为直接偏好优化(DPO)或基于人类反馈的强化学习(RLHF)提供了标准化的训练样本。研究者通过该数据集可有效模拟模型输出质量排序,从而引导生成模型更贴近人类的价值观与偏好。
实际应用
在实际部署中,该数据集被用于优化对话系统、智能助手与内容生成模型。例如,通过微调使模型在开放域问答中避免有害或冗余输出,提升回答的准确性与礼貌度。企业可基于此类数据构建内部审核管道,自动化筛选模型响应质量,从而在客服、教育等场景中实现更安全、更符合用户期望的交互体验。
衍生相关工作
基于该数据集,学术界衍生出多项经典工作,包括离线偏好优化(Offline Preference Optimization)框架的提出,以及对比研究不同奖励模型对对齐效果的影响。部分工作进一步扩展了多轮对话中的偏好建模,或结合合成数据增强策略提升小样本场景下的对齐鲁棒性。这些研究共同构成了从静态偏好数据到动态对齐算法的知识脉络。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务