Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Zero-Shot Composed Image Retrieval with Textual Inversion

About

Composed Image Retrieval (CIR) aims to retrieve a target image based on a query composed of a reference image and a relative caption that describes the difference between the two images. The high effort and cost required for labeling datasets for CIR hamper the widespread usage of existing methods, as they rely on supervised learning. In this work, we propose a new task, Zero-Shot CIR (ZS-CIR), that aims to address CIR without requiring a labeled training dataset. Our approach, named zero-Shot composEd imAge Retrieval with textuaL invErsion (SEARLE), maps the visual features of the reference image into a pseudo-word token in CLIP token embedding space and integrates it with the relative caption. To support research on ZS-CIR, we introduce an open-domain benchmarking dataset named Composed Image Retrieval on Common Objects in context (CIRCO), which is the first dataset for CIR containing multiple ground truths for each query. The experiments show that SEARLE exhibits better performance than the baselines on the two main datasets for CIR tasks, FashionIQ and CIRR, and on the proposed CIRCO. The dataset, the code and the model are publicly available at https://github.com/miccunifi/SEARLE.

Alberto Baldrati, Lorenzo Agnolucci, Marco Bertini, Alberto Del Bimbo• 2023

Related benchmarks

TaskDatasetResultRank
Composed Image RetrievalCIRR (test)
Recall@154.89
786
Composed Image RetrievalFashionIQ (val)
Average Recall@1034.82
601
Composed Image RetrievalCIRCO (test)
mAP@1016.92
360
Composed Image RetrievalFashion-IQ (test)
Average Recall@100.3481
176
Composed Image RetrievalFashion-IQ
Average Recall@5055.71
129
Composed Image Retrieval (Image-Text to Image)CIRR
Recall@134.8
128
Composed Image RetrievalCIRCO
mAP@513.2
96
Composed Image RetrievalFashionIQ Shirt
Recall@1036.46
64
Compositional Image RetrievalFashionIQ 1.0 (val)
Average Recall@1025.6
42
Composed Image RetrievalFashionIQ (Dress)
Recall@1028.16
39
Showing 10 of 64 rows

Other info

Code

Follow for update