arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2509.00039cs.CV

AMMKD:面向轻量级视觉-语言模型的自适应多模态多教师蒸馏

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

Yuqi Li, Chuanguang Yang, Junhao Dong, Zhengtao Yao, Haoyan Xu, Zeyu Dong, Hansheng Zeng, Zhulin An, Yingli Tian

AI总结:

针对大规模VLP模型难以部署在移动设备的问题,提出融合多模态特征融合、多教师蒸馏与自适应优化的AMMKD框架,在降低模型复杂度的同时提升了图文检索性能。

摘要:

大规模视觉语言预训练(VLP)模型的成功推动了图文检索任务的广泛应用,但这类模型因参数量大、计算复杂度高,在移动设备上的部署仍受限制。本文提出自适应多模态多教师知识蒸馏框架(AMMKD),该框架融合多模态特征融合、多教师蒸馏与自适应优化技术,旨在构建轻量且高效的检索模型。具体而言,该方法首先通过特征融合网络提取并融合图像与文本模态的判别性特征。为减少模型参数量并进一步提升性能,我们设计了多教师知识蒸馏框架,对两个CLIP教师模型进行预训练;通过教师文本编码器预计算并存储文本特征作为类别向量,实现模态解耦以提升效率。为更好地对齐教师与学生模型的输出,我们采用KL散度进行概率分布匹配。最后,我们设计了自适应动态加权策略,将多教师蒸馏视为多目标优化问题,利用梯度空间多样性动态调整每个教师的影响权重,减少冲突并引导学生模型向更优的学习方向训练。在三个基准数据集上的大量实验表明,AMMKD在显著降低模型复杂度的同时取得了更优的性能,验证了其有效性与灵活性。

英文摘要:

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.

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

↑