arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.16005cs.CV

知识增强的双流零样本组合图像检索

Knowledge-Enhanced Dual-stream Zero-shot Composed Image Retrieval

Zhejiang University(浙江大学)

Yucheng Suo, Fan Ma, Linchao Zhu, Yi Yang

AI总结:

本文提出知识增强的双流零样本组合图像检索框架KEDs,通过引入数据库隐式建模参考图像的细粒度属性,并利用伪三元组将伪词token与文本概念显式对齐,在多个基准测试上优于现有方法。

摘要:

我们研究零样本组合图像检索(ZS-CIR)任务,即在未对三元组数据集进行训练的情况下,给定参考图像和描述来检索目标图像。先前的工作通过将参考图像特征投影到文本嵌入空间来生成伪词token。然而,它们专注于全局视觉表示,忽略了详细属性的表示,例如颜色、对象数量和布局。为了应对这一挑战,我们提出了一种知识增强的双流零样本组合图像检索框架(KEDs)。KEDs通过引入数据库隐式地建模参考图像的属性。该数据库通过提供相关图像和标题来丰富伪词token,强调各个方面的共享属性信息。通过这种方式,KEDs从多个视角识别参考图像。此外,KEDs采用额外的流,利用从图像-文本对中挖掘的伪三元组将伪词token与文本概念对齐。该流中生成的伪词token在文本嵌入空间中与细粒度语义显式对齐。在广泛使用的基准测试(即ImageNet-R、COCO object、Fashion-IQ和CIRR)上的大量实验表明,KEDs优于先前的零样本组合图像检索方法。

英文摘要:

We study the zero-shot Composed Image Retrieval (ZS-CIR) task, which is to retrieve the target image given a reference image and a description without training on the triplet datasets. Previous works generate pseudo-word tokens by projecting the reference image features to the text embedding space. However, they focus on the global visual representation, ignoring the representation of detailed attributes, e.g., color, object number and layout. To address this challenge, we propose a Knowledge-Enhanced Dual-stream zero-shot composed image retrieval framework (KEDs). KEDs implicitly models the attributes of the reference images by incorporating a database. The database enriches the pseudo-word tokens by providing relevant images and captions, emphasizing shared attribute information in various aspects. In this way, KEDs recognizes the reference image from diverse perspectives. Moreover, KEDs adopts an extra stream that aligns pseudo-word tokens with textual concepts, leveraging pseudo-triplets mined from image-text pairs. The pseudo-word tokens generated in this stream are explicitly aligned with fine-grained semantics in the text embedding space. Extensive experiments on widely used benchmarks, i.e. ImageNet-R, COCO object, Fashion-IQ and CIRR, show that KEDs outperforms previous zero-shot composed image retrieval methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑