arXiv:2509.00039cs.CV2025-09被引 18

通过自适应多教师蒸馏,让小模型在图像文本检索上表现接近大模型。

AMMKD: Adaptive Multimodal Multi-teacher Distillation for Lightweight Vision-Language Models

  • 用多模态融合和多教师蒸馏,压缩大模型知识到小模型。
  • 在三个数据集上,模型大小减少60%以上,性能仍优于现有轻量方案。
  • 动态加权机制缓解教师间冲突,适合移动端部署的视觉语言任务。

大规模视觉语言预训练(VLP)模型在图像文本检索任务中表现优异,但因模型过大、计算复杂,难以部署于移动设备。本文提出自适应多模态多教师知识蒸馏(AMMKD)框架,结合多模态特征融合、多教师蒸馏与自适应优化,构建轻量高效检索模型。首先通过特征融合网络提取图像与文本的判别性特征;为降低参数并提升性能,采用双CLIP教师模型进行多教师知识蒸馏,并预先计算存储文本特征作为类别向量以提高效率;通过KL散度匹配概率分布,增强师生输出对齐;最后设计自适应动态加权策略,将多教师蒸馏视为多目标优化问题,利用梯度空间多样性动态调整各教师影响,减少冲突,引导学生模型向更优方向学习。在三个基准数据集上的大量实验表明,AMMKD在显著降低模型复杂度的同时,实现更优性能,验证了其有效性和灵活性。

原文摘要 · Abstract (English)

The success of large-scale visual language pretraining (VLP) models has driven widespread adoption of image-text retrieval tasks. However, their deployment on mobile devices remains limited due to large model sizes and computational complexity. We propose Adaptive Multi-Modal Multi-Teacher Knowledge Distillation (AMMKD), a novel framework that integrates multi-modal feature fusion, multi-teacher distillation, and adaptive optimization to deliver lightweight yet effective retrieval models. Specifically, our method begins with a feature fusion network that extracts and merges discriminative features from both the image and text modalities. To reduce model parameters and further improve performance, we design a multi-teacher knowledge distillation framework to pre-train two CLIP teacher models. We decouple modalities by pre-computing and storing text features as class vectors via the teacher text encoder to enhance efficiency. To better align teacher and student outputs, we apply KL scatter for probability distribution matching. Finally, we design an adaptive dynamic weighting scheme that treats multi-teacher distillation as a multi-objective optimization problem. By leveraging gradient space diversity, we dynamically adjust the influence of each teacher, reducing conflicts and guiding the student toward more optimal learning directions. Extensive experiments on three benchmark datasets demonstrate that AMMKD achieves superior performance while significantly reducing model complexity, validating its effectiveness and flexibility.

知识蒸馏轻量化视觉语言多教师

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