Reasoning-Augmented Representations for Multimodal Retrieval

Jianrui Zhang, Anirudh Sundara Rajan, Brandon Han, Soochahn Lee, Sukanta Ganguly +1 more
2/6/2026
cs.IRcs.AIcs.CV

Official code: AugmentedRetrieval/ReasoningAugmentedRetrieval (0 stars, last commit 2026-04-13).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

01PythonFeb 6, 20266 months ago

Similar Papers

Cite this paper

@article{zhang2026reasoningaugmented,
  title  = {Reasoning-Augmented Representations for Multimodal Retrieval},
  author = {Jianrui Zhang and Anirudh Sundara Rajan and Brandon Han and Soochahn Lee and Sukanta Ganguly and Yong Jae Lee},
  year   = {2026},
  eprint = {2602.07125},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.07125v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.07125.svg)](https://codeofpaper.com/paper/2602.07125)

Discussion