arXiv:2410.05160cs.CVcs.AI2024-10ICLR被引 241

让视觉语言模型变身为通用多模态嵌入工具,性能提升10%-20%。

VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Tasks

论文配图:VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Tasks
图 1 · 摘自论文原文
  • 通过对比学习将主流VLM转化为可处理任意图文组合的嵌入模型
  • 在4类任务36个数据集上实现比现有模型高10%-20%的平均性能
  • 适合需要统一多模态表征的科研与工业场景

嵌入模型在语义相似度、信息检索和聚类等下游任务中至关重要。尽管通用文本嵌入模型(如MTEB)发展迅速,但通用多模态嵌入模型进展缓慢。本文提出两个贡献:(1) MMEB(大规模多模态嵌入基准),涵盖4个元任务(分类、视觉问答、多模态检索、视觉定位)和36个数据集(20个训练集、16个评估集),包含分布内与分布外任务;(2) VLM2Vec(视觉语言模型→向量),一种对比训练框架,可将任意先进VLM转为嵌入模型。与CLIP、BLIP等独立编码图文不同,VLM2Vec根据任务指令处理任意图文组合,生成固定维度向量。我们在Phi-3.5-V、LLaVA-1.6等先进VLM上构建系列VLM2Vec模型,并在MMEB评估集上测试。结果表明,其在分布内与分布外数据集上均比现有模型绝对提升10%至20%。

原文摘要 · Abstract (English)

Embedding models have been crucial in enabling various downstream tasks such as semantic similarity, information retrieval, and clustering. Recently, there has been a surge of interest in developing universal text embedding models that can generalize across tasks (e.g., MTEB). However, progress in learning universal multimodal embedding models has been relatively slow despite its importance and practicality. In this work, we aim to explore the potential for building universal embeddings capable of handling a wide range of downstream tasks. Our contributions are twofold: (1) MMEB (Massive Multimodal Embedding Benchmark), which covers 4 meta-tasks (i.e. classification, visual question answering, multimodal retrieval, and visual grounding) and 36 datasets, including 20 training and 16 evaluation datasets covering both in-distribution and out-of-distribution tasks, and (2) VLM2Vec (Vision-Language Model -> Vector), a contrastive training framework that converts any state-of-the-art vision-language model into an embedding model via training on MMEB. Unlike previous models such as CLIP and BLIP, which encodes text or images independently without any task instruction, VLM2Vec can process any combination of images and text to generate a fixed-dimensional vector based on task instructions. We build a series of VLM2Vec models on SoTA VLMs like Phi-3.5-V, LLaVA-1.6 and evaluate them on MMEB's evaluation split. Our results show that VLM2Vec achieves an absolute average improvement of 10% to 20% over existing multimodal embedding models on both in-distribution and out-of-distribution datasets in MMEB. We show that VLMs are secretly strong embedding models.

多模态嵌入视觉语言模型对比学习基准测试

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