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

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.

Peixi Wu, Ke Mei, Feipeng Ma, Bosong Chai, Zhibin Lan, Chenxi Zhao, Shannan Yan, Jie Chen, Zhangchi Hu, Yansong Peng, Bo Lin, Junjie Zhou, Dacheng Yin, Tianyi Wang, Fengyun Rao, Jing Lyu, Hebei Li, Xiaoyan Sun• 2026

Related benchmarks

TaskDatasetResultRank
Multimodal Visual Document RetrievalMMEB Visual Document portion v2--
31
Video RetrievalUVRB (test)
AVG Score55.6
28
Multimodal RetrievalMMEB Image v2 (test)
CLS (Hit@1)70.3
18
Multimodal Video RetrievalMMEB Video V2 (test)
CLS Hit@152.6
18
Universal Multimodal RetrievalMMEB Full v2 (test)
Overall Average Score68.6
18
Multimodal RetrievalMRMR
Score (Art Domain)76.8
11
Multimodal Reasoning-Intensive RetrievalMR2-Bench
nDCG@10 (Bio.)38.5
8
Showing 7 of 7 rows

Other info

Follow for update