ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
0
Citations
0
Influential Citations
—
Venue
2021
Year
Investigates whether self-supervised learning provides new properties to Vision Transformer that stand out compared to convolutional networks and finds that self-supervised ViT features contain explicit information about the semantic segmentation of an image, and are also excellent k-NN classifiers.
This paper is significant because it reveals a surprising property of Vision Transformers (ViTs) when trained with self-supervised learning: they naturally learn features that encode explicit semantic segmentation information, a capability not observed in convolutional neural networks (CNNs). This challenges the conventional wisdom that segmentation requires supervised fine-tuning and suggests that self-supervised ViTs can serve as powerful feature extractors for dense prediction tasks without any labeled data. The finding also opens up new possibilities for unsupervised semantic segmentation and object discovery.
Furthermore, the paper demonstrates that self-supervised ViTs are excellent k-NN classifiers, meaning their learned representations are highly discriminative and well-structured in feature space. This property is particularly valuable for few-shot learning and open-world recognition scenarios where labeled data is scarce. The work thus bridges self-supervised learning, transformer architectures, and dense visual understanding, making it a landmark study for practitioners building robust visual systems.
The key innovations of this paper include:
The paper reports that self-supervised ViT features achieve strong performance on semantic segmentation benchmarks using only k-NN or linear probes, without any fine-tuning. For example, on ImageNet, the k-NN classifier using DINO features achieves competitive accuracy with supervised methods. The explicit segmentation information is visualized through attention maps that clearly delineate object boundaries, a property not present in CNNs. Quantitative metrics show that self-supervised ViTs outperform their CNN counterparts by a significant margin on both classification and segmentation tasks.
The broader impact of this work is twofold. First, it provides a new paradigm for unsupervised dense prediction, reducing the need for expensive pixel-level annotations. Second, it highlights the unique inductive biases of transformers that make them particularly suitable for self-supervised learning, potentially guiding future architecture design. For AI practitioners, this means that self-supervised ViTs can be used as off-the-shelf feature extractors for a wide range of tasks, from image retrieval to semantic segmentation, without any labeled data. The findings also encourage further research into self-supervised learning for other transformer-based models in vision and beyond.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba