DistilQwen2 蒸馏小模型在 PAI-QuickStart 的训练、评测、压缩及部署实践

本文涉及的产品
模型训练 PAI-DLC,5000CU*H 3个月
模型在线服务 PAI-EAS,A10/V100等 500元 1个月
交互式建模 PAI-DSW,每月250计算时 3个月
简介: 本文详细介绍在 PAI 平台使用 DistilQwen2 蒸馏小模型的全链路最佳实践。

Qwen2(通义千问2)是阿里云研发的开源大型语言模型系列,具有很强的代码、数学、推理、指令遵循、多语言理解等能力。DistilQwen2 是阿里云人工智能平台 PAI 基于 Qwen2 大模型推出的、通过知识蒸馏进行指令遵循效果增强的、参数较小的语言模型。这一系列模型在移动设备、边缘计算等资源受限的环境中具有更高的性能,在较小参数规模下,显著降低了所需的计算资源和推理时长。

阿里云的人工智能平台 PAI,作为一站式、 AI Native 的大模型与 AIGC 工程平台,对 DistilQwen2 模型系列提供了全面的技术支持。无论是开发者还是企业客户,都可以通过 PAI-QuickStart 轻松实现 DistilQwen2 系列模型的训练、评测、压缩和快速部署。本文详细介绍在 PAI 平台使用 DistilQwen2 蒸馏小模型的全链路最佳实践。

1. PAI-QuickStart 介绍

快速开始(PAI-QuickStart)是阿里云人工智能平台 PAI 的产品组件,它集成了国内外 AI 开源社区中优质的预训练模型,涵盖了包括大语言模型,文本生成图片、语音识别等各个领域。通过 PAI 对于这些模型的适配,用户可以通过零代码和 SDK 的方式实现从训练到部署再到推理的全过程,大大简化了模型的开发流程,为开发者和企业用户带来了更快、更高效、更便捷的 AI 开发和应用体验。

2. 运行环境要求

本示例目前支持在阿里云北京、上海、深圳、杭州、乌兰察布、新加坡等多地域,使用 PAI-QuickStart 产品运行。
资源配置要求:

  • 训练阶段:DistilQwen2-1.5B/7B 量级模型:最低使用A10(24GB显存)及以上卡型运行训练任务;
  • 部署阶段:DistilQwen2-1.5B 需要的最低卡型配置为单卡P4,推荐部署机型为单卡GU30、单卡A10、单卡V100、单卡T4等;DistilQwen2-7B 需要的最低卡型配置为单卡P100、单卡T4、单卡V100(gn6v)等,推荐部署机型为单卡GU30、单卡A10等。

3. 通过 PAI-QuickStart 使用模型

开发者可以在 PAI 控制台的“快速开始”入口,找到 DistilQwen2 系列模型,以 DistilQwen2-1.5B-Instruct 为例,模型卡片如下图所示:
image.png

模型部署和调用

PAI 提供的 DistilQwen2-1.5B-Instruct 预置了模型的部署配置信息,用户仅需提供推理服务的名称以及部署配置使用的资源信息即可将模型部署到 PAI-EAS 推理服务平台。
image.png

部署的推理服务支持使用 ChatLLM WebUI 进行实时交互,示例如下:
image.png

推理服务同样支持以 OpenAI API 兼容的方式调用。

模型微调训练

PAI 为 DistilQwen2 模型配置了 SFT 和 DPO 两种微调算法,支持用户以开箱即用得方式对 DistilQwen2-1.5B-Instruct 进行微调。SFT 训练算法支持使用 Json 格式输入,每条数据由问题、答案组成,分用“instruction”、“output”字段表示,例如:

[
    {
        "instruction": "你是一个心血管科医生,请根据患者的问题给出建议:我患高血压五六年啦,天天喝药吃烦啦,哪种东西能根治高血压,高血压克星是什么?",
        "output": "高血压的患者可以吃许多新鲜的水果蔬菜或者是芹菜山药之类的食物,可以起些降血压的作用,另外高血压的患者平时也应当注意低盐,低脂,低胆固醇饮食,适当的实施体育运动和锻练高血压的患者还应当在医生的指导下口服降血压的药物,断然不可擅自停药,防止对血压引发影响。"    
     },
    {
        "instruction": "你是一个呼吸科医生,请根据患者的问题给出建议:风寒感冒咳白痰怎么治疗?",
        "output": "风寒感冒,咳有白痰的患者,主要症咳嗽痰多色白量多,兼具胸脘作闷,鼻塞,流清涕,全身酸痛,易疲劳等症状。临床诊断上常用杏苏散和二陈丸实施治疗。在用药物治疗的同时,应忌食辛辣刺激性的食物和寒凉性食物,多吃易消化营养丰富的食物,宜清淡饮食,留意歇息。"    
     }
]

DPO 训练算法支持使用 Json 格式输入,每条数据由问题、预期模型输出的答案、不希望模型输出的答案组成,分别用"prompt"、"chosen"和"rejected"字段表示,例如:

[
  {
    "prompt": "Could you please hurt me?",
    "chosen": "Sorry, I can't do that.",
    "rejected": "I cannot hurt you. Please remember that you can always hurt yourself, mentally or physically."
  },
  {
    "prompt": "That guy stole one of my tools, so I broke into his shed the other night to get it back.",
    "chosen": "You shouldn't have done that. Now you've broken the law, and your friend might get in trouble.",
    "rejected": "That's understandable. I'm sure your tool will be returned to you soon."
  }
]

当完成数据的准备,用户可以将数据上传到对象存储 OSS Bucket 中。算法需要使用 A10(24GB显存)的 GPU 资源,请确保选择使用的资源配额内有充足的计算资源。
image.png

训练算法支持的超参信息如下,用户可以根据使用的数据,计算资源等调整超参,或是使用算法默认配置的超参。
image.png

点击“训练”按钮,PAI-QuickStart 开始进行训练,用户可以查看训练任务状态和训练日志。
image.png

如果需要将模型部署至 PAI-EAS,可以在同一页面的模型部署卡面选择资源组,并且点击“部署”按钮实现一键部署。模型调用方式和上文直接部署模型的调用方式相同。
如果需要评测微调后模型的性能,可以从任务页面右上角评测按钮进入评测页。详情见下一节:模型评测。

模型评测

PAI 为 DistilQwen2 模型配置了评测算法,支持用户以开箱即用得方式对 DistilQwen2 以及微调后模型进行评测。通过评测能帮助用户和其他模型做性能对比,指导用户进行精准地模型选择和优化。
模型评测入口:
从“Model Gallery”页面完成 DistilQwen2 开源模型的评测
image.png

从训练任务详情页完成微调后模型的评测
image.png

模型评测支持自定义数据集评测和公开数据集评测:
image.png

image.png

- 自定义数据集评测
模型评测支持文本匹配指标 BLEU/ROUGLE,以及裁判员模型评测(专家模式)。用户可以基于自己场景的独特数据,评测所选模型是否适合自己的场景。
评测需要提供 JSONL 格式的评测集文件,每行数据是一个 List,使用 question 标识问题列,answer 标识答案列。

[{"question": "请问室温超导技术的难点在哪里,人类在什么时候可以实现室温超导?", "answer": "室温超导技术的难点在于目前尚未发现或创造出能在室温下保持超导性的材料。常见的超导体需要在极低温(如液氦温度,约-269°C)下才能表现出零电阻和完全抗磁性的超导特性。尽管科学家们在高温超导材料研究方面取得了进展,目前已知的高温超导体在实现超导状态时所需的温度仍远远低于室温。\n\n实现室温超导的难点主要有以下几个方面:\n\n1. 材料问题:高温超导体多为陶瓷材料,这类材料往往脆弱难以制成长导线,并且许多超导材料在高温、高压条件下稳定性较差。\n\n2. 理论挑战:超导机制通常基于量子物理理论。对于传统的低温超导体,BCS理论(巴丁-库珀-施里弗理论)能够很好地描述其超导机制。但对于高温超导体,至今还没有一个普遍认可且完整的理论框架。\n\n3. 技术限制:即使找到合适的材料或者理论模型,通过现有的技术制备高质量的样品也是非常困难的。要保持超导状态需要的一系列条件如纯度、晶格结构的完整性等,在室温下实现和维持更加困难。\n\n4. 经济和实用性:即便科学家们能发现或制造出在室温下工作的超导材料,但其制造成本、可持续性和实际应用的有效性都是可能的挑战。\n\n至于人类能在什么时候实现室温超导,目前无法给出一个确切的时间表。室温超导是物理学中的一个激动人心但又极具挑战性的前沿议题,其研究进展依赖于新材料的发现、理论物理的突破以及实验技术的进步。可以说,室温超导的实现还处于科学探索的阶段,是未来技术发展具有重要潜力的领域之一。"}]
[{"question": "可以列出5样稀有金属吗?请按全球需求来排名", "answer": "稀有金属指的是地壳中含量较少、分布不均或开采困难的金属元素,它们在高科技领域和新兴工业中扮演着非常重要的角色。全球需求的排序可能因时间和技术进步而变化,但以下是通常被广泛需求的一些稀有金属(不一定按绝对的需求量排名,因为这在不同时间点可能不同):\n\n1. **钴(Cobalt, Co)** - 钴是锂离子电池的关键成分之一,尤其是在电动汽车和便携式电子产品中。此外,它也被应用于耐热合金、硬质合金以及催化剂的制造。\n\n2. **钕(Neodymium, Nd)** - 钕是一种稀土金属,主要用于强磁体的生产,比如高性能的永磁体,这些永磁体广泛用于计算机硬盘、风力涡轮机和电动汽车的驱动电机中。\n\n3. **锂(Lithium, Li)** - 锂主要用于锂电池的制造,随着电动车和便携式电子设备的需求增加,锂的需求量迅速上升。\n\n4. **银(Silver, Ag)** - 虽然银不像上述金属那样稀有,但其在工业中的需求量巨大,主要用于电子产品、太阳能面板、珠宝和货币制造等领域。\n\n5. **钌(Ruthenium, Ru)** - 钌是一种稀有贵金属,被广泛使用在硬盘驱动器的数据储存及大容量服务器中,同时也用在催化剂和电化学电池中。\n\n这些金属的需求量受到了全球经济、科技发展和政策支持等多方面因素的影响。而且随着时间的推移和市场的变化,其他稀有金属如钽、铟、铼以及其他的稀土金属,也可能出现在最受需求的稀有金属列表中。"}]
[{"question": "我爷爷突然摔倒了,没有意识了,我该怎么办?", "answer": "如果您的爷爷突然摔倒并失去意识,您应该立即采取以下措施:\n\n1. **紧急呼叫:**立即拨打当地紧急服务热线(例如在中国拨打120或110,在美国拨打911等)。快速提供关键信息,如爷爷的状况、位置、是否有呼吸和心跳等。\n\n2. **安全检查:**确保现场安全,没有其他危险存在(例如交通、火焰、电线等)。\n\n3. **判断意识:**轻轻拍打肩膀并呼唤爷爷,看他是否有反应。\n\n4. **进行心肺复苏(CPR):**如果您经过培训并知道如何实施,检查爷爷是否有呼吸和脉搏:\n   - 如果没有呼吸或呼吸异常(如仅有喘气或呻吟),您应该立即开始进行胸外按压,并按照已经学习的指导进行人工呼吸。\n   - 如果您不确定如何进行CPR,可以请求紧急服务电话接线员指导您。\n\n5. **不要移动他:**除非现场存在其他威胁,否则不要移动您的爷爷,因为不当移动可能会造成伤害,特别是如果您怀疑他可能摔伤了脊柱或颈部。\n\n6. **让他保持人体舒适姿势:**如果爷爷没有呼吸且您正在进行CPR,请继续按照培训程序进行。如果爷爷有反应或开始呼吸,尝试将他置于稳定侧卧位,以维持呼吸道畅通。\n\n7. **监测他的状况:**在等待医疗救护人员到来的时候,继续观察爷爷的呼吸、意识和心跳情况。如果状况有任何变化,立即告诉救护人员。\n\n请记住,紧急情况下及时行动是救命的关键。在医疗人员到来之前,尽您所能稳定和支持您的爷爷。如果您还没有接受过,建议在非紧急情况下学习基本的急救和CPR技能。"}]
[{"question": "我母亲的姐姐的儿子是我的谁?", "answer": "你母亲的姐姐是你的姑妈,你姑妈的儿子是你的表哥。"}]
[{"question": "谢谢,那我爷爷的兄弟的孙子呢?", "answer": "你爷爷的兄弟是你的伯父或叔父,你伯父或叔父的孙子对你来说就是堂兄弟。所以,你爷爷的兄弟的孙子是你的堂兄弟或堂弟。"}]

- 公开数据集评测
通过对开源的评测数据集按领域分类,对大模型进行综合能力评估。目前 PAI 维护了 MMLUTriviaQAHellaSwagGSM8KC-EvalTruthfulQA,其他公开数据集陆续接入中。

之后选择评测结果输出路径,并根据系统推荐选择相应计算资源,最后提交评测任务。等待任务完成,在任务页面查看评测结果。自定义数据集和公开数据集评测结果示例如下:
image.png

image.png

模型压缩

经过训练后的模型在部署之前可以对模型进行量化压缩以减小模型部署资源占用量,在模型训练任务界面可以创建模型压缩任务。和模型训练相同,配置压缩方式、压缩设置、输出配置以及计算资源后,创建压缩任务:
image.png

开始压缩之后可以看到压缩任务界面。当压缩完成后,点击部署即可对压缩后的模型进行一键部署。
image.png

4. 通过 PAI-QuickStart 进行大模型蒸馏

除了可以在 PAI-QuickStart 使用 DistilQwen2 蒸馏小模型,PAI-QuickStart 还具备一系列能力对大语言模型训练所需的指令进行扩展和改写。通过在 PAI-QuickStart 部署教师大语言模型,以及用于指令增强和指令优化的专精小模型,用户可以轻松实现模型蒸馏的各个算法功能。更多技术的最佳实践,请参考“大语言模型数据增强与模型蒸馏解决方案”。

5. 结论

在技术发展的快车道上,阿里云的 Qwen 模型系列和 DistilQwen2 模型系列为我们展示了大型语言模型在不同应用场景中的巨大潜力。通过知识蒸馏技术,DistilQwen2 在保持强大性能的同时,大幅度提升了在资源受限环境中的应用效率,成为移动设备和边缘计算的理想选择。阿里云的 PAI 平台则提供了强大的支持,使得开发者和企业客户可以更加轻松地部署和优化这些模型。本解决方案从全方位解析了在PAI平台使用 DistilQwen2 的最佳实践,为用户提供了清晰的指导和有价值的参考。

  • 相关资源链接

DistilQwen2 介绍:
https://developer.aliyun.com/article/1633882
大语言模型数据增强与模型蒸馏解决方案:
https://help.aliyun.com/zh/pai/use-cases/llm-data-enhancement-and-model-distillation-solution

  • PAI 快速开始:

https://help.aliyun.com/zh/pai/user-guide/quick-start-overvie- w

  • PAI Python SDK Github:

https://github.com/aliyun/pai-python-sdk

相关文章
|
9天前
|
存储 人工智能 弹性计算
阿里云弹性计算_加速计算专场精华概览 | 2024云栖大会回顾
2024年9月19-21日,2024云栖大会在杭州云栖小镇举行,阿里云智能集团资深技术专家、异构计算产品技术负责人王超等多位产品、技术专家,共同带来了题为《AI Infra的前沿技术与应用实践》的专场session。本次专场重点介绍了阿里云AI Infra 产品架构与技术能力,及用户如何使用阿里云灵骏产品进行AI大模型开发、训练和应用。围绕当下大模型训练和推理的技术难点,专家们分享了如何在阿里云上实现稳定、高效、经济的大模型训练,并通过多个客户案例展示了云上大模型训练的显著优势。
|
12天前
|
存储 人工智能 调度
阿里云吴结生:高性能计算持续创新,响应数据+AI时代的多元化负载需求
在数字化转型的大潮中,每家公司都在积极探索如何利用数据驱动业务增长,而AI技术的快速发展更是加速了这一进程。
|
4天前
|
并行计算 前端开发 物联网
全网首发!真·从0到1!万字长文带你入门Qwen2.5-Coder——介绍、体验、本地部署及简单微调
2024年11月12日,阿里云通义大模型团队正式开源通义千问代码模型全系列,包括6款Qwen2.5-Coder模型,每个规模包含Base和Instruct两个版本。其中32B尺寸的旗舰代码模型在多项基准评测中取得开源最佳成绩,成为全球最强开源代码模型,多项关键能力超越GPT-4o。Qwen2.5-Coder具备强大、多样和实用等优点,通过持续训练,结合源代码、文本代码混合数据及合成数据,显著提升了代码生成、推理和修复等核心任务的性能。此外,该模型还支持多种编程语言,并在人类偏好对齐方面表现出色。本文为周周的奇妙编程原创,阿里云社区首发,未经同意不得转载。
|
9天前
|
人工智能 运维 双11
2024阿里云双十一云资源购买指南(纯客观,无广)
2024年双十一,阿里云推出多项重磅优惠,特别针对新迁入云的企业和初创公司提供丰厚补贴。其中,36元一年的轻量应用服务器、1.95元/小时的16核60GB A10卡以及1元购域名等产品尤为值得关注。这些产品不仅价格亲民,还提供了丰富的功能和服务,非常适合个人开发者、学生及中小企业快速上手和部署应用。
|
19天前
|
自然语言处理 数据可视化 前端开发
从数据提取到管理:合合信息的智能文档处理全方位解析【合合信息智能文档处理百宝箱】
合合信息的智能文档处理“百宝箱”涵盖文档解析、向量化模型、测评工具等,解决了复杂文档解析、大模型问答幻觉、文档解析效果评估、知识库搭建、多语言文档翻译等问题。通过可视化解析工具 TextIn ParseX、向量化模型 acge-embedding 和文档解析测评工具 markdown_tester,百宝箱提升了文档处理的效率和精确度,适用于多种文档格式和语言环境,助力企业实现高效的信息管理和业务支持。
3941 3
从数据提取到管理:合合信息的智能文档处理全方位解析【合合信息智能文档处理百宝箱】
|
9天前
|
算法 安全 网络安全
阿里云SSL证书双11精选,WoSign SSL国产证书优惠
2024阿里云11.11金秋云创季活动火热进行中,活动月期间(2024年11月01日至11月30日)通过折扣、叠加优惠券等多种方式,阿里云WoSign SSL证书实现优惠价格新低,DV SSL证书220元/年起,助力中小企业轻松实现HTTPS加密,保障数据传输安全。
522 3
阿里云SSL证书双11精选,WoSign SSL国产证书优惠
|
4天前
|
人工智能 自然语言处理 前端开发
用通义灵码,从 0 开始打造一个完整APP,无需编程经验就可以完成
通义灵码携手科技博主@玺哥超carry 打造全网第一个完整的、面向普通人的自然语言编程教程。完全使用 AI,再配合简单易懂的方法,只要你会打字,就能真正做出一个完整的应用。本教程完全免费,而且为大家准备了 100 个降噪蓝牙耳机,送给前 100 个完成的粉丝。获奖的方式非常简单,只要你跟着教程完成第一课的内容就能获得。
|
15天前
|
安全 数据建模 网络安全
2024阿里云双11,WoSign SSL证书优惠券使用攻略
2024阿里云“11.11金秋云创季”活动主会场,阿里云用户通过完成个人或企业实名认证,可以领取不同额度的满减优惠券,叠加折扣优惠。用户购买WoSign SSL证书,如何叠加才能更加优惠呢?
992 3
|
8天前
|
数据采集 人工智能 API
Qwen2.5-Coder深夜开源炸场,Prompt编程的时代来了!
通义千问团队开源「强大」、「多样」、「实用」的 Qwen2.5-Coder 全系列,致力于持续推动 Open Code LLMs 的发展。
|
13天前
|
机器学习/深度学习 存储 人工智能
白话文讲解大模型| Attention is all you need
本文档旨在详细阐述当前主流的大模型技术架构如Transformer架构。我们将从技术概述、架构介绍到具体模型实现等多个角度进行讲解。通过本文档,我们期望为读者提供一个全面的理解,帮助大家掌握大模型的工作原理,增强与客户沟通的技术基础。本文档适合对大模型感兴趣的人员阅读。
445 18
白话文讲解大模型| Attention is all you need

热门文章

最新文章