arXiv:2505.02278cs.CV2025-05CVPR

通过实体定位增强CLIP,让图文匹配更精准。

Compositional Image-Text Matching and Retrieval by Grounding Entities

  • 用开放词汇检测器定位物体和关系,生成子图像嵌入
  • 动态调整全局图像嵌入,加权组合提升匹配精度
  • 零样本适配,显著提升图文检索性能,适合视觉推理场景

在大规模图像-文本对上进行视觉语言预训练是当前视觉语言模型的核心基础。尽管经过额外训练后,这些模型在视觉问答、图像描述和视觉常识推理等下游任务中表现优异,但像CLIP这类预训练模型存在无法进行实体定位和组合式图文匹配的明显缺陷。本文提出一种无需训练的零样本增强方法,对CLIP嵌入进行改进,具备良好的组合性。通过使用先进的开放词汇检测器定位图像中的物体和关系,分别计算子图像嵌入,并动态调整基础全局图像嵌入。最终嵌入由子图像嵌入加权组合而成,用于与文本嵌入的相似度计算,在Visual Genome和SVO Probes数据集上平均提升1.5%的图文匹配准确率。特别地,增强后的嵌入在Flickr30K和MS-COCO检索基准上表现更优,分别实现Recall@1提升12%和0.4%,达到新的领先水平。代码已公开于https://github.com/madhukarreddyvongala/GroundingCLIP。

原文摘要 · Abstract (English)

Vision-language pretraining on large datasets of images-text pairs is one of the main building blocks of current Vision-Language Models. While with additional training, these models excel in various downstream tasks, including visual question answering, image captioning, and visual commonsense reasoning. However, a notable weakness of pretrained models like CLIP, is their inability to perform entity grounding and compositional image and text matching~\cite{Jiang2024ComCLIP, yang2023amc, Rajabi2023GroundedVSR, learninglocalizeCVPR24}. In this work we propose a novel learning-free zero-shot augmentation of CLIP embeddings that has favorable compositional properties. We compute separate embeddings of sub-images of object entities and relations that are localized by the state of the art open vocabulary detectors and dynamically adjust the baseline global image embedding. % The final embedding is obtained by computing a weighted combination of the sub-image embeddings. The resulting embedding is then utilized for similarity computation with text embedding, resulting in a average 1.5\% improvement in image-text matching accuracy on the Visual Genome and SVO Probes datasets~\cite{krishna2017visualgenome, svo}. Notably, the enhanced embeddings demonstrate superior retrieval performance, thus achieving significant gains on the Flickr30K and MS-COCO retrieval benchmarks~\cite{flickr30ke, mscoco}, improving the state-of-the-art Recall@1 by 12\% and 0.4\%, respectively. Our code is available at https://github.com/madhukarreddyvongala/GroundingCLIP.

图文匹配实体定位CLIP增强零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。