让ViT更懂局部细节,通过选择性注意力提升分割精度
Vision Transformers Need Better Token Interaction

- 用稀疏注意力替代softmax,让视觉令牌间交互更精准
- 在ImageNet上训练200轮后,分割指标显著提升:VOC mIoU升至48.78
- 适合关注图像分割与密集预测的视觉模型研究者
视觉变换器(ViTs)在长期训练中能学习强大的图像级表征,但其块级表征在密集预测任务中表现下降。我们重新审视这一现象,发现其并非仅由高范数伪影导致,而是源于‘语义扩散’——全局语义信息通过块令牌过度传播。分析表明,局部性无法完全解释表征质量:浅层特征虽更对齐前景区域,却仍逊于深层特征;[CLS]特征对密集预测仍有互补价值。因此,不应消除全局上下文,而应使令牌交互更具选择性。我们以稀疏注意力(entmax-1.5)作为最小干预,保留全局连接性的同时替换softmax注意力。在ImageNet-1K上训练200轮的DINOv1 ViT-S/16上,该方法保持线性探测准确率,大幅改善语义分割性能:VOC mIoU从42.80提升至48.78,ADE20K从19.85升至21.97,Cityscapes从36.79增至37.87。结果表明,选择性令牌混合是一种简单有效的改进策略。
原文摘要 · Abstract (English)
Vision Transformers (ViTs) can learn strong image-level representations while their patch representations become less effective for dense prediction during prolonged training. We revisit this dense degradation phenomenon and argue that it is not fully explained by high-norm artifacts alone. Instead, we characterize \emph{semantic diffusion}: an optimization shortcut in which global semantic information spreads through patch tokens beyond what is locally justified. Our analysis shows that dense representation quality is not captured by locality alone: shallow features can remain better aligned with foreground regions yet underperform deeper features, and \texttt{[CLS]} features remain complementary for dense prediction. These observations suggest that the goal should not be to remove global context, but to make token interactions more selective. We therefore study sparse attention as a minimal intervention, replacing softmax attention with entmax-1.5 while preserving global token connectivity. On DINOv1 ViT-S/16 trained for 200 epochs on ImageNet-1K, this change preserves ImageNet linear probing accuracy and substantially improves semantic segmentation performance: VOC mIoU increases from 42.80 to 48.78, ADE20K from 19.85 to 21.97, and Cityscapes from 36.79 to 37.87. These results suggest that selective token mixing is a simple and effective bias for improving dense ViT representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。