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

UniIR: Training and Benchmarking Universal Multimodal Information Retrievers

About

Existing information retrieval (IR) models often assume a homogeneous format, limiting their applicability to diverse user needs, such as searching for images with text descriptions, searching for a news article with a headline image, or finding a similar photo with a query image. To approach such different information-seeking demands, we introduce UniIR, a unified instruction-guided multimodal retriever capable of handling eight distinct retrieval tasks across modalities. UniIR, a single retrieval system jointly trained on ten diverse multimodal-IR datasets, interprets user instructions to execute various retrieval tasks, demonstrating robust performance across existing datasets and zero-shot generalization to new tasks. Our experiments highlight that multi-task training and instruction tuning are keys to UniIR's generalization ability. Additionally, we construct the M-BEIR, a multimodal retrieval benchmark with comprehensive results, to standardize the evaluation of universal multimodal information retrieval.

Cong Wei, Yang Chen, Haonan Chen, Hexiang Hu, Ge Zhang, Jie Fu, Alan Ritter, Wenhu Chen• 2023

Related benchmarks

TaskDatasetResultRank
Text-to-Image RetrievalMS-COCO--
187
Image-to-Text RetrievalMS-COCO--
168
Image-to-Text RetrievalMSCOCO--
152
Text-to-Image RetrievalMSCOCO--
142
Composed Image Retrieval (Image-Text to Image)CIRR--
128
Composed Image RetrievalCIRCO
mAP@512.5
96
Multimodal RetrievalMMEB
Classification Score44.3
94
Image EmbeddingMMEB v1 (test)
Classification44.3
70
Multimodal EmbeddingMMEB
Classification Accuracy44.3
66
Text-to-Image RetrievalShareGPT4V
R@185.8
52
Showing 10 of 65 rows

Other info

Follow for update