知识蒸馏(Knowledge Distillation, KD)是一种非常流行的技术,用于让大型深度学习模型(通常被称为“教师模型”)的能力,传递到一个较小的模型(称为“学生模型”)中。这种方法的核心目标是让学生模型像教师模型一样,能够做出相似的判断,但它的计算量要小得多,因此能够更快、更高效地运行。

可以把这个过程想象成“学生向老师请教”的场景。教师模型通过训练得到非常强大的能力,而学生模型虽然小,但通过从教师模型那里学习,可以获得接近的效果。这样做的好处就是,学生模型不仅能够在计算上更加节省资源,还能保留大模型的一些优势,适应一些资源受限的设备和场景。

知识蒸馏可以分为两种方式:

  1. 白盒知识蒸馏:这种方式要求学生模型能够访问教师模型的内部结构,比如参数和梯度。这种方式的优势是,学生模型可以更深入地理解教师模型的学习过程,从而更高效地进行知识迁移。
  2. 黑盒知识蒸馏:在这种方式下,学生模型无法直接看到教师模型的内部结构,而是通过教师模型的输入输出对其进行学习。即使不能了解教师模型的“心思”,学生模型也可以通过观察输入和输出之间的关系,来模仿教师模型的表现。

通过这种方式,知识蒸馏能够让学生模型在不需要大量计算资源的情况下,尽可能地复现教师模型的强大能力。这使得它在许多实际应用中非常有用,尤其是在移动设备或嵌入式系统中,资源有限但又需要较强模型性能的场景。

在这里插入图片描述

1. 白盒知识蒸馏

白盒知识蒸馏(White-box Knowledge Distillation)是一种通过深入了解教师模型(即大模型)内部细节来指导学生模型学习的方法。在这种方法中,学生模型可以访问教师模型的各种信息,包括其内部参数和梯度,因此可以采取多种策略来优化学生模型的学习过程。

在知识蒸馏的标准方法中,目标是通过最小化教师模型分布 pT(y∣x)和学生模型分布 pS(y∣x)之间的Kullback-Leibler散度(KLD),即 KL[pT∥pS],来促使学生模型的输出分布尽可能覆盖教师模型的高概率区域。这意味着,学生模型在预测时应该尽量模仿教师模型的行为,尤其是在高概率区域。

对于像文本分类任务这样输出空间较小的任务,最小化正向KLD是有效的,因为在这种情况下,教师和学生的高概率区域往往是有限的。因此,学生模型可以有效学习到教师模型的高概率区域。

然而,在开放式文本生成任务中(如大语言模型的应用),输出空间更加复杂,教师模型的高概率区域往往比学生模型能够覆盖的区域要多得多。在这种情况下,最小化正向KLD可能导致学生模型在其无法有效学习的区域赋予过高的概率,从而生成一些教师模型几乎不会生成的内容。这种现象被称为空白区域问题(void region)。

为了解决这个问题,MiniLLM提出了一种改进的方法,即将正向KLD替换为反向KLD。这种方法能够引导学生模型关注教师模型的主要高概率区域,同时避免学生模型在教师模型的空白区域中赋予过高的概率。这在大语言模型的文本生成任务中尤为重要,能够确保生成的内容更准确且更符合实际应用的需求。

在优化反向KLD时,MiniLLM使用了策略梯度法(Policy Gradient)来推导目标函数的梯度,并通过以下几种方式进一步稳定和加速训练:

  • 单步分解:降低方差
  • 教师混合采样:缓解奖励操控问题
  • 长度归一化:消除长度偏差

此外,on-policy KD(同策略知识蒸馏)是一种将知识蒸馏与模仿学习相结合的方法。在这种方法中,学生模型基于教师模型的输出分布生成自己的输出序列,并针对其自生成输出中的错误部分获得特定的反馈。这种反馈循环类似于强化学习中的奖励反馈,有助于减少训练和推理分布的不匹配。

为了进一步优化蒸馏过程,Generalized KD (GKD) 提出了更加通用的方案。GKD允许根据具体任务灵活选择优化目标和数据来源,结合了固定数据集(例如教师生成的序列或带标签的真实数据)与学生模型生成的同策略数据,以实现更高效的学习。

此外,TED提出了一种任务感知的逐层知识蒸馏方法,它通过在每一层添加特定任务的过滤器来对学生模型进行指导。每个过滤器先在教师模型上进行训练,然后冻结教师模型的过滤器,指导学生模型对齐输出特征。

MiniMoE采用了专家混合(Mixture-of-Experts, MoE)模型作为学生模型,进一步缩小学生模型和教师模型之间的能力差距。

KPTD则提出了一种通过将实体定义的知识转移到大语言模型的参数中的方法。这种方法基于实体定义生成转移集,并通过这些定义指导学生模型的学习,使其输出分布更接近教师模型。

这种类型的知识蒸馏方法,不仅通过减少学生模型的计算负担,还能够提高学生模型在复杂任务中的表现,尤其在需要高效和高可靠性的大语言模型生成任务中具有重要应用价值。

2. 黑盒知识蒸馏

黑盒知识蒸馏(Black-box Knowledge Distillation)是一种在无法访问大模型内部细节的情况下,通过教师模型的输出(例如分类概率或生成文本)来指导学生模型的学习的方法。与白盒知识蒸馏不同,黑盒方法不要求学生模型直接接触教师模型的内部参数,而是通过模仿教师模型的输出分布,来逼近其行为。这意味着,学生模型可以通过学习教师模型生成的答案(如分类结果或文本生成),从而获得类似的能力,达到性能压缩和迁移的效果。

在这里插入图片描述

关键目标:

黑盒知识蒸馏的核心目标是在没有大模型内部参数的帮助下,学生模型能通过模仿教师模型的输出,学习如何完成特定任务。例如,学生模型可以在大语言模型的指导下,学习如何更好地进行上下文学习(ICL)、思维链推理(CoT)或指令跟随(IF)等任务。

例如:TAPIR框架

TAPIR(Task-Aware Curriculum Planning for Instruction Refinement)是一种黑盒蒸馏的框架,它通过精心设计的多任务课程规划来提升学生模型的能力。TAPIR框架的主要思想是,从教师模型中选取那些学生模型较难理解的指令,并通过难度重采样的方式,提升学生模型对这些复杂任务的学习效果。

具体步骤包括:

  1. 初始化学生模型:从一个预训练的学生模型开始。
  2. 难度筛选:利用开源指令数据集(如 Alpaca 数据集),根据学生模型难以拟合的指令挑选出适合的种子数据集。
  3. 多任务规划:根据任务的难度和类型,使用教师模型(如 ChatGPT)生成更多类似难度的指令-响应对,以提升推理类任务的学习效果。
  4. 回答风格增强:通过调整教师模型的回答风格(如思维链或代码注释)来帮助学生模型更好地理解任务。
  5. 多轮优化:通过反馈循环,不断评估学生模型的输出,并用裁判模型对其质量进行打分,逐步引导学生模型提高。

在TAPIR框架中,模型拟合难度(MFD)指标被用来衡量学生模型对特定指令的学习能力。MFD分数是通过评估学生模型生成的答案与教师模型生成的答案之间的差异来计算的。只有那些MFD分数较高的样本对,才会被用于进一步的训练,帮助学生模型逐步提升其任务能力。

Distilling Step-by-Step

另一种方法是Distilling Step-by-Step,这种方法包括两个主要步骤:

  1. 生成标签和推理依据:教师模型基于无标签数据集生成预测标签,并且还生成推理依据,解释为什么这个标签是正确的。推理依据是对标签背后原因的自然语言解释,帮助学生理解如何从输入推导出输出。
  2. 学生模型训练:学生模型不仅仅学习任务标签,还学习推理依据。这使得学生模型能够理解背后的逻辑,并能更好地处理复杂任务。

这种方法通过提供更多的上下文和解释,帮助学生模型不仅学习到正确的答案,还能理解为什么这个答案是对的,从而提升其泛化能力。

总结

知识蒸馏通过两种方式:白盒和黑盒。将教师模型的知识转移给学生模型。白盒知识蒸馏依赖于直接访问教师模型的内部细节,适合用于模型结构优化;黑盒知识蒸馏则通过模仿教师模型的输出,引导学生模型学习,适用于无法访问教师模型内部信息的场景。无论哪种方式,知识蒸馏都能够有效地压缩模型,同时保持较高的性能,在许多应用中具有重要价值,尤其是在大语言模型和复杂任务处理方面。

普通人如何抓住AI大模型的风口?

领取方式在文末

为什么要学习大模型?

目前AI大模型的技术岗位与能力培养随着人工智能技术的迅速发展和应用 , 大模型作为其中的重要组成部分 , 正逐渐成为推动人工智能发展的重要引擎 。大模型以其强大的数据处理和模式识别能力, 广泛应用于自然语言处理 、计算机视觉 、 智能推荐等领域 ,为各行各业带来了革命性的改变和机遇 。

目前,开源人工智能大模型已应用于医疗、政务、法律、汽车、娱乐、金融、互联网、教育、制造业、企业服务等多个场景,其中,应用于金融、企业服务、制造业和法律领域的大模型在本次调研中占比超过 30%。
在这里插入图片描述

随着AI大模型技术的迅速发展,相关岗位的需求也日益增加。大模型产业链催生了一批高薪新职业:
在这里插入图片描述

人工智能大潮已来,不加入就可能被淘汰。如果你是技术人,尤其是互联网从业者,现在就开始学习AI大模型技术,真的是给你的人生一个重要建议!

最后

只要你真心想学习AI大模型技术,这份精心整理的学习资料我愿意无偿分享给你,但是想学技术去乱搞的人别来找我!

在当前这个人工智能高速发展的时代,AI大模型正在深刻改变各行各业。我国对高水平AI人才的需求也日益增长,真正懂技术、能落地的人才依旧紧缺。我也希望通过这份资料,能够帮助更多有志于AI领域的朋友入门并深入学习。

真诚无偿分享!!!
vx扫描下方二维码即可
加上后会一个个给大家发

在这里插入图片描述

大模型全套学习资料展示

自我们与MoPaaS魔泊云合作以来,我们不断打磨课程体系与技术内容,在细节上精益求精,同时在技术层面也新增了许多前沿且实用的内容,力求为大家带来更系统、更实战、更落地的大模型学习体验。

图片

希望这份系统、实用的大模型学习路径,能够帮助你从零入门,进阶到实战,真正掌握AI时代的核心技能!

01 教学内容

图片

  • 从零到精通完整闭环:【基础理论 →RAG开发 → Agent设计 → 模型微调与私有化部署调→热门技术】5大模块,内容比传统教材更贴近企业实战!

  • 大量真实项目案例: 带你亲自上手搞数据清洗、模型调优这些硬核操作,把课本知识变成真本事‌!

02适学人群

应届毕业生‌: 无工作经验但想要系统学习AI大模型技术,期待通过实战项目掌握核心技术。

零基础转型‌: 非技术背景但关注AI应用场景,计划通过低代码工具实现“AI+行业”跨界‌。

业务赋能突破瓶颈: 传统开发者(Java/前端等)学习Transformer架构与LangChain框架,向AI全栈工程师转型‌。

image.png

vx扫描下方二维码即可
在这里插入图片描述

本教程比较珍贵,仅限大家自行学习,不要传播!更严禁商用!

03 入门到进阶学习路线图

大模型学习路线图,整体分为5个大的阶段:
图片

04 视频和书籍PDF合集

图片

从0到掌握主流大模型技术视频教程(涵盖模型训练、微调、RAG、LangChain、Agent开发等实战方向)

图片

新手必备的大模型学习PDF书单来了!全是硬核知识,帮你少走弯路(不吹牛,真有用)
图片

05 行业报告+白皮书合集

收集70+报告与白皮书,了解行业最新动态!
图片

06 90+份面试题/经验

AI大模型岗位面试经验总结(谁学技术不是为了赚$呢,找个好的岗位很重要)图片
在这里插入图片描述

07 deepseek部署包+技巧大全

在这里插入图片描述

由于篇幅有限

只展示部分资料

并且还在持续更新中…

真诚无偿分享!!!
vx扫描下方二维码即可
加上后会一个个给大家发

在这里插入图片描述

Logo

中国智能体开发者社区,聚焦智能体与大模型开发,提供前沿资讯、实用工具链、开源项目及行业案例。通过技术沙龙、开发者大赛等活动,促进经验交流与协作,助力开发者快速构建创新智能应用。

更多推荐