Beyond Chain-of-Thought: Rewrite as a Universal Interface for Generative Multimodal Embeddings
About
Multimodal Large Language Models (MLLMs) have emerged as a promising foundation for universal multimodal embeddings. Recent studies have shown that reasoning-driven generative multimodal embeddings can outperform discriminative embeddings on several embedding tasks. However, Chain-of-Thought (CoT) reasoning tends to generate redundant thinking steps and introduce semantic ambiguity in the summarized answers in broader retrieval scenarios. To address this limitation, we propose Rewrite-driven Multimodal Embedding (RIME), a unified framework that jointly optimizes generation and embedding through a retrieval-friendly rewrite. Meanwhile, we present the Cross-Mode Alignment (CMA) to bridge the generative and discriminative embedding spaces, enabling flexible mutual retrieval to trade off efficiency and accuracy. Based on this, we also introduce Refine Reinforcement Learning (Refine-RL) that treats discriminative embeddings as stable semantic anchors to guide the rewrite optimization. Extensive experiments on MMEB-V2, MRMR and UVRB demonstrate that RIME substantially outperforms prior generative embedding models while significantly reducing the length of thinking.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multimodal Visual Document Retrieval | MMEB Visual Document portion v2 | -- | 31 | |
| Video Retrieval | UVRB (test) | AVG Score55.6 | 28 | |
| Multimodal Retrieval | MMEB Image v2 (test) | CLS (Hit@1)70.3 | 18 | |
| Multimodal Video Retrieval | MMEB Video V2 (test) | CLS Hit@152.6 | 18 | |
| Universal Multimodal Retrieval | MMEB Full v2 (test) | Overall Average Score68.6 | 18 | |
| Multimodal Retrieval | MRMR | Score (Art Domain)76.8 | 11 | |
| Multimodal Reasoning-Intensive Retrieval | MR2-Bench | nDCG@10 (Bio.)38.5 | 8 |