arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.01998cs.CVcs.IR

零样本组合图像检索的仅语言高效训练

Language-only Efficient Training of Zero-shot Composed Image Retrieval

NAVER Vision(NAVER视觉部门);NAVER AI Lab(NAVER人工智能实验室)

Geonmo Gu, Sanghyuk Chun, Wonjae Kim, Yoohoon Kang, Sangdoo Yun

AI总结:

本文提出仅用语言训练的LinCIR框架,通过自掩码投影实现零样本组合图像检索,以CLIP ViT-G为主干在48分钟内完成训练,并在多个基准上取得最佳性能。

摘要:

组合图像检索(CIR)任务接受由图像和文本组成的组合查询,旨在搜索同时满足两个条件的相关图像。传统的CIR方法需要一个由查询图像、查询文本和目标图像组成的三元组训练数据集,而收集这样的数据集成本非常高。近期一些工作致力于零样本(ZS)CIR范式,以在不使用预先收集的三元组的情况下解决该问题。然而,现有的ZS-CIR方法由于训练期间输入文本缺乏多样性,表现出有限的主干网络可扩展性和泛化能力。我们提出了一种新颖的CIR框架,仅使用语言进行训练。我们的LinCIR(Language-only training for CIR,仅语言训练的CIR)可以通过一种名为自掩码投影(SMP)的新型自监督方法仅使用文本数据集进行训练。我们将文本潜在嵌入投影到token嵌入空间,并通过替换原始文本的关键词token来构造新文本。然后,我们让新文本和原始文本具有相同的潜在嵌入向量。凭借这一简单策略,LinCIR出人意料地高效且效果显著;使用CLIP ViT-G主干网络的LinCIR在48分钟内完成训练,并在四个不同的CIR基准——CIRCO、GeneCIS、FashionIQ和CIRR上展现出最佳的ZS-CIR性能,甚至在FashionIQ上超越了有监督方法。代码可在 https://github.com/navervision/lincir 获取。

英文摘要:

Composed image retrieval (CIR) task takes a composed query of image and text, aiming to search relative images for both conditions. Conventional CIR approaches need a training dataset composed of triplets of query image, query text, and target image, which is very expensive to collect. Several recent works have worked on the zero-shot (ZS) CIR paradigm to tackle the issue without using pre-collected triplets. However, the existing ZS-CIR methods show limited backbone scalability and generalizability due to the lack of diversity of the input texts during training. We propose a novel CIR framework, only using language for its training. Our LinCIR (Language-only training for CIR) can be trained only with text datasets by a novel self-supervision named self-masking projection (SMP). We project the text latent embedding to the token embedding space and construct a new text by replacing the keyword tokens of the original text. Then, we let the new and original texts have the same latent embedding vector. With this simple strategy, LinCIR is surprisingly efficient and highly effective; LinCIR with CLIP ViT-G backbone is trained in 48 minutes and shows the best ZS-CIR performances on four different CIR benchmarks, CIRCO, GeneCIS, FashionIQ, and CIRR, even outperforming supervised method on FashionIQ. Code is available at https://github.com/navervision/lincir

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

↑