中文

Multimodal Distribution Matching for Vision-Language Dataset Distillation

计算机视觉与模式识别 2026-05-25 v1 人工智能

摘要

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.

关键词

引用

@article{arxiv.2605.23482,
  title  = {Multimodal Distribution Matching for Vision-Language Dataset Distillation},
  author = {Jongoh Jeong and Hoyong Kwon and Minseok Kim and Kuk-Jin Yoon},
  journal= {arXiv preprint arXiv:2605.23482},
  year   = {2026}
}

备注

Accepted for publication at CVPR 2026. Project Page: https://andyj1.github.io/mdm