Reconstructing Content with Collaborative Attention for Universal Multimodal Representation Learning
Authors: Jiahan Chen, Da Li, Hengran Zhang, Yinqiong Cai, Lixin Su, Jiafeng Guo, Daiting Shi, Dawei Yin, Keping Bi
Organizations: State Key Laboratory of AI Safety, Institute of Computing Technology, Chinese Academy of Sciences, University of Chinese Academy of Sciences, Beijing, China · Baidu Inc., Beijing, China
Multimodal embedding models, rooted in multimodal large language models (MLLMs), have yielded significant performance improvements across diverse tasks such as retrieval and classification. However, most existing approaches rely heavily on large-scale contrastive learning, with limited exploration of how the architectural and training paradigms of MLLMs affect embedding quality. While effective for generation, the causal attention and next-token prediction paradigm of MLLMs does not explicitly encourage the formation of globally compact representations, limiting their effectiveness as multimodal embedding backbones. To address this, we propose CoCoA, a Content reconstruction pre-training paradigm based on Collaborative Attention for multimodal embedding optimization. Specifically, we restructure the attention flow and introduce an EOS-based reconstruction task, encouraging the model to reconstruct input from the corresponding <EOS> embeddings. This drives the multimodal model to compress the semantic information of the input into the <EOS> token, laying the foundations for subsequent contrastive learning. Extensive experiments on MMEB-V1 demonstrate that CoCoA built upon Qwen2-VL and Qwen2.5-VL significantly improves embedding quality. Results validate that content reconstruction serves as an effective strategy to maximize the value of existing data, enabling multimodal embedding models generate compact and informative representations, raising their performance ceiling.
Figures & tables
Figure 1. Overview of CoCoA. Stage 1: Bidirectional attention warm-up using MAE and MNTP for different modalities; Stage 2: EOS-bridged caption or answer tokens reconstruction forces the multi-modalities’ semantics to be compressed into a single token, i.e., ⟨EOS⟩ ; Stage 3: Contrastive learning using ⟨EOS⟩ .
Meta-Task
Block A (Compression)
Block B (Reconstruction)
Classification
qry img
tgt txt
VQA
qry img + qry txt
tgt txt
Retrieval
qry img
tgt txt
Table 1. Construction of training instances in the form of BlockA⟨EOS⟩BlockB for different tasks during reconstruction.
Figure 2. An example of synthetic data. The fine-grained details are highlighted in bold to emphasize that these elements are newly introduced by the synthetic data generation process and are not explicitly present in the original data.
Model
Size
Per Meta-Task Score
Average Score
Classification
VQA
Retrieval
Grounding
IND
OOD
Overall
Only contrastive learning
CLIP (ViT-L)
428M
55.2
19.7
53.2
62.2
47.6
42.8
45.4
OpenCLIP (ViT-L)
428M
56.0
21.9
55.4
64.1
50.5
43.1
47.2
GME (Qwen2-VL)
2B
56.9
41.2
67.8
53.4
–
–
55.8
UNITE (Qwen2-VL)
2B
63.2
55.9
65.4
75.6
65.8
60.1
63.3
Table 2. Results on MMEB-V1. “IND” denotes in-distribution, and “OOD” refers to out-of-distribution. Bold and underline indicate the optimal and suboptimal performance, respectively.
Model
MMEB-V1
Our Method
CoCoA (Qwen2-VL 2B)
62.9
Pretraining Stage Ablation
w/o. Bidirectional Attention Warm-Up
61.6
w/o. EOS-Bridged Reconstruction
60.7
Baseline
Table 3. Ablation study on different pretraining stages.
Image Loss
Text Loss
MMEB-V1
MAE
MNTP
62.9
MAE
–
61.8
–
MNTP
62.3
–
MLM
57.9
Table 4. Impact of different loss functions in Bidirectional Attention Warm-Up.
Figure 3. Performance under different mask ratios. We adopt three different mask ratios: 20%, 50%, and 70% from left to right.
Task
Avg. Tokens
<4
4–10
>10
Classification
2.41
89.3%
9.1%
1.6%
VQA
3.12
67.8%
28.4%
3.8%
Retrieval
18.09
0.3%
20.1%
79.6%
Table 5. Token length statistics of reconstructed text across different tasks.
Figure 4. Performance under different pre-training data volumes. Baseline is VLM2Vec based on Qwen2-VL-2B. 100K-400K denote the amounts of MMEB-V1 original data, and “+” indicates additional synthetic data.
Figure 5. The effect of pretraining data scale on In-Domain (IND) and Out-of-Domain (OOD) performance.
Figure 6. Case study of visual information reconstruction from ⟨EOS⟩ in different multimodal embedding models.
Contrastive learning has become the standard paradigm for multimodal representation learning, aligning semantically matched pairs while separating mismatched ones in a shared embedding space. Although various contrastive training strategies have been proposed across negative selection, batch construction, and gradient modulation, their actual effectiveness under controlled, reproducible conditions remains poorly understood. This motivates a fundamental question: Do sophisticated contrastive strategies consistently bring meaningful downstream gains in multimodal embedding learning? To address this, we systematically evaluate representative strategies under a unified benchmarking framework on MMEB. Surprisingly, direct evaluations reveal that most existing strategies offer marginal gains over a standard aligned baseline, with several even degrading overall performance. To uncover the underlying failure modes, we introduce an MLLM-as-a-Judge diagnostic framework that dissects top-1 retrieval behaviors into labeled positives, false negatives (unlabeled valid matches), and hard/easy negative errors. Our systematic analysis yields three key insights: (1) Gradient distortion: Upweighting gradients from high-similarity negatives consistently hurts downstream generalization; (2) Distribution illusion: Wider separation between positive and negative similarity distributions does not necessarily translate to better retrieval performance; and (3) Synergistic training: While false-negative mitigation or hard-batch construction yields limited gains in isolation, integrating hard-negative mining, false-negative mitigation, and hard-batch construction achieves optimal performance, delivering an average improvement of 6.0% on MMEB with Qwen2-VL-2B. Overall, our findings demystify popular contrastive strategies and provide actionable recipes for training universal multimodal embedding models.
Duo Li, Haojun Pan, Zuhao Yang +4
CCDS, Nanyang Technological University · Alibaba Group · Universität Hamburg +1
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. Code is available at https://github.com/PeppaWu/RIME.
Peixi Wu, Ke Mei, Feipeng Ma +15
WeChat Vision, Tencent Inc. · Zhejiang University · Tsinghua University +1
Multimodal Large Language Models (MLLMs) have shown remarkable success in comprehension tasks such as visual description and visual question answering. However, their direct application to embedding-based tasks like retrieval remains challenging due to the discrepancy between output formats and optimization objectives. Previous approaches often employ contrastive fine-tuning to adapt MLLMs for retrieval, but at the cost of losing their generative capabilities. We argue that both generative and embedding tasks fundamentally rely on shared cognitive mechanisms, specifically cross-modal representation alignment and contextual comprehension. To this end, we propose CREM (Compression-driven Representation Enhanced Model), with a unified framework that enhances multimodal representations for retrieval while preserving generative ability. Specifically, we introduce a compression-based prompt design with learnable chorus tokens to aggregate multimodal semantics and a compression-driven training strategy that integrates contrastive and generative objectives through compression-aware attention. Extensive experiments demonstrate that CREM achieves state-of-the-art retrieval performance on MMEB while maintaining strong generative performance on multiple comprehension benchmarks. Our findings highlight that generative supervision can further improve the representational quality of MLLMs under the proposed compression-driven paradigm.