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

Reasoning-Augmented Representations for Multimodal Retrieval

About

Universal Multimodal Retrieval (UMR) seeks any-to-any search across text and vision, yet modern embedding models remain brittle when queries require latent reasoning (e.g., resolving underspecified references or matching compositional constraints). We argue this brittleness is often data-induced: when images carry "silent" evidence and queries leave key semantics implicit, a single embedding pass must both reason and compress, encouraging spurious feature matching. We propose a data-centric framework that decouples these roles by externalizing reasoning before retrieval. Using a strong Vision--Language Model, we make implicit semantics explicit by densely captioning visual evidence in corpus entries, resolving ambiguous multimodal references in queries, and rewriting verbose instructions into concise retrieval constraints. Inference-time enhancement alone is insufficient; the retriever must be trained on these semantically dense representations to avoid distribution shift and fully exploit the added signal. Across M-BEIR, our reasoning-augmented training method yields consistent gains over strong baselines, with ablations showing that corpus enhancement chiefly benefits knowledge-intensive queries while query enhancement is critical for compositional modification requests. We publicly release our code at https://github.com/AugmentedRetrieval/ReasoningAugmentedRetrieval.

Jianrui Zhang, Anirudh Sundara Rajan, Brandon Han, Soochahn Lee, Sukanta Ganguly, Yong Jae Lee• 2026

Related benchmarks

TaskDatasetResultRank
Modification RetrievalCIRR 7
Recall@549.4
6
Multimodal RetrievalMSCOCO 0
Recall@580.2
6
Multimodal RetrievalMSCOCO 3
Recall@590.02
6
Multimodal RetrievalNights-4
R@531.56
6
Multimodal RetrievalInfoSeek 6
R@542.08
6
Multimodal RetrievalInfoSeek-8
Recall@552.38
6
Multimodal RetrievalWebQA 2
R@579.61
6
Multimodal RetrievalOVEN-6
R@548.32
6
Multimodal RetrievalOVEN-8
R@568.56
6
Multimodal RetrievalEDIS-2
Recall@553.29
6
Showing 10 of 17 rows

Other info

GitHub

Follow for update