用想象的图像代理提升零样本图像检索精度
Imagine and Seek: Improving Composed Image Retrieval with an Imagined Proxy
- 用大模型生成与图文描述对齐的虚拟图像作为查询代理
- 在CIRR数据集上Recall@10达70.07,刷新SOTA
- 无需训练,适合快速部署于跨模态检索场景
零样本组合图像检索(ZSCIR)需同时匹配查询图像和相对文本描述。现有方法将查询图像投影至文本特征空间,再与文本特征融合进行检索,但因图像与文本间存在天然鸿沟,难以保证细节对齐。本文提出无需训练的IP-CIR方法,通过大语言模型生成与查询图像和文本描述对齐的虚拟图像代理,增强查询表征。利用查询文本与图像对代理图像进行条件生成,并融合代理图像、查询图像及文本语义扰动,构建鲁棒查询特征。新提出的平衡度量融合文本与代理检索相似性,更精准定位目标图像,同时引入图像侧信息。在三个公开数据集上的实验表明,本方法显著提升检索性能:在CIRR上达到Recall@10为70.07,优于现有方法;FashionIQ上从45.11提升至45.74;CIRCO上mAPK@10由32.24升至34.26。
原文摘要 · Abstract (English)
The Zero-shot Composed Image Retrieval (ZSCIR) requires retrieving images that match the query image and the relative captions. Current methods focus on projecting the query image into the text feature space, subsequently combining them with features of query texts for retrieval. However, retrieving images only with the text features cannot guarantee detailed alignment due to the natural gap between images and text. In this paper, we introduce Imagined Proxy for CIR (IP-CIR), a training-free method that creates a proxy image aligned with the query image and text description, enhancing query representation in the retrieval process. We first leverage the large language model's generalization capability to generate an image layout, and then apply both the query text and image for conditional generation. The robust query features are enhanced by merging the proxy image, query image, and text semantic perturbation. Our newly proposed balancing metric integrates text-based and proxy retrieval similarities, allowing for more accurate retrieval of the target image while incorporating image-side information into the process. Experiments on three public datasets demonstrate that our method significantly improves retrieval performances. We achieve state-of-the-art (SOTA) results on the CIRR dataset with a Recall@K of 70.07 at K=10. Additionally, we achieved an improvement in Recall@10 on the FashionIQ dataset, rising from 45.11 to 45.74, and improved the baseline performance in CIRCO with a mAPK@10 score, increasing from 32.24 to 34.26.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。