用扩散模型生成假图+文本对,再通过细粒度融合修复缺陷,实现零样本图像描述
Unleashing Text-to-Image Diffusion Prior for Zero-Shot Image Captioning
- 通过局部视觉-文本特征混合,修复扩散生成图像的细节错误
- 在MSCOCO和Flickr30k上超越现有方法,跨域零样本任务排名第一
- 适合做零样本图文生成、无真实图像标注场景的开发者使用
零样本图像描述近期受到关注,仅用文本数据训练。文本到图像扩散模型可生成合成图像-文本对,但合成图像显著区域存在缺陷,导致图文语义错位。为此,我们提出分块跨模态特征混合法(PCM),在训练中细粒度自适应缓解内容失真,兼容多数编码器-解码器框架,构建PCM-Net。具体地,先基于CLIP空间中图像-文本相似性检测图像显著视觉概念;随后选择性融合输入图像的分块视觉特征与显著概念的文本特征,生成含较少缺陷内容的混合特征图;最后通过视觉-语义编码器优化该特征图,并输入句子解码器生成描述。此外,设计了基于CLIP权重的交叉熵损失,优先高质量图像-文本对以提升训练效果。在MSCOCO和Flickr30k数据集上的实验表明,PCM-Net优于当前基于视觉语言模型的方法,在域内与跨域零样本任务中均排名第一。合成数据集SynthImgCap及代码已公开。
原文摘要 · Abstract (English)
Recently, zero-shot image captioning has gained increasing attention, where only text data is available for training. The remarkable progress in text-to-image diffusion model presents the potential to resolve this task by employing synthetic image-caption pairs generated by this pre-trained prior. Nonetheless, the defective details in the salient regions of the synthetic images introduce semantic misalignment between the synthetic image and text, leading to compromised results. To address this challenge, we propose a novel Patch-wise Cross-modal feature Mix-up (PCM) mechanism to adaptively mitigate the unfaithful contents in a fine-grained manner during training, which can be integrated into most of encoder-decoder frameworks, introducing our PCM-Net. Specifically, for each input image, salient visual concepts in the image are first detected considering the image-text similarity in CLIP space. Next, the patch-wise visual features of the input image are selectively fused with the textual features of the salient visual concepts, leading to a mixed-up feature map with less defective content. Finally, a visual-semantic encoder is exploited to refine the derived feature map, which is further incorporated into the sentence decoder for caption generation. Additionally, to facilitate the model training with synthetic data, a novel CLIP-weighted cross-entropy loss is devised to prioritize the high-quality image-text pairs over the low-quality counterparts. Extensive experiments on MSCOCO and Flickr30k datasets demonstrate the superiority of our PCM-Net compared with state-of-the-art VLMs-based approaches. It is noteworthy that our PCM-Net ranks first in both in-domain and cross-domain zero-shot image captioning. The synthetic dataset SynthImgCap and code are available at https://jianjieluo.github.io/SynthImgCap.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。