用检索图文联合提示增强轻量图像描述,提升生成质量。
ViPCap: Retrieval Text-Based Visual Prompts for Lightweight Image Captioning

- 用文本检索结合图像信息生成视觉提示,替代仅依赖CLIP嵌入
- 在COCO、Flickr30k、NoCaps上显著优于现有轻量模型
- 可即插即用,适合资源受限场景下的图像描述任务
近期轻量级图像描述模型多依赖检索文本作为提示,但以往方法仅将检索文本用于文本提示,视觉信息仍完全依赖CLIP视觉嵌入。这一局限导致提示中蕴含的图像描述信息未能充分反映在视觉嵌入空间中。为此,我们提出ViPCap,一种基于检索文本的视觉提示新方法。该方法将检索到的图文信息映射至CLIP空间,生成多个随机高斯分布,并通过采样探索随机增强后的分布,有效检索包含图像信息的语义特征。这些特征被整合进原图,作为视觉提示,从而提升模型捕捉相关视觉信息的能力。在COCO、Flickr30k和NoCaps数据集上的实验表明,ViPCap在效率与效果上均显著优于现有轻量级描述模型,具备良好的即插即用潜力。代码已开源:https://github.com/taewhankim/VIPCAP。
原文摘要 · Abstract (English)
Recent lightweight image captioning models using retrieved data mainly focus on text prompts. However, previous works only utilize the retrieved text as text prompts, and the visual information relies only on the CLIP visual embedding. Because of this issue, there is a limitation that the image descriptions inherent in the prompt are not sufficiently reflected in the visual embedding space. To tackle this issue, we propose ViPCap, a novel retrieval text-based visual prompt for lightweight image captioning. ViPCap leverages the retrieved text with image information as visual prompts to enhance the ability of the model to capture relevant visual information. By mapping text prompts into the CLIP space and generating multiple randomized Gaussian distributions, our method leverages sampling to explore randomly augmented distributions and effectively retrieves the semantic features that contain image information. These retrieved features are integrated into the image and designated as the visual prompt, leading to performance improvements on the datasets such as COCO, Flickr30k, and NoCaps. Experimental results demonstrate that ViPCap significantly outperforms prior lightweight captioning models in efficiency and effectiveness, demonstrating the potential for a plug-and-play solution. The source code is available at https://github.com/taewhankim/VIPCAP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。