arXiv:2412.01282cs.CVcs.AI2024-12CVPR被引 32

通过蒸馏跨模态对齐知识,提升移动端视觉语言模型性能。

Align-KD: Distilling Cross-Modal Alignment Knowledge for Mobile Vision-Language Model Enhancement

  • 在浅层引导学生模型学习视觉与文本的跨模态匹配。
  • 1.7B学生模型从7B教师模型获益,6个基准平均分提升2.0。
  • 适合移动端多模态模型优化,尤其关注小模型性能提升。

视觉语言模型(VLMs)为多模态任务带来强大的理解与推理能力。与此同时,移动设备上人工智能应用的需求日益增长,例如AI助手软件。一些研究尝试将VLM迁移到边缘设备以扩展其应用范围。简化模型结构是常用方法,但随着模型缩小,性能与尺寸之间的权衡愈发困难。知识蒸馏(KD)可在不增加模型大小或数据量的情况下提升模型综合能力。然而,现有大模型蒸馏技术大多仅适用于单模态大语言模型,或仅利用教师模型生成新的数据环境,未考虑在VLM中最重要的跨模态对齐知识的蒸馏。我们提出Align-KD方法,指导学生模型在浅层学习跨模态匹配,并基于文本关注点学习视觉标记到文本嵌入空间的投影。在Align-KD引导下,1.7B的MobileVLM V2模型可从7B教师模型中学习丰富知识,采用轻量级训练损失,在两个训练子集上分别实现6个基准平均分提升2.0。代码已公开于:https://github.com/fqhank/Align-KD。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) bring powerful understanding and reasoning capabilities to multimodal tasks. Meanwhile, the great need for capable aritificial intelligence on mobile devices also arises, such as the AI assistant software. Some efforts try to migrate VLMs to edge devices to expand their application scope. Simplifying the model structure is a common method, but as the model shrinks, the trade-off between performance and size becomes more and more difficult. Knowledge distillation (KD) can help models improve comprehensive capabilities without increasing size or data volume. However, most of the existing large model distillation techniques only consider applications on single-modal LLMs, or only use teachers to create new data environments for students. None of these methods take into account the distillation of the most important cross-modal alignment knowledge in VLMs. We propose a method called Align-KD to guide the student model to learn the cross-modal matching that occurs at the shallow layer. The teacher also helps student learn the projection of vision token into text embedding space based on the focus of text. Under the guidance of Align-KD, the 1.7B MobileVLM V2 model can learn rich knowledge from the 7B teacher model with light design of training loss, and achieve an average score improvement of 2.0 across 6 benchmarks under two training subsets respectively. Code is available at: https://github.com/fqhank/Align-KD.

视觉语言模型知识蒸馏移动端

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。