Vision-language foundation models, represented by Contrastive language-image pre-training (CLIP), have gained increasing attention for jointly understanding both vision and textual tasks. However, existing approaches primarily focus on training models to match global image representations with textual descriptions, thereby overlooking the critical alignment between local regions and corresponding text tokens. This paper extends CLIP with multi-granularity alignment. Notably, we deliberately construct a new dataset comprising pseudo annotations at various levels of granularities, encompassing image-level, region-level, and pixel-level captions/tags. Accordingly, we develop a unified multi-granularity learning framework, named UMG-CLIP, that simultaneously empowers the model with versatile perception abilities across different levels of detail. Equipped with parameter efficient tuning, UMG-CLIP surpasses current widely used CLIP models and achieves state-of-the-art performance on diverse image understanding benchmarks, including open-world recognition, retrieval, semantic segmentation, and panoptic segmentation tasks. We hope UMG-CLIP can serve as a valuable option for advancing vision-language foundation models.
翻译:视觉语言基础模型,以对比语言-图像预训练(CLIP)为代表,因能联合理解视觉与文本任务而备受关注。然而,现有方法主要聚焦于训练模型匹配全局图像表征与文本描述,从而忽略了局部区域与对应文本标记之间的关键对齐。本文将CLIP扩展为多粒度对齐。值得注意的是,我们精心构建了一个包含多层级粒度伪标注的新数据集,涵盖图像级、区域级和像素级字幕/标签。据此,我们开发了名为UMG-CLIP的统一多粒度学习框架,该框架同时赋予模型在不同细节层次上的通用感知能力。通过参数高效微调,UMG-CLIP超越了当前广泛使用的CLIP模型,在开放世界识别、检索、语义分割及全景分割等多种图像理解基准任务中取得了最先进性能。我们希望UMG-CLIP能成为推动视觉语言基础模型发展的宝贵选择。