Vision Transformers Need More Than Registers
Organizations: School of Computing and Data Science, The University of Hong Kong · Sun Yat-sen University
Abstract
Vision Transformers (ViTs), when pre-trained on large-scale data, provide general-purpose representations for diverse downstream tasks. However, artifacts in ViTs are widely observed across different supervision paradigms and downstream tasks. Through systematic analysis of artifacts in ViTs, we find that their fundamental mechanisms have yet to be sufficiently elucidated. In this paper, through systematic analysis, we conclude that these artifacts originate from a lazy aggregation behavior: ViT uses semantically irrelevant background patches as shortcuts to represent global semantics, driven by global attention and Coarse-grained semantic supervision. Our solution selectively integrates patch features into the CLS token, reducing the influence of background-dominated shortcuts and consistently improving performance across 12 benchmarks under label-, text-, and self-supervision. We hope this work offers a new perspective on ViT behavior.
Figures & tables
| Model | Backbone | COCO-Obj. | ADE20K | City. | VOC20 | Context59 | COCO-Stf. |
|---|---|---|---|---|---|---|---|
| CLIP [ 27 ] | ViT-B/16 | 8.8 | 3.1 | 6.5 | 49.0 | 11.2 | 7.2 |
| CLIP ( + LazyStrike ) | ViT-B/16 | 13.3 ( +4.5 ) | 8.3 ( +5.2 ) | 12.1 ( +5.6 ) | 75.0 ( +26.0 ) | 15.2 ( +4.0 ) | 11.8 ( +4.6 ) |
| MetaCLIP [ 39 ] | ViT-B/16 | 4.8 | 2.9 | 5.8 | 39.6 | 9.3 | 6.2 |
| MetaCLIP ( + LazyStrike ) | ViT-B/16 | 14.1 ( +9.3 ) | 7.9 ( +5.0 ) | 11.1 ( +5.3 ) | 72.8 ( +33.2 ) | 15.5 ( +6.2 ) | 12.0 ( +5.8 ) |
| EVACLIP [ 32 ] | ViT-B/16 | 15.0 | 6.7 | 12.2 | 56.5 | 14.1 | 9.7 |
| EVACLIP ( + LazyStrike ) | ViT-B/16 | 26.2 ( +11.2 ) | 14.8 ( +8.1 ) | 24.5 ( +12.3 ) | 79.6 ( +23.1 ) | 24.7 ( +10.6 ) | 18.3 ( +8.6 ) |
| Method | Backbone | COCO Detection | LVIS Segmentation | ||||||
|---|---|---|---|---|---|---|---|---|---|
| AP50 | AP50 | AP50 | AP | AP | AP | AP | |||
| ConvNet based | |||||||||
| F-VLM [ 16 ] | RN50 | 1 39.6 | 1 / | 1 28.0 | 1 24.2 | 1 26.9 | 1 24.0 | 1 18.6 | |
| F-VLM [ 16 ] | RN50x64 | 1 / | 1 / | 1 / | 1 34.9 | 1 / | 1 / | 1 32.8 | |
| ViT based | |||||||||
| F-ViT [ 37 ] | ViT-B/16 | 1 34.9 | 1 41.0 | 1 17.5 | 1 15.4 | 1 20.6 | 1 12.3 | 1 11.5 | |
Explore similar work
Vision Transformers Need Better Token Interaction
[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.