arXiv:2411.02571cs.CLcs.AI2024-11ICLR被引 169

用多模态大模型实现跨模态通用检索,支持图文混合查询

MM-Embed: Universal Multimodal Retrieval with Multimodal LLMs

论文配图:MM-Embed: Universal Multimodal Retrieval with Multimodal LLMs
图 1 · 摘自论文原文
  • 用多模态大模型做双编码器检索,支持多任务多模态输入
  • 提出感知模态的难负样本挖掘,解决模型偏见问题
  • 可零样本提示重排序,适合复杂图文混合查询场景

现有先进检索模型通常只处理单一模态的固定任务(如文本问答)。本文提出通用多模态检索框架,支持多种模态和多样化任务。通过在10个数据集、16种任务上微调多模态大语言模型(MLLM)作为双编码器检索器,发现其虽能理解图文混合查询,但在跨模态检索中性能低于小型CLIP检索器,因存在模态偏见。为此提出模态感知难负样本挖掘以缓解该问题。进一步持续微调模型,提升文本检索能力并保持多模态性能。最终模型MM-Embed在涵盖多领域多任务的M-BEIR基准上达到最优,在MTEB文本检索基准上超越NV-Embed-v1。此外,探索将现成MLLM作为零样本重排序器,通过提示+重排序可进一步提升复杂图文查询的检索效果。

原文摘要 · Abstract (English)

State-of-the-art retrieval models typically address a straightforward search scenario, in which retrieval tasks are fixed (e.g., finding a passage to answer a specific question) and only a single modality is supported for both queries and retrieved results. This paper introduces techniques for advancing information retrieval with multimodal large language models (MLLMs), enabling a broader search scenario, termed universal multimodal retrieval, where multiple modalities and diverse retrieval tasks are accommodated. To this end, we first study fine-tuning an MLLM as a bi-encoder retriever on 10 datasets with 16 retrieval tasks. Our empirical results show that the fine-tuned MLLM retriever is capable of understanding challenging queries, composed of both text and image, but it underperforms compared to a smaller CLIP retriever in cross-modal retrieval tasks due to the modality bias exhibited by MLLMs. To address the issue, we propose modality-aware hard negative mining to mitigate the modality bias exhibited by MLLM retrievers. Second, we propose continuously fine-tuning the universal multimodal retriever to enhance its text retrieval capability while preserving multimodal retrieval capability. As a result, our model, MM-Embed, achieves state-of-the-art performance on the multimodal retrieval benchmark M-BEIR, which spans multiple domains and tasks, while also surpassing the state-of-the-art text retrieval model, NV-Embed-v1, on the MTEB retrieval benchmark. We also explore prompting the off-the-shelf MLLMs as zero-shot rerankers to refine the ranking of the candidates from the multimodal retriever. We find that, through prompt-and-reranking, MLLMs can further improve multimodal retrieval when the user queries (e.g., text-image composed queries) are more complex and challenging to understand. These findings also pave the way for advancing universal multimodal retrieval in the future.

多模态检索大模型图文混合检索优化

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