【解决方案】DistilQwen2.5-R1蒸馏小模型在PAI-ModelGallery的训练、评测、压缩及部署实践

本文涉及的产品
模型训练 PAI-DLC,100CU*H 3个月
交互式建模 PAI-DSW,每月250计算时 3个月
模型在线服务 PAI-EAS,A10/V100等 500元 1个月
简介: 阿里云的人工智能平台 PAI,作为一站式的机器学习和深度学习平台,对DistilQwen2.5-R1模型系列提供了全面的技术支持。无论是开发者还是企业客户,都可以通过 PAI-ModelGallery 轻松实现 Qwen2.5 系列模型的训练、评测、压缩和快速部署。本文详细介绍在 PAI 平台使用 DistilQwen2.5-R1 蒸馏模型的全链路最佳实践。

随着 DeepSeek-R1 和 QwQ-32B 等面向深度推理的大语言模型的开源,“大模型+慢思考”已成为拓展大语言模型智能边界的标准配置。然而,这些模型在资源受限的移动设备和边缘计算场景中的普及仍面临巨大挑战。因此,学术界和工业界迫切需要解决如何有效利用知识蒸馏技术,将这些超大规模深度推理模型的知识迁移到小模型中,从而提升计算效率并降低部署成本的问题。为此,阿里云人工智能平台 PAI 在 DistilQwen2.5 系列蒸馏小模型的基础上,推出了更为强大的 DistilQwen2.5-R1 系列深度推理模型


DistilQwen2.5-R1 系列以少量来自 DeepSeek-R1 的思维链蒸馏数据为基础,通过一系列创新的蒸馏策略,有效强化了小模型的深度思考能力。实验评估结果显示,DistilQwen2.5-R1 系列中的多种小规模模型在各项基准测试中表现优异,DistilQwen2.5-R1-7B 性能显著超越了其他开源蒸馏模型,包括 OpenThinker-7B。


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

一、PAI-ModelGallery介绍

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



二、运行环境要求

  • 本示例目前支持在阿里云北京、上海、深圳、杭州、乌兰察布、新加坡等多地域,使用 PAI-ModelGallery 产品运行。
  • 资源配置要求:
  • 训练阶段:PAI-DistilQwen2.5-R1-3B/7B 量级模型:最低使用A10(24GB显存)及以上卡型运行训练任务;
  • 部署阶段:

image.png


三、通过PAI-ModelGallery使用模型

  • 登录 PAI 控制台,选择地域并进入工作空间;
  • 左侧导航栏进入快速开始 > Model Gallery
  • 找到 PAI-DistilQwen2.5-R1 系列模型,以 PAI-DistilQwen2.5-3B-R1 为例,模型卡片如下图所示:

image.png


模型部署和调用

PAI 平台提供的 DistilQwen2.5-3B-R1 预置了模型的部署配置信息,可以选择BladeLLM 部署/VLLM 部署/Transformers 标准部署等多样化的部署,零代码即可完成模型部署,并支持以 OpenAI API 兼容的方式调用及使用 ChatLLM WebUI 进行实时交互。

image.png


模型微调训练

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

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

DPO 训练算法支持使用 Json 格式输入,每条数据由问题、预期模型输出的答案、不希望模型输出的答案组成,分别用 “rompt”、“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 中。DistilQwen2.5-3B-R1 至少需要使用A10(24GB显存)的 GPU 资源,请确保选择使用的资源配额内有充足的计算资源。

image.png

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

image.png

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

image.png

如果需要将模型部署至 PAI-EAS,可以在同一页面的模型部署卡面选择资源组,并且点击“部署”按钮实现一键部署。模型调用方式和上文直接部署模型的调用方式相同。

如果需要评测微调后模型的性能,可以从任务页面右上角评测按钮进入评测页。详情见下一节:模型评测。


模型评测

PAI 为 DistilQwen2.5-3B-R1 模型配置了评测算法,支持用户以开箱即用的方式对 DistilQwen2.5-3B-R1 以及微调后模型进行评测。通过评测能帮助用户和其他模型做性能对比,指导用户进行精准地模型选择和优化。

模型评测入口:

1.从“Model Gallery”页面完成 DistilQwen2.5 开源模型的评测

image.png

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

image.png

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

image.png

image.png

  • 自定义数据集评测

模型评测支持文本匹配指标 BLEU/ROUGLE,以及裁判员模型评测(专家模式)。用户可以基于自己场景的独特数据,评测所选模型是否适合自己的场景。

评测需要提供 JSONL 格式的评测集文件,每行数据是一个 List ,使用question标识问题列,answer标识答案列

  • 公开数据集评测

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


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

image.png

image.png

模型压缩

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

image.png

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

image.png


四、通过PAI-ModelGallery进行大模型蒸馏

除了可以在 PAI-ModelGallery 使用 DistilQwen2.5-3B-R1 蒸馏小模型,PAI-ModelGallery 还具备一系列能力对大语言模型训练所需的指令进行扩展和改写。通过在 PAI-ModelGallery 部署教师大语言模型,以及用于指令增强和指令优化的专精小模型,用户可以轻松实现模型蒸馏的各个算法功能。更多技术的最佳实践,请参考先前发布的“大语言模型数据增强与模型蒸馏解决方案。对于新出的 DeepSeek-R1 类推理模型的蒸馏,用户也可以参考“蒸馏DeepSeek-R1等深度推理大模型”来训练部署自己的推理模型



五、结论

在技术发展的快车道上,阿里云的 Qwen 模型系列和 DistilQwen2.5-3B-R1 模型系列为我们展示了大型语言模型在推理场景下的巨大潜力。它在少量来自  DeepSeek-R1 的思维链数据基础上,通过创新蒸馏策略增强了小模型的深度思考能力。实验结果表明,该系列模型在多个基准测试中表现出色,尤其是 DistilQwen2.5-R1-7B 的性能全面超越了其他开源蒸馏模型。为了方便实际应用,这些模型的 Checkpoint 已在 Hugging Face 和 Model Scope 社区中公开。在未来,随着大语言模型和知识蒸馏技术更进一步的发展,我们将推出各种领域、各种规格的 DistilQwen 系列模型,充分促进大语言模型在实际应用中的降本增效。阿里云的PAI平台则提供了强大的支持,使得开发者和企业客户可以更加轻松地部署和优化这些模型。本解决方案从全方位解析了在PAI平台使用 DistilQwen2.5-3B-R1 的最佳实践,为用户提供了清晰的指导和有价值的参考。



六、相关资源链接



技术交流答疑群

image.png

相关实践学习
使用PAI-EAS一键部署ChatGLM及LangChain应用
本场景中主要介绍如何使用模型在线服务(PAI-EAS)部署ChatGLM的AI-Web应用以及启动WebUI进行模型推理,并通过LangChain集成自己的业务数据。
机器学习概览及常见算法
机器学习(Machine Learning, ML)是人工智能的核心,专门研究计算机怎样模拟或实现人类的学习行为,以获取新的知识或技能,重新组织已有的知识结构使之不断改善自身的性能,它是使计算机具有智能的根本途径,其应用遍及人工智能的各个领域。 本课程将带你入门机器学习,掌握机器学习的概念和常用的算法。
目录
打赏
0
7
8
0
3595
分享
相关文章
【新模型速递】PAI一键云上零门槛部署DeepSeek-V3-0324、Qwen2.5-VL-32B
PAI-Model Gallery 集成国内外 AI 开源社区中优质的预训练模型,涵盖了 LLM、AIGC、CV、NLP 等各个领域,用户可以通过 PAI 以零代码方式实现从训练到部署再到推理的全过程,获得更快、更高效、更便捷的 AI 开发和应用体验。 现阿里云PAI-Model Gallery已同步接入DeepSeek-V3-0324、Qwen2.5-VL-32B-Instruct两大新模型,提供企业级部署方案。
AI训练师入行指南(三):机器学习算法和模型架构选择
从淘金到雕琢,将原始数据炼成智能珠宝!本文带您走进数字珠宝工坊,用算法工具打磨数据金砂。从基础的经典算法到精密的深度学习模型,结合电商、医疗、金融等场景实战,手把手教您选择合适工具,打造价值连城的智能应用。掌握AutoML改装套件与模型蒸馏术,让复杂问题迎刃而解。握紧算法刻刀,为数字世界雕刻文明!
64 6
生物医药蛋白分子数据采集:支撑大模型训练的技术实践分享
作为生物信息学领域的数据工程师,近期在为蛋白质相互作用预测AI大模型构建训练集时,我面临着从PDB、UniProt等学术数据库获取高质量三维结构、序列及功能注释数据的核心挑战。通过综合运用反爬对抗技术,成功突破了数据库的速率限制、验证码验证等反爬机制,将数据采集效率提升4倍,为蛋白质-配体结合预测模型训练提供了包含10万+条有效数据的基础数据集,提高了该模型预测的准确性。
69 1
DistilQwen2.5蒸馏小模型在PAI-ModelGallery的训练、评测、压缩及部署实践
DistilQwen2.5 是阿里云人工智能平台 PAI 推出的全新蒸馏大语言模型系列。通过黑盒化和白盒化蒸馏结合的自研蒸馏链路,DistilQwen2.5各个尺寸的模型在多个基准测试数据集上比原始 Qwen2.5 模型有明显效果提升。这一系列模型在移动设备、边缘计算等资源受限的环境中具有更高的性能,在较小参数规模下,显著降低了所需的计算资源和推理时长。阿里云的人工智能平台 PAI,作为一站式的机器学习和深度学习平台,对 DistilQwen2.5 模型系列提供了全面的技术支持。本文详细介绍在 PAI 平台使用 DistilQwen2.5 蒸馏小模型的全链路最佳实践。
【重磅发布】AllData数据中台核心功能:机器学习算法平台
杭州奥零数据科技有限公司成立于2023年,专注于数据中台业务,维护开源项目AllData并提供商业版解决方案。AllData提供数据集成、存储、开发、治理及BI展示等一站式服务,支持AI大模型应用,助力企业高效利用数据价值。
K-means聚类算法是机器学习中常用的一种聚类方法,通过将数据集划分为K个簇来简化数据结构
K-means聚类算法是机器学习中常用的一种聚类方法,通过将数据集划分为K个簇来简化数据结构。本文介绍了K-means算法的基本原理,包括初始化、数据点分配与簇中心更新等步骤,以及如何在Python中实现该算法,最后讨论了其优缺点及应用场景。
292 6
基于机器学习的人脸识别算法matlab仿真,对比GRNN,PNN,DNN以及BP四种网络
本项目展示了人脸识别算法的运行效果(无水印),基于MATLAB2022A开发。核心程序包含详细中文注释及操作视频。理论部分介绍了广义回归神经网络(GRNN)、概率神经网络(PNN)、深度神经网络(DNN)和反向传播(BP)神经网络在人脸识别中的应用,涵盖各算法的结构特点与性能比较。
机器学习算法的优化与改进:提升模型性能的策略与方法
机器学习算法的优化与改进:提升模型性能的策略与方法
606 13
机器学习算法的优化与改进:提升模型性能的策略与方法
使用 PAI-DSW x Free Prompt Editing图像编辑算法,开发个人AIGC绘图小助理
使用 PAI-DSW x Free Prompt Editing图像编辑算法,开发个人AIGC绘图小助理
CCS 2024:如何严格衡量机器学习算法的隐私泄露? ETH有了新发现
在2024年CCS会议上,苏黎世联邦理工学院的研究人员提出,当前对机器学习隐私保护措施的评估可能存在严重误导。研究通过LiRA攻击评估了五种经验性隐私保护措施(HAMP、RelaxLoss、SELENA、DFKD和SSL),发现现有方法忽视最脆弱数据点、使用较弱攻击且未与实际差分隐私基线比较。结果表明这些措施在更强攻击下表现不佳,而强大的差分隐私基线则提供了更好的隐私-效用权衡。
90 14

热门文章

最新文章

相关产品

  • 人工智能平台 PAI
  • AI助理

    你好,我是AI助理

    可以解答问题、推荐解决方案等