Multimodal Distribution Matching for Vision-Language Dataset Distillation
About
Dataset distillation compresses large training sets into compact synthetic datasets while preserving downstream performance. As modern systems increasingly operate on paired vision-language inputs, multimodal distillation must preserve representation quality and cross-modal alignment under tight compute and memory budgets, yet prior methods often require heavy computes and overlook their correlations. To address this, we present Multimodal Distribution Matching (MDM), a geometry-aware framework for efficient and generalizable multimodal distillation. Specifically, MDM integrates complementary components at the data, model, and loss levels. At the data level, it initializes synthetic image-text pairs by sampling from clusters in the joint embedding space. At the model level, it forms a mixed teacher by interpolating independently fine-tuned models in weight space according to their angular deviation from the pretrained anchor. At the loss level, it matches joint distributions on the unit hypersphere using a geometry-aware matching objective that exploits the joint features in the cross-modal agreement and discrepancy directions along with symmetric contrastive learning. Across image-text retrieval benchmarks with cross-architecture evaluation, MDM yields compact synthetic sets that preserve multimodal semantics, substantially reduce distillation cost, and remain robust across architectures.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Retrieval | Flickr30K | -- | 559 | |
| Image-to-Text Retrieval | COCO | R@14.9 | 152 | |
| Image-Text Retrieval | Flickr8K | Mean Score26.2 | 75 | |
| Image-Text Retrieval | COCO | Image Retrieval @ 1 (IR@1)3.7 | 6 | |
| Image-Text Retrieval | Flickr30K | IR@1 Recall10 | 5 | |
| Image-to-Text Retrieval | Flickr30K | TR@113.7 | 4 |