arXiv:2506.23115cs.CVcs.AI2025-06ACL被引 21

用双向注意力提升多模态嵌入效果,更鲁棒且可扩展。

MoCa: Modality-aware Continual Pre-training Makes Better Bidirectional Multimodal Embeddings

  • 分两阶段训练:先用联合重建增强双向理解,再用多样数据微调对齐
  • 在MMEB和ViDoRe-v2上达新SOTA,模型越大数据越多表现越好
  • 不依赖高质量成对数据,适合大规模无标注多模态训练

多模态嵌入模型基于因果视觉语言模型(VLM)在各类任务中展现潜力。然而现有方法存在三大局限:VLM主干中的因果注意力对嵌入任务不理想;依赖高质量成对标注数据进行对比学习导致可扩展性差;训练目标与数据多样性不足。为此,我们提出MoCa,一种将预训练VLM转化为高效双向多模态嵌入模型的两阶段框架。第一阶段为模态感知持续预训练,引入联合重建目标,同时对交错的文本与图像输入进行去噪,增强双向上下文感知推理能力。第二阶段为异质对比微调,利用超越简单图文对的丰富语义多模态数据,提升泛化性与对齐效果。该方法通过持续预训练引入双向注意力,借助联合重建目标实现大规模无标注数据的有效扩展,并利用多样化多模态数据增强表征鲁棒性。实验表明,MoCa在MMEB和ViDoRe-v2基准上持续提升性能,达到新状态最优水平,并在模型规模与训练数据量上均表现出强可扩展性。

原文摘要 · Abstract (English)

Multimodal embedding models, built upon causal Vision Language Models (VLMs), have shown promise in various tasks. However, current approaches face three key limitations: the use of causal attention in VLM backbones is suboptimal for embedding tasks; scalability issues due to reliance on high-quality labeled paired data for contrastive learning; and limited diversity in training objectives and data. To address these issues, we propose MoCa, a two-stage framework for transforming pre-trained VLMs into effective bidirectional multimodal embedding models. The first stage, Modality-aware Continual Pre-training, introduces a joint reconstruction objective that simultaneously denoises interleaved text and image inputs, enhancing bidirectional context-aware reasoning. The second stage, Heterogeneous Contrastive Fine-tuning, leverages diverse, semantically rich multimodal data beyond simple image-caption pairs to enhance generalization and alignment. Our method addresses the stated limitations by introducing bidirectional attention through continual pre-training, scaling effectively with massive unlabeled datasets via joint reconstruction objectives, and utilizing diverse multimodal data for enhanced representation robustness. Experiments demonstrate that MoCa consistently improves performance across MMEB and ViDoRe-v2 benchmarks, achieving new state-of-the-art results, and exhibits strong scalability with both model size and training data on MMEB.

多模态嵌入持续预训练双向建模无监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。