arXiv:2410.02746cs.CVcs.LG2024-10ICML被引 47

提升CLIP的局部视觉理解能力,让模型更精准定位图像区域。

Contrastive Localized Language-Image Pre-Training

  • 引入区域-文本对比损失,增强图像局部与文本的对齐能力。
  • 在10亿级图像上训练,实现高精度的区域识别与检索性能。
  • 适合需要精确定位的多模态大模型任务,如指代表达与视觉定位。

对比语言-图像预训练(CLIP)通过对网络抓取的噪声文本标注进行图像级对齐,成功构建了视觉-语言基础模型。然而,这种图像级对齐难以满足下游任务中细粒度视觉表征的需求,尤其在多模态大语言模型(MLLM)需要区域级理解时。本文提出对比局部语言-图像预训练(CLOC),通过引入区域-文本对比损失和新模块,增强模型的局部定位能力。我们定义了可提示嵌入(promptable embeddings),使编码器能根据空间提示轻松生成区域表示。为支持大规模预训练,设计了视觉丰富且空间定位的伪标签生成框架。在超过10亿张标注图像上训练后,CLOC实现了高质量的区域嵌入,在区域识别与检索任务中表现优异,可直接替换CLIP以提升MLLM在指代和定位任务中的性能。

原文摘要 · Abstract (English)

Contrastive Language-Image Pre-training (CLIP) has been a celebrated method for training vision encoders to generate image/text representations facilitating various applications. Recently, CLIP has been widely adopted as the vision backbone of multimodal large language models (MLLMs) to connect image inputs for language interactions. The success of CLIP as a vision-language foundation model relies on aligning web-crawled noisy text annotations at image levels. Nevertheless, such criteria may become insufficient for downstream tasks in need of fine-grained vision representations, especially when region-level understanding is demanding for MLLMs. In this paper, we improve the localization capability of CLIP with several advances. We propose a pre-training method called Contrastive Localized Language-Image Pre-training (CLOC) by complementing CLIP with region-text contrastive loss and modules. We formulate a new concept, promptable embeddings, of which the encoder produces image embeddings easy to transform into region representations given spatial hints. To support large-scale pre-training, we design a visually-enriched and spatially-localized captioning framework to effectively generate region-text pseudo-labels at scale. By scaling up to billions of annotated images, CLOC enables high-quality regional embeddings for image region recognition and retrieval tasks, and can be a drop-in replacement of CLIP to enhance MLLMs, especially on referring and grounding tasks.

视觉定位多模态预训练区域理解

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