five

heegyu/hh-rlhf-vicuna-format

收藏
Hugging Face2023-09-06 更新2024-03-04 收录
下载链接:
https://hf-mirror.com/datasets/heegyu/hh-rlhf-vicuna-format
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集来源于Antrophic/hh-rlhf,并转换为Vicuna格式。数据集包含训练集和测试集,分别有160800和8552个样本。数据集的特征包括chosen、rejected、source、context和instruction。chosen和rejected字段包含from和value子字段,source字段有四个可能的值:harmless-base、helpful-base、helpful-online、helpful-rejection-sampled。context字段是一个列表,包含from和value子字段,instruction字段也包含from和value子字段。在chosen、rejected和context字段中,from键的值要么是human,要么是gpt。

本数据集源自Antrophic/hh-rlhf,并转换为Vicuna格式。该数据集包含训练集与测试集,样本量分别为160800条与8552条。数据集的字段包括chosen、rejected、source、context及instruction:其中chosen与rejected字段均包含from与value两个子字段;source字段存在四种可选取值,即harmless-base、helpful-base、helpful-online与helpful-rejection-sampled;context字段为列表类型,同样包含from与value子字段;instruction字段亦包含from与value子字段。在chosen、rejected及context字段中,from键的取值仅可为human或gpt。
提供机构:
heegyu
原始信息汇总

数据集概述

数据配置

  • 默认配置
    • 训练集:路径为 data/train-*
    • 测试集:路径为 data/test-*

数据特征

  • 特征列表
    • chosen
      • from:数据类型为 string
      • value:数据类型为 string
    • rejected
      • from:数据类型为 string
      • value:数据类型为 string
    • source:数据类型为 string
    • context:列表类型
      • from:数据类型为 string
      • value:数据类型为 string
    • instruction
      • from:数据类型为 string
      • value:数据类型为 string

数据分割

  • 训练集
    • 字节数:210,193,081
    • 样本数:160,800
  • 测试集
    • 字节数:11,248,194
    • 样本数:8,552

数据大小

  • 下载大小:128,509,060 字节
  • 数据集大小:221,441,275 字节
搜集汇总
数据集介绍
main_image_url
构建方式
该数据集源自Anthropic发布的hh-rlhf数据集,经格式转换以适配Vicuna对话框架。原始数据包含人类与AI助手的多轮对话,其中chosen和rejected字段分别记录优选与劣质回应,用于偏好学习。转换过程保留了对话结构,将原始的多轮交互重组为统一的instruction、context、chosen与rejected字段,并添加source字段标识数据来源(如helpful-base、harmless-base等)。最终形成包含约16万训练样本和8500余测试样本的标准化格式数据集。
使用方法
该数据集可直接用于训练偏好对齐模型,如RLHF中的奖励模型或DPO算法。使用时需按默认配置加载train和test分割,通过chosen与rejected字段构建对比损失。instruction字段可作为独立指令输入,context字段提供多轮对话历史。对于需要统一对话格式的微调任务,可将chosen或rejected中的gpt回应作为目标输出,结合instruction和context形成完整的训练样本。建议根据source字段筛选特定子集以适配不同应用场景。
背景与挑战
背景概述
heegyu/hh-rlhf-vicuna-format 数据集源于 Anthropic 团队于 2022 年发布的 hh-rlhf 数据集,旨在探索基于人类反馈的强化学习(RLHF)在对话系统中的优化路径。该数据集由 Anthropic 的研究人员主导,核心研究问题聚焦于如何通过人类偏好标注来训练语言模型生成更安全、更有帮助的回复。通过将原始数据转换为 Vicuna 格式,该数据集扩展了在开源对话模型上的适用性,为后续研究提供了标准化接口。其影响力体现在推动了 RLHF 领域的可复现性研究,并成为评估模型对齐能力的重要基准之一,尤其在帮助性与无害性平衡的探讨中占据关键地位。
当前挑战
该数据集所解决的领域挑战在于语言模型回复的偏好对齐问题,即如何通过有限的人类标注数据引导模型生成既符合用户需求又避免有害内容的输出。构建过程中面临多重挑战:首先,原始数据中‘chosen’与‘rejected’标签的区分依赖于标注者主观判断,导致偏好边界模糊,尤其在‘helpful-base’与‘harmless-base’等来源间存在冲突;其次,对话上下文长度不一,部分样本缺乏足够历史信息以支撑偏好决策,增加了标注噪声;此外,从 Anthropic 原始格式到 Vicuna 格式的转换需处理字段映射与结构一致性,确保‘from’与‘value’键的语义无损,这对数据清洗与验证提出了严格要求。
常用场景
经典使用场景
在强化学习与人类反馈(RLHF)领域,heegyu/hh-rlhf-vicuna-format数据集被广泛用于训练对话模型以区分有益与无益的回复。该数据集将原始的Anthropic hh-rlhf数据转换为Vicuna格式,保留了对话上下文、人类指令、以及模型生成的被采纳(chosen)与被拒绝(rejected)回复。研究者通常利用该数据集进行偏好学习,如通过对比学习或奖励模型训练,使语言模型学会在开放式对话中生成更符合人类偏好的回答。其经典使用场景包括基于人类反馈的强化学习(RLHF)中的奖励模型训练,以及直接偏好优化(DPO)方法的模型微调,为对齐研究提供了标准化数据基础。
解决学术问题
该数据集的核心学术价值在于解决语言模型与人类价值观对齐的难题。传统监督学习难以捕捉人类对回复质量的细微偏好,而heegyu/hh-rlhf-vicuna-format通过提供成对的偏好数据(chosen vs. rejected),使研究者能够量化模型输出与人类期望的偏差。它支持了对奖励过拟合、偏好噪声鲁棒性等关键问题的探索,推动了RLHF理论框架的完善。此外,数据集包含harmless-base、helpful-base等多源标签,助力研究不同对齐目标(如无害性与有帮助性)间的权衡,其意义在于为构建安全、可控的AI系统提供了实证支撑。
实际应用
在实际应用中,该数据集被部署于对话系统的优化流程中。例如,企业可基于该数据训练奖励模型,对客服机器人生成的回复进行排序,筛选出更礼貌、更准确的答案。在内容审核场景中,模型通过对比chosen与rejected回复,学会规避有害或误导性输出。此外,该数据集还可用于个性化推荐系统的对话模块,通过偏好学习使虚拟助手适应不同用户的表达风格。其轻量化的Vicuna格式降低了数据预处理成本,使得中小型团队也能快速集成到现有的微调管线中,提升人机交互的流畅度与安全性。
数据集最近研究
最新研究方向
基于人类反馈的强化学习(RLHF)是当前大语言模型对齐研究的前沿方向,而heegyu/hh-rlhf-vicuna-format数据集作为Anthropic原始hh-rlhf数据的结构化重构版本,其将对话偏好数据转化为Vicuna格式,为探索指令遵循与安全性平衡提供了关键资源。该数据集通过保留多源标签(如helpful-base与harmless-base),支持研究者深入分析模型在有益性与无害性之间的权衡机制,尤其在近期关于AI伦理对齐的热点事件中,此类细粒度偏好数据成为评估模型过度拒绝或过度顺从行为的重要基准。其引入的上下文结构化设计,更推动了对话系统中长程依赖与价值锚定研究的发展,对构建可信赖AI系统具有里程碑意义。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务