用视觉粒化增强文本提示,让CLIP更好区分细粒度图像
Causality-guided Prompt Learning for Vision-language Models via Visual Granulation
- 通过视觉粒化分解图像特征,分离共性与个性属性
- 在15个数据集上显著超越现有提示学习方法,尤其在细粒度任务中
- 适合需要高精度图像分类的视觉语言模型研究者
提示学习近年来受到广泛关注,用于将预训练视觉语言模型(如CLIP)适配到下游识别任务。然而,大多数基于CLIP的提示学习方法在处理细粒度数据集时表现有限。为解决此问题,我们提出一种基于因果引导的文本提示学习方法——CaPL,通过视觉粒化技术构建视觉粒化集,利用因果推断捕捉不同细粒度类别间的细微差异。CaPL包含两个模块:(1) 属性解耦模块,使用布朗桥扩散模型将视觉特征分解为非个体属性(部分类别共享)和个体属性(仅对应单一类别);(2) 粒化学习模块,结合上述属性,通过两种因果推断策略构建视觉粒化以支持识别。得益于学习到的视觉粒化,可生成更具判别力的文本提示。在15个数据集上的大量实验表明,我们的方法显著优于当前最优提示学习方法,尤其在细粒度数据集上表现突出。
原文摘要 · Abstract (English)
Prompt learning has recently attracted much attention for adapting pre-trained vision-language models (e.g., CLIP) to downstream recognition tasks. However, most of the existing CLIP-based prompt learning methods only show a limited ability for handling fine-grained datasets. To address this issue, we propose a causality-guided text prompt learning method via visual granulation for CLIP, called CaPL, where the explored visual granulation technique could construct sets of visual granules for the text prompt to capture subtle discrepancies among different fine-grained classes through casual inference. The CaPL method contains the following two modules: (1) An attribute disentanglement module is proposed to decompose visual features into non-individualized attributes (shared by some classes) and individualized attributes (specific to single classes) using a Brownian Bridge Diffusion Model; (2) A granule learning module is proposed to construct visual granules by integrating the aforementioned attributes for recognition under two causal inference strategies. Thanks to the learned visual granules, more discriminative text prompt is expected to be learned. Extensive experimental results on 15 datasets demonstrate that our CaPL method significantly outperforms the state-of-the-art prompt learning methods, especially on fine-grained datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。