用大模型提升多模态嵌入,让图文匹配更准更懂组合
Breaking the Modality Barrier: Universal Embedding Learning with Multimodal LLMs
- 分两阶段训练:先用大语言模型教多模态模型理解文本,再用难例增强指令调优
- 在多个检索任务上性能全面超越现有方法,尤其在复杂组合查询上提升显著
- 适合做多模态搜索、图像描述生成等需要精准理解图文关系的任务
对比语言-图像预训练(CLIP)框架虽广泛用于多模态表示学习,但受限于文本截断、图文独立编码和词袋行为导致的组合能力不足。尽管多模态大语言模型(MLLM)在视觉-语言理解上取得进展,其可迁移多模态表示学习潜力仍待挖掘。本文提出UniME(通用多模态嵌入)框架,通过两阶段训练:第一阶段,利用强大语言模型作为教师,对MLLM的语言组件进行文本判别性知识蒸馏;第二阶段,引入硬负样本增强的指令调优,先缓解假负样本干扰,再在每批样本中采样多个难负例,迫使模型聚焦于挑战性样本。该方法不仅增强判别能力,也提升下游任务中的指令遵循能力。在MMEB基准及多种检索任务(包括短/长标题检索与组合检索)上的实验表明,UniME在所有任务上均实现一致性能提升,展现出更强的判别性与组合理解能力。
原文摘要 · Abstract (English)
The Contrastive Language-Image Pre-training (CLIP) framework has become a widely used approach for multimodal representation learning, particularly in image-text retrieval and clustering. However, its efficacy is constrained by three key limitations: (1) text token truncation, (2) isolated image-text encoding, and (3) deficient compositionality due to bag-of-words behavior. While recent Multimodal Large Language Models (MLLMs) have demonstrated significant advances in generalized vision-language understanding, their potential for learning transferable multimodal representations remains underexplored.In this work, we present UniME (Universal Multimodal Embedding), a novel two-stage framework that leverages MLLMs to learn discriminative representations for diverse downstream tasks. In the first stage, we perform textual discriminative knowledge distillation from a powerful LLM-based teacher model to enhance the embedding capability of the MLLMś language component. In the second stage, we introduce hard negative enhanced instruction tuning to further advance discriminative representation learning. Specifically, we initially mitigate false negative contamination and then sample multiple hard negatives per instance within each batch, forcing the model to focus on challenging samples. This approach not only improves discriminative power but also enhances instruction-following ability in downstream tasks. We conduct extensive experiments on the MMEB benchmark and multiple retrieval tasks, including short and long caption retrieval and compositional retrieval. Results demonstrate that UniME achieves consistent performance improvement across all tasks, exhibiting superior discriminative and compositional capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。