用反向生成模型提升CLIP对图像细节的捕捉能力。
un$^2$CLIP: Improving CLIP's Visual Detail Capturing Ability via Inverting unCLIP
- 用unCLIP反向构建图像生成器,增强视觉细节建模。
- 在多个任务上超越原版CLIP和已有改进方法。
- 适合需要精细图像理解的多模态应用研究者。
对比语言-图像预训练(CLIP)已成为基础模型,广泛应用于视觉与多模态任务。然而,近期研究表明,CLIP在区分图像细微差异方面表现不足,且在密集预测与以视觉为中心的多模态任务中效果不佳。为此,本文聚焦于提升现有CLIP模型的视觉细节捕捉能力。我们发现,一种特定的生成模型——unCLIP,提供了实现目标的合适框架。unCLIP通过条件化CLIP图像嵌入来训练图像生成器,本质上是对CLIP图像编码器进行逆向建模。相较于判别式模型如CLIP,生成模型更擅长捕捉图像细节,因其需学习图像数据分布。此外,unCLIP的条件输入空间与原始CLIP的图文嵌入空间一致。因此,我们提出将unCLIP反向(称为un$^2$CLIP),以改进CLIP模型。该方法使图像编码器在保持与原文本编码器对齐的同时,获得unCLIP的视觉细节捕捉能力。我们在多种任务上评估改进后的CLIP,包括挑战性的MMVP-VLM基准、开放词汇语义分割等密集预测任务,以及多模态大语言模型任务。实验表明,un$^2$CLIP显著优于原版CLIP及此前改进方法。代码与模型将公开于https://github.com/LiYinqi/un2CLIP。
原文摘要 · Abstract (English)
Contrastive Language-Image Pre-training (CLIP) has become a foundation model and has been applied to various vision and multimodal tasks. However, recent works indicate that CLIP falls short in distinguishing detailed differences in images and shows suboptimal performance on dense-prediction and vision-centric multimodal tasks. Therefore, this work focuses on improving existing CLIP models, aiming to capture as many visual details in images as possible. We find that a specific type of generative models, unCLIP, provides a suitable framework for achieving our goal. Specifically, unCLIP trains an image generator conditioned on the CLIP image embedding. In other words, it inverts the CLIP image encoder. Compared to discriminative models like CLIP, generative models are better at capturing image details because they are trained to learn the data distribution of images. Additionally, the conditional input space of unCLIP aligns with CLIP's original image-text embedding space. Therefore, we propose to invert unCLIP (dubbed un$^2$CLIP) to improve the CLIP model. In this way, the improved image encoder can gain unCLIP's visual detail capturing ability while preserving its alignment with the original text encoder simultaneously. We evaluate our improved CLIP across various tasks to which CLIP has been applied, including the challenging MMVP-VLM benchmark, the dense-prediction open-vocabulary segmentation task, and multimodal large language model tasks. Experiments show that un$^2$CLIP significantly improves the original CLIP and previous CLIP improvement methods. Code and models will be available at https://github.com/LiYinqi/un2CLIP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。