用双向注意力提升多模态嵌入效果,更鲁棒且可扩展。
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 官方产品;中文卡片由大模型生成,请以原文为准。