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
887
Composed Image RetrievalFashionIQ (val)
Average Recall@1034.82
653
Composed Image RetrievalCIRCO (test)
mAP@1016.92
432
Composed Image RetrievalFashion-IQ (test)
Average Recall@100.3481
200
Composed Image Retrieval (Image-Text to Image)CIRR
Recall@134.8
168
Composed Image RetrievalFashion-IQ
Average Recall@5055.71
133
Composed Image RetrievalCIRCO
mAP@513.2
122
Composed Image RetrievalFashionIQ Shirt
Recall@1036.46
92
Composed Image RetrievalFashionIQ (Dress)
Recall@1028.16
67
Compositional Image RetrievalFashionIQ 1.0 (val)
Average Recall@1025.6
42
Showing 10 of 68 rows

Other info

Code

Follow for update