Search papers, labs, and topics across Lattice.
This paper introduces UniME-R1, a novel embedder-adviser framework that enhances unified multimodal retrieval by conditioning retrieval reasoning on feedback from initially retrieved candidates. By generating Retrieval-Centric Chain-of-Thought (RC-CoT) rationales, the framework effectively identifies and addresses discriminative cues that the embedder confuses, leading to improved candidate reranking and retrieval accuracy. Extensive experiments reveal that UniME-R1 outperforms existing methods on various multimodal retrieval benchmarks, showcasing its potential to refine retrieval processes significantly.
Retrieval reasoning that learns from failures can dramatically boost the accuracy of multimodal retrieval systems.
Unified multimodal retrieval aims to identify candidates that satisfy complex user intent expressed through heterogeneous inputs. Although Large Vision-Language Model (LVLM)-based retrievers are efficient and scalable, directly encoding raw multimodal inputs often misses fine-grained discriminative cues, leading to confusion among semantically similar candidates. Recent methods mitigate this limitation by generating Chain-of-Thought (CoT) rationales to enrich the query representation. However, such reasoning is typically derived from the query alone: it explains what the query describes, but not what the retriever misunderstands. We argue that effective retrieval reasoning should instead be conditioned on retrieval feedback. Based on this insight, we introduce UniME-R1, an embedder-adviser framework that learns to reason over initially retrieved candidates and generate Retrieval-Centric Chain-of-Thought (RC-CoT). The adviser analyzes candidates individually to identify the discriminative cues confused by the embedder. If the target appears in the initial top-k set, UniME-R1 directly reranks the candidates; otherwise, it generates RC-CoT to refine the retrieval direction and performs full-corpus re-retrieval with a dual-mode embedder. To train the framework, we mine hard negatives to simulate realistic retrieval failures, jointly optimize direct retrieval and RC-CoT-augmented retrieval, and align the adviser with retrieval outcomes through supervised learning and retrieval-oriented reinforcement learning. Extensive experiments on MMEB-V2 and a diverse set of general multimodal retrieval benchmarks demonstrate that UniME-R1 consistently improves retrieval performance over strong baselines.
