让生成图像与真实图像在隐空间对齐,提升视觉语言模型性能
GMAIL: Generative Modality Alignment for generated Image Learning

- 将生成图像和真实图像视为不同模态,在隐空间进行对齐
- 在图像描述、零样本检索等任务上显著提升性能
- 适用于多种视觉语言模型,特别适合大模型训练
生成模型已能合成高度逼真的图像,为机器学习训练提供丰富数据。然而,直接将生成图像当作真实图像使用,可能因真实与合成域之间的模态差异导致模式崩溃。本文提出GMAIL框架,明确将生成图像视为与真实图像不同的模态。不直接在像素空间替换,而是在同一隐空间通过多模态学习对齐两类图像。具体而言,先用跨模态对齐损失在生成图像上微调模型,再利用该对齐模型进一步训练各类视觉语言模型。实验表明,该方法在图像描述、零样本图像检索、零样本图像分类及长文本检索任务中均显著提升性能,并展现出生成数据的正向扩展趋势,对大型多模态模型LLaVA的描述能力有明显增强。
原文摘要 · Abstract (English)
Generative models have made it possible to synthesize highly realistic images, potentially providing an abundant data source for training machine learning models. Despite the advantages of these synthesizable data sources, the indiscriminate use of generated images as real images for training can even cause mode collapse due to modality discrepancies between real and synthetic domains. In this paper, we propose a novel framework for discriminative use of generated images, coined GMAIL, that explicitly treats generated images as a separate modality from real images. Instead of indiscriminately replacing real images with generated ones in the pixel space, our approach bridges the two distinct modalities in the same latent space through a multi-modal learning approach. To be specific, we first fine-tune a model exclusively on generated images using a cross-modality alignment loss and then employ this aligned model to further train various vision-language models with generated images. By aligning the two modalities, our approach effectively leverages the benefits of recent advances in generative models, thereby boosting the effectiveness of generated image learning across a range of vision-language tasks. Our framework can be easily incorporated with various vision-language models, and we demonstrate its efficacy throughout extensive experiments. For example, our framework significantly improves performance on image captioning, zero-shot image retrieval, zero-shot image classification, and long caption retrieval tasks. It also shows positive generated data scaling trends and notable enhancements in the captioning performance of the large multimodal model, LLaVA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。