解决视觉令牌不平衡问题,提升高效多模态模型的感知能力
EM-KD: Distilling Efficient Multimodal Large Language Model with Unbalanced Vision Tokens
- 用匈牙利匹配对齐师生模型视觉令牌空间
- 提出两种蒸馏策略,显著提升准确率与效率
- 适合关注多模态模型压缩与性能优化的研究者
高效多模态大模型通过压缩视觉令牌降低资源消耗,但会损失视觉信息导致理解能力下降。现有知识蒸馏方法忽视了高效学生模型与原始教师模型之间因视觉令牌数量不均导致的细粒度理解差异。本文提出EM-KD,一种新型蒸馏范式。首先计算教师与学生视觉逻辑值间的曼哈顿距离,并用匈牙利算法在空间维度上对齐;随后引入两种蒸馏策略:1)视觉-语言亲和性蒸馏(VLAD),通过最小化学生与教师亲和矩阵的平滑L1距离实现对齐;2)视觉语义蒸馏(VSD),利用反KL散度衡量最终层视觉逻辑值在词汇空间上的离散概率分布。在多个基准测试中,经EM-KD训练的模型在准确率与效率上均大幅超越现有高效多模态模型。与采用相同视觉令牌匹配策略的对比方法相比,EM-KD仍表现更优。
原文摘要 · Abstract (English)
Efficient Multimodal Large Language Models (MLLMs) compress vision tokens to reduce resource consumption, but the loss of visual information can degrade comprehension capabilities. Although some priors introduce Knowledge Distillation to enhance student models, they overlook the fundamental differences in fine-grained vision comprehension caused by unbalanced vision tokens between the efficient student and vanilla teacher. In this paper, we propose EM-KD, a novel paradigm that enhances the Efficient MLLMs with Knowledge Distillation. To overcome the challenge of unbalanced vision tokens, we first calculate the Manhattan distance between the vision logits of teacher and student, and then align them in the spatial dimension with the Hungarian matching algorithm. After alignment, EM-KD introduces two distillation strategies: 1) Vision-Language Affinity Distillation (VLAD) and 2) Vision Semantic Distillation (VSD). Specifically, VLAD calculates the affinity matrix between text tokens and aligned vision tokens, and minimizes the smooth L1 distance of the student and the teacher affinity matrices. Considering the semantic richness of vision logits in the final layer, VSD employs the reverse KL divergence to measure the discrete probability distributions of the aligned vision logits over the vocabulary space. Comprehensive evaluation on diverse benchmarks demonstrates that EM-KD trained model outperforms prior Efficient MLLMs on both accuracy and efficiency with a large margin, validating its effectiveness. Compared with previous distillation methods, which are equipped with our proposed vision token matching strategy for fair comparison, EM-KD also achieves better performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。