Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Jongoh Jeong, Hoyong Kwon, Minseok Kim, Kuk-Jin Yoon• 2026

Related benchmarks

TaskDatasetResultRank
Text-to-Image RetrievalFlickr30K--
559
Image-to-Text RetrievalCOCO
R@14.9
152
Image-Text RetrievalFlickr8K
Mean Score26.2
75
Image-Text RetrievalCOCO
Image Retrieval @ 1 (IR@1)3.7
6
Image-Text RetrievalFlickr30K
IR@1 Recall10
5
Image-to-Text RetrievalFlickr30K
TR@113.7
4
Showing 6 of 6 rows

Other info

Follow for update