five

Magpie-Align/Magpie-Air-MT-300K-v0.1

收藏
Hugging Face2024-07-03 更新2024-06-29 收录
下载链接:
https://hf-mirror.com/datasets/Magpie-Align/Magpie-Air-MT-300K-v0.1
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是通过Magpie方法从Llama 3 8B Instruct模型中生成的,包含了300K高质量的多轮对话数据。数据集的过滤条件包括输入质量、输入难度、奖励差异等,并且去除了重复和不完整的指令。数据集的结构包括模型名称、输入配置、对话内容等字段。

This dataset is generated by Llama 3 8B Instruct using the Magpie method, containing 300K high-quality multi-turn conversation data. The filtering conditions for the dataset include input quality, input difficulty, reward difference, etc., and remove repetitive and incomplete instructions. The structure of the dataset includes fields such as model name, input configuration, conversation content, etc.
提供机构:
Magpie-Align
原始信息汇总

数据集概述

数据集信息

  • 特征:

    • model: 模型名称,类型为字符串。
    • gen_input_config: 生成输入配置,包含以下子字段:
      • temperature: 温度参数,类型为浮点数。
      • top_p: 采样概率参数,类型为浮点数。
    • input1: 输入1,类型为字符串。
    • output1: 输出1,类型为字符串。
    • input2: 输入2,类型为字符串。
    • output2: 输出2,类型为字符串。
    • conversations: 对话列表,包含以下子字段:
      • from: 对话来源,类型为字符串。
      • value: 对话内容,类型为字符串。
    • uuid: 唯一标识符,类型为字符串。
  • 分割:

    • train: 训练集,包含300,000个样本,总大小为3,714,318,522字节。
  • 下载大小: 1,946,163,678字节。

  • 数据集大小: 3,714,318,522字节。

配置

  • 默认配置:
    • data_files:
      • train: 训练集文件路径为data/train-*

过滤设置

  • 输入质量: ≥ 良好
  • 输入难度: ≥ 中等
  • 奖励差异: ≥ 0
  • 移除重复和不完整的指令(例如以冒号结尾的指令)
  • 选择300K个具有最长响应的数据

数据集导航

模型名称 数据集 类型 描述
Llama 3 70B Instruct Magpie-Pro-1M SFT 使用Meta Llama 3 70B构建的1M原始对话。
Llama 3 70B Instruct Magpie-Pro-300K-Filtered SFT 应用过滤器并选择300K高质量对话。
Llama 3 70B Instruct Magpie-Pro-MT-300K SFT 选择300K困难问题并扩展为多轮对话。
Llama 3 8B Instruct Magpie-Air-3M SFT 使用Meta Llama 3 8B构建的3M原始对话。
Llama 3 8B Instruct Magpie-Air-300K-Filtered SFT 应用过滤器并选择300K高质量数据。
Llama 3 8B Instruct Magpie-Air-MT-300K SFT 选择300K困难问题并扩展为多轮对话。
搜集汇总
数据集介绍
main_image_url
构建方式
在大型语言模型对齐数据稀缺的背景下,Magpie-Align/Magpie-Air-MT-300K-v0.1数据集应运而生。该数据集采用创新的自合成方法,利用Llama-3-8B-Instruct模型的自回归特性,仅输入用户消息之前的左侧模板,即可自动生成用户查询及其对应回复。原始数据量达到400万条,经过严格筛选,最终保留30万条高质量实例。筛选标准包括输入质量不低于“良好”、输入难度不低于“中等”、奖励差异非负,并剔除重复和不完整的指令,同时优先选取回复长度最长的数据。此外,数据集还通过多轮对话扩展,增强了交互的深度与多样性。
使用方法
使用时,用户可直接通过HuggingFace Datasets库加载该数据集的训练分割,其中包含30万条样本。每条样本以对话形式组织,涵盖模型名称、生成配置(如温度和top-p参数)、多轮输入输出对以及唯一标识符。该数据集专为监督式微调(SFT)设计,适用于训练指令跟随型语言模型。需注意,该数据集不应与Magpie-Air-300K-Filtered同时使用,因为两者在第一轮对话上高度重叠。用户可参照官方代码库和论文中的实现细节,将其集成到微调流程中,以提升模型的对齐能力。
背景与挑战
背景概述
在大规模语言模型(LLM)对齐领域,高质量指令数据的获取始终是制约模型性能提升的关键瓶颈。尽管诸如Llama-3-Instruct等模型已开放权重,但其对齐数据却长期处于私密状态,这无疑阻碍了人工智能技术的民主化进程。现有开源数据创建方法受限于高昂的人力成本与预设的有限提示范围,难以实现规模化扩展,进而限制了公共对齐数据集的多样性与质量。针对这一困境,Magpie-Align团队于2024年提出了一种名为Magpie的自合成方法,旨在通过直接提取已对齐LLM内部的指令数据来突破上述局限。该研究由Magpie-Align团队主导,其核心洞察在于:利用Llama-3-Instruct等对齐模型的自回归特性,仅输入左侧模板至用户消息预留位置,即可自动生成用户查询。基于此方法,团队成功生成了400万条指令及其对应响应,并从中筛选出30万条高质量实例,构建了Magpie-Air-MT-300K-v0.1数据集。实验表明,使用该数据集微调的Llama-3-8B-Base模型在多项对齐基准测试中表现优异,甚至可与官方Llama-3-8B-Instruct相媲美,彰显了其在推动LLM对齐研究民主化与高效化方面的巨大潜力。
当前挑战
该数据集所面临的挑战首先来自领域问题的复杂性:LLM对齐任务要求指令数据兼具广泛覆盖性与精准导向性,而传统人工标注方式难以在规模与质量间取得平衡,且易受主观偏见影响。Magpie自合成方法虽能规模化生成数据,却需应对生成指令的多样性不足与潜在噪声问题。其次,在构建过程中,团队遭遇了多重技术挑战:如何确保自动生成的用户查询真实反映人类意图,避免模板化重复;如何设计有效的过滤机制以剔除不完整或低质量指令,例如通过引入输入质量、难度及奖励差异等评估维度;如何将单轮对话自然扩展为多轮交互,维持对话逻辑的连贯性与任务导向性。此外,数据集的规模达300K样本,其存储与处理对计算资源提出了较高要求,而筛选过程中响应长度的偏好设置亦可能引入偏差,需在后续应用中审慎考量。
常用场景
经典使用场景
Magpie-Air-MT-300K-v0.1 数据集最经典的用途在于为大型语言模型(LLM)的监督微调(SFT)提供高质量、多轮对话的指令数据。该数据集通过自合成方法从 Llama-3-8B-Instruct 中提取用户查询与模型回复,并经过严格过滤,选取了 300K 条具有较高输入质量、难度适中且回复较长的实例。研究者可利用该数据集对基础模型进行对齐训练,显著提升其在复杂对话场景中的表现能力。
解决学术问题
该数据集巧妙地解决了开源社区中高质量对齐数据匮乏的难题。传统指令数据构建依赖高昂的人力成本且预设范围有限,难以规模化扩展。Magpie-Air-MT-300K-v0.1 通过自回归机制直接从对齐模型中提取指令,突破了数据生成的瓶颈,使得大规模、多样化的对齐数据合成成为可能。实验表明,仅使用该数据集进行 SFT 即可媲美官方 Llama-3-8B-Instruct 在 AlpacaEval、ArenaHard 等基准上的表现,为低成本、高效率的模型对齐研究提供了新的范式。
实际应用
在实际应用中,该数据集可赋能企业或研究机构快速构建具备多轮对话能力的智能客服、虚拟助手及教育辅导系统。由于数据集中包含大量难度递增的指令和对应的模型回复,开发者可直接用于微调基础模型,使其掌握更自然、更精准的交互技巧。此外,其过滤机制确保了数据质量,降低了后续部署中产生有害或低质量输出的风险,从而加速了 LLM 在真实场景中的落地进程。
数据集最近研究
最新研究方向
当前,大语言模型的对齐技术正面临高质量指令数据匮乏与人工标注成本高昂的双重困境。Magpie-Align/Magpie-Air-MT-300K-v0.1 数据集的出现,为这一领域开辟了一条全新的自合成路径。该研究敏锐地捕捉到已对齐模型(如 Llama-3-Instruct)的自回归特性,仅通过输入左侧模板即可诱使其自主生成用户查询与对应回复,从而在无需人工干预的情况下,以极低成本合成大规模、多样化的指令数据。这一创新方法不仅成功破解了开源社区因缺乏对齐数据而面临的“黑箱”难题,更通过精细的过滤与多轮扩展策略,筛选出300K高质量实例。实验证明,基于该数据集微调的模型在 AlpacaEval、ArenaHard 等前沿对齐基准上,性能可与官方使用千万级数据且经偏好优化后的模型相媲美,甚至超越传统公开数据集。这标志着自合成对齐数据从理论走向实用,为AI民主化与高效模型对齐提供了可扩展、低成本的全新范式,深刻影响了未来大模型训练的数据生态。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务