Revisiting [CLS] and Patch Token Interaction in Vision Transformers
Not on the map. Why?
Abstract (arXiv)
Vision Transformers have emerged as powerful, scalable and versatile representation learners. To capture both global and local features, a learnable [CLS] class token is typically prepended to the input sequence of patch tokens. Despite their distinct nature, both token types are processed identically throughout the model. In this work, we investigate the friction between global and local feature learning under different pre-training strategies by analyzing the interactions between class and patch tokens. Our analysis reveals that standard normalization layers introduce an implicit differentiation between these token types. Building on this insight, we propose specialized processing paths that selectively disentangle the computational flow of class and patch tokens, particularly within normalization layers and early query-key-value projections. This targeted specialization leads to significantly improved patch representation quality for dense prediction tasks. Our experiments demonstrate segmentation performance gains of over 2 mIoU points on standard benchmarks, while maintaining strong classification accuracy. The proposed modifications introduce only an 8% increase in parameters, with no additional computational overhead. Through comprehensive ablations, we provide insights into which architectural components benefit most from specialization and how our approach generalizes across model scales and learning frameworks.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 28 of its 37 references that are in the map. First posted on arXiv 9 Feb 2026.
Builds on
28 of 37 references.
References · 28
- 2017Attention Is All You Needcited by 59,932
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2020End-to-End Object Detection with Transformerscited by 7,384
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
16 more
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2025DINOv3cited by 1,297
- 2025SigLIP 2: Multilingual Vision-Language Encoders with Improved Semantic Understanding, Localization, and Dense Featurescited by 1,127
- 2021Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetcited by 1,096
- 2022Token Merging: Your ViT But Fastercited by 995
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2023Vision Transformers Need Registerscited by 844
- 2021Going deeper with Image Transformerscited by 759
- 2021ImageNet-21K Pretraining for the Massescited by 607
- 2022DeiT III: Revenge of the ViTcited by 417
- 2025Perception Encoder: The best visual embeddings are not at the output of the networkcited by 315
- 2024Massive Activations in Large Language Modelscited by 277
- 2024On the Role of Attention Masks and LayerNorm in Transformerscited by 45
- 2025Systematic Outliers in Large Language Modelscited by 32
- 2024Near, far: Patch-ordering enhances vision foundation models' scene understandingcited by 15
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026UniDriveVLA: Unifying Understanding, Perception, and Action Planning for Autonomous Drivingcited by 17
- 2026Registers Matter for Pixel-Space Diffusion Transformerscited by 2
- 2026Resolving Representation Ambiguity in Feedforward Novel View Synthesis Transformer via Semantic-Spatial Decouplingcited by 0
- 2026Gated Spatial Redundancy Projection for Pathology Transformer Attentionscited by 0
- 2026Gated Spatial Redundancy Projection for Pathology Transformer Attentions8 Aug 2026
- 2026Resolving Representation Ambiguity in Feedforward Novel View Synthesis Transformer via Semantic-Spatial Decoupling18 May 2026
- 2026Registers Matter for Pixel-Space Diffusion Transformers15 May 2026
- 2026UniDriveVLA: Unifying Understanding, Perception, and Action Planning for Autonomous Driving2 Apr 2026