arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2501.01709cs.CVcs.AI

MoVE-KD: 面向视觉语言模型的混合视觉编码器知识蒸馏

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

Peking University(北京大学);NIO(蔚来);Shanghai Jiao Tong University(上海交通大学)

Jiajun Cao, Yuan Zhang, Tao Huang, Ming Lu, Qizhe Zhang, Ruichuan An, Ningning MA, Shanghang Zhang

AI总结:

针对多视觉编码器集成到VLM计算成本高的问题,提出MoVE-KD框架,结合LoRA、MoE和注意力蒸馏策略,将多编码器专长蒸馏到单高效编码器,经LLaVA等模型验证有效。

摘要:

视觉编码器是视觉语言模型(VLM)的核心组件,每种编码器都依托不同的预训练视觉基础模型展现出独特优势。为整合这些编码器的多元能力,近期研究将多个编码器集成到单个VLM中,却导致计算成本大幅攀升。本文提出混合视觉编码器知识蒸馏(MoVE-KD)框架,这一全新框架可将多个视觉编码器的独特专长蒸馏到单个高效编码器模型中。具体而言,为缓解冲突并保留每个教师编码器的独有特性,我们采用低秩适配(LoRA)和混合专家(MoE),根据输入特征选择性激活专业知识,同时提升适应性与效率。为规范知识蒸馏过程并提升性能,我们提出一种基于注意力的蒸馏策略,可自适应地为不同编码器分配权重并突出有价值的视觉token,降低复制多个教师全面且差异化特征的负担。在LLaVA、LLaVA-NeXT等主流VLM上开展的综合实验验证了该方法的有效性。我们的代码可在以下链接获取:https://github.com/hey-cjj/MoVE-KD。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑