DistilQwen2.5-R1 发布:知识蒸馏助推小模型深度思考
作者:蔡文睿(清素)、汪诚愚(熊兮)、严俊冰(玖烛)、黄俊(临在)
引言
随着 DeepSeek-R1 和 QwQ-32B 等面向深度推理的大语言模型的开源,“大模型+慢思考”已成为拓展大语言模型智能边界的标准配置。然而,这些模型在资源受限的移动设备和边缘计算场景中的普及仍面临巨大挑战。因此,学术界和工业界迫切需要解决如何有效利用知识蒸馏技术,将这些超大规模深度推理模型的知识迁移到小模型中,从而提升计算效率并降低部署成本的问题。为此,我们在 DistilQwen2.5 系列蒸馏小模型(看这里)的基础上,推出了更为强大的 DistilQwen2.5-R1 系列深度推理模型。
DistilQwen2.5-R1 系列以少量来自 DeepSeek-R1 的思维链蒸馏数据为基础,通过一系列创新的蒸馏策略,有效强化了小模型的深度思考能力。实验评估结果显示,DistilQwen2.5-R1 系列中的多种小规模模型在各项基准测试中表现优异(见下图)。例如,DistilQwen2.5-R1-7B 性能显著超越了其他开源蒸馏模型,包括 OpenThinker-7B。


为方便开发者和企业在实际应用中使用 DistilQwen2.5-R1 系列模型,其所有的 Checkpoint 已在 Hugging Face 和 Model Scope 开源社区中公开。本文将深入阐述 DistilQwen2.5-R1 的蒸馏算法、性能评估,并且提供在阿里云人工智能平台 PAI 上的使用指南及相关下载教程。
DistilQwen2.5-R1 中的知识蒸馏技术
本节中,我们主要描述 DistilQwen2.5-R1 模型训练中使用的数据增强与知识蒸馏技术。
由于自身参数量的显著差异,大模型与小模型的认知与推理轨迹有时并不完全一致。以数学问题为例:对于有的数学问题,小模型由于自身参数量的限制,会倾向于使用更基础的方法去解决问题。而大模型基于其强大的推理能力,会采用较为高阶的方法。比如经典的鸡兔同笼问题,小模型倾向于使用简单枚举法逐一试错,而大模型会直接通过列方程的较高级方法求解。
正是由于大小模型的认知轨迹偏差,小模型有时无法有效理解大模型的思维链,此时如果直接该思维链(Chain-of-Thought,CoT)蒸馏到小模型中,往往效果不佳。为此,我们设计了一种小型推理模型训练框架,以消除这种认知轨迹偏差带来的负面影响。在后续训练中,我们还利用这种偏差数据进一步提升小模型的推理能力,最终推出基于该训练框架的 DistilQwen2.5-R1 系列模型。我们提出的训练技术框架包含两个阶段:CoT 数据“评价-改进-验证”机制,以及基于不同认知轨迹数据的偏好优化算法。总体而言,DistilQwen2.5-R1 模型蒸馏的详细算法框架如下图所示:

给定原始的大模型思维链数据集,例如从 DeepSeek-R1 蒸馏的数据集,在一阶段,我们先对其进行数据难度评价,接着根据数据的难度等级对其进行相应的优化,优化之后还要对结果进行验证。我们使用改进且被验证的 CoT 数据集对模型进行 SFT 训练,获取模型的基础推理能力。在二阶段,我们利用一阶段已有的不同难度的 CoT 数据构造偏好数据集,在一阶段的基础上进一步提升小模型的推理能力。
CoT 数据“评价-改进-验证”机制
正如上文中提到的,大小模型间的认知推理轨迹有时存在显著偏差。因此,对于待蒸馏的大模型思维链数据集,小模型无法完全理解。阶段一正是基于这种认知偏差对数据集进行优化,采用了 LLM-as-a-Judge 的范式,对大模型的推理过程进行评价并改进。
给定问题、大模型的推理过程和问题的答案,我们使用模型判断这个推理过程是简单、中等还是困难。难度等级的核心标准是小模型是否能够遵循给定的推理过程得到问题的答案。以下是思维链的难度等级及定义:
中等: 小模型可以遵循该推理过程得到问题的答案。
简单: 给定的推理过程过于简单,缺少小模型所需的必要步骤,导致大模型依赖其强大的推理能力解决问题,而小模型无法遵循该过程得到答案。
困难: 给定的推理过程过于复杂或过于困难,导致小模型无法遵循该过程得到答案。
基于一个大模型的问题与思维链集合,我们可以将其分为简单、中等和困难三类。对于评级为中等的部分,我们予以保留。对于被评为简单和困难的数据,我们使用模型对思维链进行改进。具体来说:对于简单部分,我们扩展其推理过程,直至小模型可以遵循扩展的过程得到答案。对于评级为困难的部分,我们精简其推理过程,直至小模型可以遵循精简的过程得到答案。
我们之后对改进结果进行进一步验证,包括:对改进后的思维链再次评价难度等级,检测其是否被归类为中等难度,以及验证小模型是否能够遵循改进的思维链解决问题。如果改进后的思维链通过验证,说明改进有效,该数据可以被小模型有效理解,我们将其保留。如果验证不通过,说明改进无效,我们将返回到改进步骤,重新进行改进,直至通过验证。最终,我们获取了优化后的思维链数据集,其组成部分如下:
初始难度评级为中等的数据。
初始难度评级为简单,经过改进扩展后评为中等并通过验证的数据。
初始难度评级为困难,经过改进精简后评为中等并通过验证的数据。
此时,数据集内所有思维链的最终难度评级均为中等,意味着小模型可以有效理解数据集内的所有思维链,并能遵循这些思维链解决相应推理问题。上文提到的大小模型认知轨迹偏差问题在改进后的数据集中得到妥善解决,其可能带来的负面影响也被消除。我们使用优化后的思维链数据集对 Qwen2.5 系列基座模型进行监督微调(SFT),得到 DistilQwen2.5-R1 系列模型的基础结果。
基于多种认知轨迹数据的偏好优化
在第二阶段,我们基于第一阶段得到的不同难度等级数据对模型进行进一步提升。
具体来说,在第一阶段中,评级难度为中等的思维链数据是正确且适合小模型的思维链,小模型能够有效理解该思维链并解决问题。而难度评级为简单或困难的思维链数据依然是正确的思维链,只是不适合小模型。在此基础上,我们使用模型将正确的推理过程改写为一个错误的推理过程。错误的推理过程没有逻辑性,且会误导小模型,使得小模型完全无法遵循该错误的推理过程解决问题。
基于改写得到的错误思维链,我们将其与简单、中等和困难的思维链进行两两组合,组成多种偏好数据对。这些偏好数据对中有的偏差大,有的偏差小。基于不同种类的偏好数据对及其特点,我们分别使用针对性的参数配置,在第一阶段模型的基础上,采用 DPO 算法进一步优化小模型的推理能力。
最终,我们利用第一阶段得到的不同难度等级的认知轨迹(思维链)数据以及基础模型结果,得到了 DistilQwen2.5-R1 系列模型。
DistilQwen2.5-R1 模型效果评测
在本节中,我们从多个角度评测 DistilQwen2.5-R1 系列蒸馏小模型的实际效果;同时,我们将 DistilQwen2.5-R1 系列模型和当前业界的前沿模型对比效果。
模型综合能力评测
我们在多个模型推理能力评测基准上测试了 DistilQwen2.5-R1 系列模型的能力,涵盖数学、代码和科学问题三个主流推理领域。
在数学领域,我们使用 AIME2024 和 MATH-500 这两个基准进行测试,AIME2024 是美国数学邀请赛的 2024 年测试集,包含 30 道高难度数学题,用于评估大语言模型在复杂数学推理和问题解决能力,尤其考察代数、几何等领域的综合应用。MATH-500 是一个数学推理能力的基准测试,包含 500 个测试样本,旨在全面考察模型在数学解题上的能力。它与 AIME2024 类似,但有其独特的测试目标和对比结果,用于衡量模型在不同数学题目上的准确性。
在代码领域,我们使用 LiveCodeBench 基准,LiveCodeBench 是一个动态更新的基准测试平台,用于全面评估大型语言模型在复杂编码场景中的能力。它通过从顶级竞赛平台收集高难度编程任务来测试模型的代码生成、自我修复代码执行和测试等能力,是一个综合性、无污染的评价基准。在本次评测中,我们使用 LiveCodeBench 基准的 V2 版本,其包含 2023 年 5 月-2024 年 5 月的 511 个代码问题。
在科学问题领域,我们使用 GPQA-Diamond(Grade-Level Problems in Question Answering Diamond)基准,其由纽约大学、CohereAI 及 Anthropic 的研究人员联合发布,包含 198 条结果,是 GPQA 系列中最高质量的评测数据,用于评估模型解决专家级科学问题的能力。
如下图所示,DistilQwen2.5-R1 系列模型在 3B、7B、14B 和 32B 四个参数量级的模型中,与原始 Qwen2.5 模型的效果进行了对比。可以看出,本文描述的小型推理模型训练框架显著提升了现有语言模型的推理能力,并在多个评测基准上取得了一致而明显的效果提升。

与其他模型能力对比
为了横向比较同期发布的不同参数规模的推理模型效果,下表分别是 DistilQwen2.5-R1 系列模型在各个参数量级上与其他前沿推理模型在上文提到的 4 个基准的评测结果。我们重点对比了 DistilQwen2.5-R1 系列与 OpenThinker、DeepSeek-R1-Distill-Qwen 等系列模型。
以下是 7B 量级的对比结果,可以看出,DistilQwen2.5-R1-7B 模型超越了 Bespoke-Stratos-7B 和 OpenThinker-7B。值得注意的是,相较于 OpenThinker-7B,DistilQwen2.5-R1-7B 在使用更少训练数据的情况下在所有基准上达到了更高的结果。DeepSeek-R1-Distill-Qwen-7B 使用了 800k 闭源训练数据,而 DistilQwen2.5-R1-7B 使用了开源数据进行训练(OpenThoughts 数据集过滤和改写得到的子集),在基于开源数据模型领域内处于领先地位。

以下是 32B 量级的对比结果。同样地,DistilQwen2.5-R1-32B 在所有已知基准上超越了 Sky-T1-32B-Preview,以及在绝大多数基准上超越了 OpenThinker-32B。

模型多次推理评测
我们还测试了 DistilQwen2.5-R1 系列模型在上文提到的四个基准上多次推理的结果,模型会对同一个问题生成 k 个回答进行评测,即 Pass@k 指标。以下是 DistilQwen2.5-R1-7B 和 DistilQwen2.5-R1-32B 在四个基准上 Pass@k 结果(k=2、4、8、16、32、64)。
可以看出,随着模型推理次数 k 的逐步增加,两个模型在所有基准上的评测准确率大幅提高。值得注意的是,随着 k 的增加,DistilQwen2.5-R1-7B 在 MATH-500 和 GPQA-Diamond 上涨幅巨大,并且不断逼近 DistilQwen2.5-R1-32B 水准。这表明我们的推理模型训练框架在小模型领域内拥有巨大潜力。我们可以通过多次推理的方式使 7B 模型拥有媲美 32B 模型的能力,极大减少了推理所需的计算资源。

模型输出
对同一数学问题,我们对比了 DistilQwen2.5-R1 系列模型在 7B、32B 量级和同等量级模型的推理结果。从输出结果可以看出,DistilQwen2.5-R1 系列模型在同量级推理模型中处于领先地位。
模型下载和使用
DistilQwen2.5-R1 在阿里云人工智能平台 PAI 上的实践
以下 HuggingFace transformers 库为例,简要介绍如何在 PAI-DSW 上使用 DistilQwen2.5-R1 模型。首先需要保证 PAI-DSW 镜像内 transformers 版本大于等于 4.37.0,否则会在加载模型时报错:
以 DistilQwen2.5-R1-7B 为例,我们可以使用如下代码调用模型:
DistilQwen2.5-R1 在开源社区的下载
我们在 Hugging Face 和 Model Scope 上开源了我们蒸馏后的模型,分别为DistilQwen2.5-R1-3B、DistilQwen2.5-R1-7B、DistilQwen2.5-R1-14B、DistilQwen2.5-R1-32B。以 Hugging Face 为例,用户可以使用如下代码下载这两个模型:
小结与未来工作
本文介绍了 DistilQwen2.5-R1 系列深度推理模型,它在少量来自 DeepSeek-R1 的思维链数据基础上,通过创新蒸馏策略增强了小模型的深度思考能力。实验结果表明,该系列模型在多个基准测试中表现出色,尤其是 DistilQwen2.5-R1-7B 的性能全面超越了其他开源蒸馏模型。为了方便实际应用,这些模型的 Checkpoint 已在 Hugging Face 和 Model Scope 社区中公开,并提供了在阿里云人工智能平台 PAI 上的操作指南。在未来,随着大语言模型和知识蒸馏技术更进一步的发展,我们将推出各种领域、各种规格的 DistilQwen 系列模型,充分促进大语言模型在实际应用中的降本增效。
参考资料
相关发表论文
Yuanhao Yue, Chengyu Wang, Jun Huang, Peng Wang. Building a Family of Data Augmentation Models for Low-cost LLM Fine-tuning on the Cloud. COLING 2025
Yuanhao Yue, Chengyu Wang, Jun Huang, Peng Wang. Distilling Instruction-following Abilities of Large Language Models with Task-aware Curriculum Planning. EMNLP 2024
技术文章
DistilQwen2.5 发布:通义千问蒸馏小模型再升级:https://developer.aliyun.com/article/1653842
DistilQwen2:通义千问大模型的知识蒸馏实践:https://developer.aliyun.com/article/1633882
DistilQwen2 蒸馏小模型的训练、评测、压缩与部署实践:https://help.aliyun.com/zh/pai/user-guide/training-evaluation-compression-and-deployment-of-distilqwen2
大语言模型数据增强与模型蒸馏解决方案:https://help.aliyun.com/zh/pai/user-guide/llm-data-enhancement-and-model-distillation-solution
技术交流答疑群

评论