系统梳理知识蒸馏方法,助力大模型轻量化部署
A Comprehensive Survey on Knowledge Distillation
- 按知识来源、架构、算法等维度分类归纳蒸馏技术
- 涵盖扩散模型、3D数据、大语言模型等前沿应用
- 适合关注模型压缩与高效推理的研究者和工程师
深度神经网络在计算机视觉与自然语言处理领域取得了显著成果,但随着参数量激增的Transformer模型及大规模基础模型(如视觉-语言模型VLMs、大语言模型LLMs)的发展,将其部署于边缘设备面临高运行时与内存消耗的挑战。知识蒸馏(KD)通过教师-学生架构,使轻量级学生模型学习大型教师模型的知识,成为缓解该问题的重要技术。本文系统综述了知识蒸馏方法,从知识源、蒸馏方案、算法、模态、应用及方法对比等多个维度展开分析。区别于多数陈旧或简单更新的综述,本工作以全新视角与结构,深入探讨了扩散模型、3D输入、基础模型、Transformer及LLMs等关键子领域的最新进展。同时讨论现有挑战与未来研究方向。项目代码开源:https://github.com/IPL-Sharif/KD_Survey
原文摘要 · Abstract (English)
Deep Neural Networks (DNNs) have achieved notable performance in the fields of computer vision and natural language processing with various applications in both academia and industry. However, with recent advancements in DNNs and transformer models with a tremendous number of parameters, deploying these large models on edge devices causes serious issues such as high runtime and memory consumption. This is especially concerning with the recent large-scale foundation models, Vision-Language Models (VLMs), and Large Language Models (LLMs). Knowledge Distillation (KD) is one of the prominent techniques proposed to address the aforementioned problems using a teacher-student architecture. More specifically, a lightweight student model is trained using additional knowledge from a cumbersome teacher model. In this work, a comprehensive survey of knowledge distillation methods is proposed. This includes reviewing KD from different aspects: distillation sources, distillation schemes, distillation algorithms, distillation by modalities, applications of distillation, and comparison among existing methods. In contrast to most existing surveys, which are either outdated or simply update former surveys, this work proposes a comprehensive survey with a new point of view and representation structure that categorizes and investigates the most recent methods in knowledge distillation. This survey considers various critically important subcategories, including KD for diffusion models, 3D inputs, foundational models, transformers, and LLMs. Furthermore, existing challenges in KD and possible future research directions are discussed. Github page of the project: https://github.com/IPL-Sharif/KD_Survey
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。