用扩散模型增强CLIP的密集知识,提升弱监督分割精度
DiCLIP: Diffusion Model Enhances CLIP's Dense Knowledge for Weakly Supervised Semantic Segmentation

- 引入扩散模型提升CLIP在视觉与文本模态的密集表征能力
- 在PASCAL VOC和MS COCO上优于当前最优方法,训练成本更低
- 适合关注弱监督分割与多模态知识融合的研究者
基于图像级标签的弱监督语义分割(WSSS)通常依赖类别激活图(CAM)实现像素级预测。近期,对比语言-图像预训练(CLIP)被用于生成CAM。然而,现有WSSS方法仅利用CLIP的跨模态对齐特性,忽视其在视觉与文本模态中固有的稀疏密集知识,导致CAM生成效果不佳。本文提出DiCLIP,一种新型WSSS框架,通过生成式扩散模型增强CLIP在双模态中的密集知识。设计了视觉相关性增强(VCE)与文本语义增强(TSA)模块:VCE利用扩散模型可靠的空间一致性缓解CLIP注意力的过平滑问题,通过注意力聚类精化(ACR)模块从扩散模型中提取多样相关图,作为多样性偏差引导CLIP自注意力向更具判别性的密集分布演进;TSA认为单一文本模态无法覆盖视觉类别的变异性,因此借助扩散模型的生成能力构建动态键值缓存模型,将CAM生成从局部块-文本匹配机制转变为视觉知识检索范式。实验表明,DiCLIP在PASCAL VOC和MS COCO数据集上均超越现有方法,且显著降低训练成本。代码已公开于https://github.com/zwyang6/DiCLIP。
原文摘要 · Abstract (English)
Weakly Supervised Semantic Segmentation (WSSS) with image-level labels typically leverages Class Activation Maps (CAMs) to achieve pixel-level predictions. Recently, Contrastive Language-Image Pre-training (CLIP) has been introduced to generate CAMs in WSSS. However, previous WSSS methods solely adopt CLIP's vision-language paired property for dense localization, neglecting its inherently limited dense knowledge across both visual and text modalities, which renders CAM generation suboptimal. In this work, we propose DiCLIP, a novel WSSS framework that leverages the generative diffusion model to enhance CLIP's dense knowledge across two modalities. Specifically, Visual Correlation Enhancement (VCE) and Text Semantic Augmentation (TSA) modules are proposed for dense prediction enhancement. To improve the spatial awareness of visual features, our VCE module utilizes diffusion's reliable spatial consistency to mitigate the over-smoothing issue in CLIP's attention. It designs the Attention Clustering Refinement (ACR) module to reliably extract diverse correlation maps from the diffusion model. The correlation maps act as a diversity bias for CLIP's self-attention, recursively pushing its visual features towards a more discriminative dense distribution. To augment the semantics of text embeddings, our TSA module argues that a single text modality is insufficient to encompass the variability of visual categories. Thus, we leverage diffusion's generative power to maintain a dynamic key-value cache model, shifting CAM generation from a patch-text matching mechanism to a novel visual knowledge retrieval paradigm. With these enhancements, DiCLIP not only outperforms state-of-the-art methods on PASCAL VOC and MS COCO but also significantly reduces training costs. Code is publicly available at https://github.com/zwyang6/DiCLIP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。