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

## TL;DR

AMMKD introduces an adaptive multi-modal multi-teacher distillation framework that effectively reduces model size and complexity for vision-language retrieval tasks, maintaining high performance on benchmarks.

## Contribution

The paper presents a novel framework combining multi-modal feature fusion, multi-teacher distillation, and adaptive optimization for lightweight vision-language models.

## Key findings

- Achieves superior retrieval performance with reduced model complexity.
- Effectively aligns teacher-student outputs using KL scatter.
- Demonstrates flexibility and effectiveness across multiple datasets.

## Abstract

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.