arXiv:2510.13515cs.CVcs.AI2025-10AAAI被引 39

用大模型当裁判,让多模态嵌入更精准区分相似内容

UniME-V2: MLLM-as-a-Judge for Universal Multimodal Embedding Learning

论文配图:UniME-V2: MLLM-as-a-Judge for Universal Multimodal Embedding Learning
图 1 · 摘自论文原文
  • 让大模型评估图文配对语义匹配度,生成软标签指导负样本挖掘
  • 在多个任务上达到当前最优,尤其提升对难负样本的区分能力
  • 适合需要高精度图文检索的研究者和工程落地场景

通用多模态嵌入模型是众多任务的基础。现有方法通常通过计算查询-候选对之间的相似性进行批次内负样本挖掘,但难以捕捉候选间的细微语义差异,且负样本多样性不足,嵌入表示对假负样本和难负样本的区分能力有限。本文利用多模态大模型(MLLM)的先进理解能力,提出新型通用多模态嵌入模型UniME-V2。首先通过全局检索构建潜在难负样本集,再引入MLLM-as-a-Judge机制,由大模型评估查询-候选对的语义对齐程度,生成软语义匹配分数。这些分数用于指导难负样本挖掘,缓解假负样本影响,并识别多样、高质量的难负样本。同时,将语义匹配分数作为软标签,打破传统一一对应约束,使模型学习候选间更精细的语义差异,显著提升判别能力。为进一步优化性能,提出UniME-V2-Reranker,基于挖掘出的难负样本,采用联合成对与列表式优化训练。在MMEB基准及多个检索任务上的实验表明,该方法在所有任务上平均达到当前最优性能。

原文摘要 · Abstract (English)

Universal multimodal embedding models are foundational to various tasks. Existing approaches typically employ in-batch negative mining by measuring the similarity of query-candidate pairs. However, these methods often struggle to capture subtle semantic differences among candidates and lack diversity in negative samples. Moreover, the embeddings exhibit limited discriminative ability in distinguishing false and hard negatives. In this paper, we leverage the advanced understanding capabilities of MLLMs to enhance representation learning and present a novel Universal Multimodal Embedding (UniME-V2) model. Our approach first constructs a potential hard negative set through global retrieval. We then introduce the MLLM-as-a-Judge mechanism, which utilizes MLLMs to assess the semantic alignment of query-candidate pairs and generate soft semantic matching scores. These scores serve as a foundation for hard negative mining, mitigating the impact of false negatives and enabling the identification of diverse, high-quality hard negatives. Furthermore, the semantic matching scores are used as soft labels to mitigate the rigid one-to-one mapping constraint. By aligning the similarity matrix with the soft semantic matching score matrix, the model learns semantic distinctions among candidates, significantly enhancing its discriminative capacity. To further improve performance, we propose UniME-V2-Reranker, a reranking model trained on our mined hard negatives through a joint pairwise and listwise optimization approach. We conduct comprehensive experiments on the MMEB benchmark and multiple retrieval tasks, demonstrating that our method achieves state-of-the-art performance on average across all tasks.

多模态嵌入大模型裁判难负样本挖掘

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