通过分层教师逐步蒸馏,让小模型高效学习大模型的视觉语言能力。
LLaVA-CKD: Bottom-Up Cascaded Knowledge Distillation for Vision-Language Models

- 设计多级教师递进式蒸馏,避免大容量差距导致的知识丢失。
- 在7个VQA基准上达到当前最佳性能,小模型效果逼近大模型。
- 适合资源受限场景下部署高性能视觉语言模型的人群。
大型视觉语言模型(VLMs)在视觉问答(VQA)等任务中表现优异,但其内存与计算开销限制了实际部署。知识蒸馏是一种有效缓解该问题的技术,将高容量教师网络的知识迁移到更小的学生网络。然而,师生间容量差距过大反而会降低知识迁移效果。为此,我们提出一种自底向上的级联知识蒸馏(CKD)框架:借鉴人类教育体系,引入一个或多个中间容量的教师,逐级引导学生网络提升能力,使后续更高容量的教师能更有效地接手。我们进行了理论分析,研究级联蒸馏对学生泛化性能的影响。在基于LLaVA方法构建的模型上应用该框架,并在7个公开标准VQA基准上评估,结果表明所获模型性能达到当前最优水平。
原文摘要 · Abstract (English)
Large Vision-Language Models (VLMs) are successful in addressing a multitude of vision-language understanding tasks, such as Visual Question Answering (VQA), but their memory and compute requirements remain a concern for practical deployment. A promising class of techniques for mitigating this concern is Knowledge Distillation, where knowledge from a high-capacity Teacher network is transferred to a considerably smaller Student network. However, the capacity gap between the two networks is both a blessing and a curse: the smaller the Student network, the better its efficiency, and the larger the Teacher, the more knowledge it carries; yet, beyond a point, the larger capacity gap between the two leads to worse knowledge transfer. To counter this effect, we propose a bottom-up cascaded knowledge distillation (CKD) framework. Instead of treating knowledge transfer as an activity involving one high-capacity Teacher (or an ensemble of such), inspired by human formal education systems, we introduce one (potentially, more) additional Teacher(s) of intermediate capacity that gradually bring the Student network to the next level, where the next (higher-capacity) Teacher can take over. We provide a theoretical analysis in order to study the effect of cascaded distillation in the generalization performance of the Student. We apply the proposed framework on models build upon the LLaVA methodology and evaluate the derived models on seven standard, publicly available VQA benchmarks, demonstrating their SotA performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。