用生成模型加速多模态数据蒸馏,18倍提速且效果更优。
Efficient Multimodal Dataset Distillation via Generative Models
- 引入双向对比损失与多样性损失,提升生成图像与文本关联性。
- 在Flickr30K等数据集上实现比现有方法快18倍的蒸馏速度。
- 适合需要高效训练多模态模型的研究者与工业应用。
数据蒸馏旨在从大规模数据集中合成小规模数据集,使模型在该小数据集上训练后仍能良好表现于原始数据集。随着大语言模型与多模态大语言模型的发展,尤其是图文数据集的重要性日益凸显。然而,现有多模态数据蒸馏方法受限于匹配训练轨迹算法(Matching Training Trajectories),计算资源需求高,耗时可达数天。本文提出EDGE,一种基于生成模型的高效多模态数据蒸馏方法。针对生成图像与文本关联弱、样本多样性不足两大挑战,我们设计了双向对比损失与多样性损失相结合的训练流程,并提出一种文本生成策略,通过引入更多文本信息提升图文检索性能。在Flickr30K、COCO和CC3M数据集上的实验表明,本方法在性能与效率上均优于现有方法,最快可达当前最优方法的18倍加速。
原文摘要 · Abstract (English)
Dataset distillation aims to synthesize a small dataset from a large dataset, enabling the model trained on it to perform well on the original dataset. With the blooming of large language models and multimodal large language models, the importance of multimodal datasets, particularly image-text datasets, has grown significantly. However, existing multimodal dataset distillation methods are constrained by the Matching Training Trajectories algorithm, which significantly increases the computing resource requirement, and takes days to process the distillation. In this work, we introduce EDGE, a generative distillation method for efficient multimodal dataset distillation. Specifically, we identify two key challenges of distilling multimodal datasets with generative models: 1) The lack of correlation between generated images and captions. 2) The lack of diversity among generated samples. To address the aforementioned issues, we propose a novel generative model training workflow with a bi-directional contrastive loss and a diversity loss. Furthermore, we propose a caption synthesis strategy to further improve text-to-image retrieval performance by introducing more text information. Our method is evaluated on Flickr30K, COCO, and CC3M datasets, demonstrating superior performance and efficiency compared to existing approaches. Notably, our method achieves results 18x faster than the state-of-the-art method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。