five

Vezora/Wizard_Math_Alpaca

收藏
Hugging Face2023-08-14 更新2024-03-04 收录
下载链接:
https://hf-mirror.com/datasets/Vezora/Wizard_Math_Alpaca
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集包含两个文件:Math.json和GM8SK.jsonl,它们被转换为Alpaca格式。GM8sk.jsonl用于评估,Math.json用于训练。数据集还提供了关于ALPACA LORA用户的使用建议,包括可以针对的模块和超参数设置。

This dataset contains two files: Math.json and GM8SK.jsonl, both of which are converted to the Alpaca format. GM8SK.jsonl is used for evaluation, while Math.json is utilized for training. The dataset also provides usage guidelines for users of Alpaca LoRA, including recommended target modules and hyperparameter settings.
提供机构:
Vezora
原始信息汇总

数据集概述

数据文件

  • MATH_Alpaca.json: 包含约5,000个示例,用于评估。
  • gm8sk_Alpaca.json: 包含约1,000个示例,用于评估。

训练与评估

  • 数据集可用于训练模型。

模型训练建议

  • ALPACA LORA用户: 可针对以下模块使用lora: "gate_proj", "down_proj", "up_proj", "q_proj", "v_proj", "k_proj", "o_proj"。
  • Platypus模型: 使用以下模块: "gate_proj", "down_proj", "up_proj"。

超参数设置

  • Platypus2-13B / 70B模型:
    • 批量大小: 16
    • 微批量大小: 1
    • 训练轮数: 1
    • 学习率: 4e-4 / 3e-4
    • 截断长度: 4096
    • lora rank: 16
    • lora alpha: 16
    • lora dropout: 0.05
    • lora目标模块: "gate_proj", "down_proj", "up_proj"
    • 训练输入: False
    • 添加eos token: False
    • 按长度分组: False
    • 提示模板: alpaca
    • 学习率调度器: cosine
    • 预热步骤: 100

训练建议

  • 建议使用批量大小为4-10,截断长度不超过2048,以避免VRAM问题。
  • 使用load_in_4bit, Normal Float, 和bf16。
  • 对于单张24GB显卡,需编辑"training.py"文件以修改目标模块。
  • 使用alpaca lora时,运行train.py命令时需添加--lora_target_modules参数。
搜集汇总
数据集介绍
main_image_url
构建方式
在数学推理与大语言模型微调的交汇领域,Vezora/Wizard_Math_Alpaca数据集应运而生。该数据集通过将Math.json和GM8SK.jsonl两个原始数学语料库转化为统一的Alpaca指令格式而构建,其中Math.json包含约5000个训练样本,GM8sk_Alpaca.json则囊括约1000个评估样本。这种格式转换使得数据能够直接适配Alpaca风格的指令微调框架,为数学推理能力的增强提供了结构化训练素材。
特点
该数据集的核心特点在于其双重用途设计:训练集与评估集彼此独立,分别服务于模型优化与性能验证。同时,数据集对LoRA微调技术进行了深度适配,明确列出了可针对性调整的模块,如gate_proj、down_proj、up_proj等,并参考了Platypus模型取得优异结果时的超参数配置,为研究者提供了可复现的调优基准。这种设计兼顾了灵活性与实用性,使得数据集能够适应不同计算资源下的微调需求。
使用方法
使用该数据集时,研究者可根据目标选择训练集或评估集进行模型微调或测试。对于Alpaca LoRA用户,建议在训练命令中通过--lora_target_modules参数指定目标模块,并参考推荐的批量大小(4-10)和截断长度(≤2048)以避免显存溢出。若使用oobabooga框架,需修改training.py中的模块配置;而采用alpaca lora时则直接传递参数即可。此外,加载4bit量化、bf16和Normal Float等设置可进一步提升训练效率。
背景与挑战
背景概述
在自然语言处理领域,数学推理能力的提升一直是大型语言模型研究的前沿课题。Vezora/Wizard_Math_Alpaca数据集由研究人员于2023年创建,旨在通过整合MATH和GSM8K两个经典数学推理基准,构建适用于Alpaca格式的指令微调资源。该数据集包含约5000个训练样本和1000个评估样本,覆盖从基础算术到复杂竞赛题的多层次数学问题。其核心研究问题在于探索如何利用有限的高质量数学数据,通过参数高效微调方法(如LoRA)提升模型的数学推理能力。该数据集在开源社区产生了广泛影响,为后续数学推理模型(如Platypus系列)提供了关键训练基础,推动了数学领域专用语言模型的发展。
当前挑战
该数据集面临的挑战首先体现在领域问题层面:数学推理要求模型具备符号操作、逻辑推导与多步计算能力,而现有语言模型常因缺乏结构化思维而陷入计算错误或逻辑断裂。其次,构建过程中存在多重困难:MATH与GSM8K的问题格式差异导致数据统一化成本高昂;仅5000个训练样本难以覆盖数学问题的多样性,易引发过拟合;Alpaca格式的转换需要精心设计指令模板以保留原始推理路径,而LoRA微调时目标模块的选择(如gate_proj与q_proj)对模型性能影响显著,却缺乏系统性研究。此外,训练资源限制(如单卡24GB显存)迫使研究者权衡批次大小与序列长度,进一步增加了模型优化的复杂性。
常用场景
经典使用场景
Vezora/Wizard_Math_Alpaca数据集为数学推理领域的自然语言处理研究提供了标准化训练与评估资源。该数据集将MATH和GSM8K数学问题集转换为Alpaca指令格式,包含约5000条训练样本和1000条评估样本,覆盖从基础算术到竞赛级数学的多层次推理任务。研究者可借助此数据集对大型语言模型进行数学推理能力的微调与评测,尤其适用于参数高效微调方法如LoRA,通过指定目标模块(如gate_proj、down_proj等)实现轻量化训练,是探索数学语言模型能力边界的重要基准。
实际应用
在实际应用中,该数据集可用于构建智能教育辅助系统,如自动解答数学问题的聊天机器人或个性化习题推荐引擎。基于该数据集微调的模型能够理解自然语言描述的数学问题并生成逐步推理过程,适用于在线学习平台中的实时答疑场景。同时,其轻量化微调特性使得在消费级GPU(如24GB显存)上部署数学推理模型成为可能,降低了教育科技企业的人工智能应用门槛,促进了数学辅导工具的普及与迭代。
衍生相关工作
基于该数据集衍生的经典工作包括Platypus系列模型,其通过采用特定的LoRA目标模块组合(gate_proj、down_proj、up_proj)在数学推理任务上取得了显著成效,验证了不同参数更新策略对模型性能的影响。此外,该数据集为后续研究如数学推理链增强、多任务联合训练及跨领域知识迁移提供了基准测试平台,推动了如MATH-Alpaca等衍生数据集的构建,进一步丰富了数学语言模型的评估体系与训练范式。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务