arXiv:2501.01709cs.CVcs.AI2025-01CVPR被引 38

用混合视觉编码器知识蒸馏,让小模型学会多个大模型的特长。

MoVE-KD: Knowledge Distillation for VLMs with Mixture of Visual Encoders

  • 通过低秩适配和专家混合机制,按输入激活不同编码器专长。
  • 在LLaVA等模型上实现更高性能,同时降低计算开销。
  • 适合需要高效视觉语言模型的应用场景,如移动端部署。

视觉编码器是视觉语言模型(VLMs)的核心组件,各自基于不同的预训练视觉基础模型展现出独特优势。为充分利用多种编码器的能力,近期研究将多个编码器集成到单一VLM中,但导致计算成本显著上升。本文提出一种新的混合视觉编码器知识蒸馏框架(MoVE-KD),将多个视觉编码器的独特能力蒸馏到一个高效单编码器模型中。为缓解编码器间的冲突并保留其特性,我们采用低秩适配(LoRA)与专家混合(MoEs)机制,根据输入特征选择性激活特定知识,提升模型适应性与效率。为进一步规范蒸馏过程并增强性能,我们设计了一种基于注意力的蒸馏策略,自适应加权不同教师编码器,并强调关键视觉标记,减轻复制多教师复杂特征的负担。在LLaVA和LLaVA-NeXT等主流VLM上的大量实验验证了该方法的有效性。代码已开源:https://github.com/hey-cjj/MoVE-KD。

原文摘要 · Abstract (English)

Visual encoders are fundamental components in vision-language models (VLMs), each showcasing unique strengths derived from various pre-trained visual foundation models. To leverage the various capabilities of these encoders, recent studies incorporate multiple encoders within a single VLM, leading to a considerable increase in computational cost. In this paper, we present Mixture-of-Visual-Encoder Knowledge Distillation (MoVE-KD), a novel framework that distills the unique proficiencies of multiple vision encoders into a single, efficient encoder model. Specifically, to mitigate conflicts and retain the unique characteristics of each teacher encoder, we employ low-rank adaptation (LoRA) and mixture-of-experts (MoEs) to selectively activate specialized knowledge based on input features, enhancing both adaptability and efficiency. To regularize the KD process and enhance performance, we propose an attention-based distillation strategy that adaptively weighs the different encoders and emphasizes valuable visual tokens, reducing the burden of replicating comprehensive but distinct features from multiple teachers. Comprehensive experiments on popular VLMs, such as LLaVA and LLaVA-NeXT, validate the effectiveness of our method. Our code is available at: https://github.com/hey-cjj/MoVE-KD.

知识蒸馏视觉编码器多模态高效模型

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