用自监督方法实现零样本图像组合检索,无需标注数据即可高效匹配图文查询。
SCOT: Self-Supervised Contrastive Pretraining For Zero-Shot Compositional Retrieval
- 利用大语言模型生成文本描述,替代真实目标图像作为对比学习的监督信号。
- 在FashionIQ和CIRR数据集上超越多数全监督方法,在零样本场景下表现最优。
- 适合希望跳过繁琐标注、快速部署跨领域图像检索系统的开发者使用。
组合图像检索(CIR)是一种多模态学习任务,要求模型将查询图像与用户提供的文本修改结合,以检索目标图像。该任务在电商商品检索和网络搜索等领域有广泛应用。现有方法主要依赖全监督学习,需人工标注包含图像-文本三元组的数据集(如FashionIQ和CIRR),存在两个核心挑战:(i) 标注成本高;(ii) 模型难以泛化到未见物体和新领域。本文提出SCOT(自监督组合预训练)策略,通过结合大规模图像-文本对数据集与大语言模型的生成能力,对嵌入组合网络进行对比学习预训练。关键创新在于:利用大规模对比预训练视觉-语言模型输出的文本嵌入作为代理目标监督信号,替代真实目标图像嵌入。在零样本设置下,该方法在FashionIQ和CIRR等标准基准上超越现有最先进零样本方法,并媲美甚至优于部分全监督方法。
原文摘要 · Abstract (English)
Compositional image retrieval (CIR) is a multimodal learning task where a model combines a query image with a user-provided text modification to retrieve a target image. CIR finds applications in a variety of domains including product retrieval (e-commerce) and web search. Existing methods primarily focus on fully-supervised learning, wherein models are trained on datasets of labeled triplets such as FashionIQ and CIRR. This poses two significant challenges: (i) curating such triplet datasets is labor intensive; and (ii) models lack generalization to unseen objects and domains. In this work, we propose SCOT (Self-supervised COmpositional Training), a novel zero-shot compositional pretraining strategy that combines existing large image-text pair datasets with the generative capabilities of large language models to contrastively train an embedding composition network. Specifically, we show that the text embedding from a large-scale contrastively-pretrained vision-language model can be utilized as proxy target supervision during compositional pretraining, replacing the target image embedding. In zero-shot settings, this strategy surpasses SOTA zero-shot compositional retrieval methods as well as many fully-supervised methods on standard benchmarks such as FashionIQ and CIRR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。