Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Zero-shot Composed Text-Image Retrieval

About

In this paper, we consider the problem of composed image retrieval (CIR), it aims to train a model that can fuse multi-modal information, e.g., text and images, to accurately retrieve images that match the query, extending the user's expression ability. We make the following contributions: (i) we initiate a scalable pipeline to automatically construct datasets for training CIR model, by simply exploiting a large-scale dataset of image-text pairs, e.g., a subset of LAION-5B; (ii) we introduce a transformer-based adaptive aggregation model, TransAgg, which employs a simple yet efficient fusion mechanism, to adaptively combine information from diverse modalities; (iii) we conduct extensive ablation studies to investigate the usefulness of our proposed data construction procedure, and the effectiveness of core components in TransAgg; (iv) when evaluating on the publicly available benckmarks under the zero-shot scenario, i.e., training on the automatically constructed datasets, then directly conduct inference on target downstream datasets, e.g., CIRR and FashionIQ, our proposed approach either performs on par with or significantly outperforms the existing state-of-the-art (SOTA) models. Project page: https://code-kunkun.github.io/ZS-CIR/

Yikun Liu, Jiangchao Yao, Ya Zhang, Yanfeng Wang, Weidi Xie• 2023

Related benchmarks

TaskDatasetResultRank
Composed Image RetrievalCIRR (test)
Recall@138.1
481
Composed Image RetrievalFashionIQ (val)
Shirt Recall@1034.45
455
Composed Image Retrieval (Image-Text to Image)CIRR--
75
Composed Image RetrievalCIRCO--
63
Compositional Image RetrievalFashionIQ 1.0 (val)
Average Recall@1028.57
42
Composed Image RetrievalFashion-IQ
Average Recall@1034.4
40
Sketch-based image retrievalSketchy
mAP@20073.4
15
Sketch-based image retrievalTU-Berlin
mAP63.1
15
Sketch-based image retrievalQuickDraw
mAP24.9
15
Composed Image RetrievalPatternCom
mAP@20037.4
4
Showing 10 of 11 rows

Other info

Code

Follow for update