不训练提升CLIP语义分割能力,通过修复视觉区分度实现高精度像素级理解。
Improving Visual Discriminability of CLIP for Training-Free Open-Vocabulary Semantic Segmentation
- 分析各层/头/令牌的视觉区分度,发现末层牺牲区分度、部分注意力头表现优异。
- 提出重加权、选头增强、异常令牌替换三项技术,显著提升分割精度。
- 无需训练或额外模型,适合快速部署于真实场景的开放词汇分割任务。
将CLIP模型扩展至语义分割仍具挑战,因其图像级预训练目标与像素级理解存在错位。现有方法虽通过重组最后一层和特征取得进展,但常继承前层全局对齐偏差,导致分割性能受限。本文提出LHT-CLIP,一种无需训练的框架,系统性挖掘CLIP在层、头、令牌三个层面的视觉区分度。经全面分析发现:(i) 最后几层(如ViT-B/16的最后3层,ViT-L/14的最后8层)主要强化图文对齐但牺牲视觉区分度;(ii) 部分注意力头(如ViT-B/16中10/144个)在多个数据集上持续展现强区分度;(iii) 异常令牌激活模式稀疏且一致。基于此,提出语义空间重加权、选择性头增强与异常令牌替换三项互补技术,无需训练、辅助网络或调参即可有效恢复视觉区分度。在8个常见分割基准上验证,LHT-CLIP在多种场景下均达当前最优,凸显其有效性与实用性。
原文摘要 · Abstract (English)
Extending CLIP models to semantic segmentation remains challenging due to the misalignment between their image-level pre-training objectives and the pixel-level visual understanding required for dense prediction. While prior efforts have achieved encouraging results by reorganizing the final layer and features, they often inherit the global alignment bias of preceding layers, leading to suboptimal segmentation performance. In this work, we propose LHT-CLIP, a novel training-free framework that systematically exploits the visual discriminability of CLIP across layer, head, and token levels. Through comprehensive analysis, we reveal three key insights: (i) the final layers primarily strengthen image-text alignment with sacrifice of visual discriminability (e.g., last 3 layers in ViT-B/16 and 8 layers in ViT-L/14), partly due to the emergence of anomalous tokens; (ii) a subset of attention heads (e.g., 10 out of 144 in ViT-B/16) display consistently strong visual discriminability across datasets; (iii) abnormal tokens display sparse and consistent activation pattern compared to normal tokens. Based on these findings, we propose three complementary techniques: semantic-spatial reweighting, selective head enhancement, and abnormal token replacement to effectively restore visual discriminability and improve segmentation performance without any additional training, auxiliary pre-trained networks, or extensive hyperparameter tuning. Extensive experiments on 8 common semantic segmentation benchmarks demonstrate that LHT-CLIP achieves state-of-the-art performance across diverse scenarios, highlighting its effectiveness and practicality for real-world deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。