ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
0
Citations
0
Influential Citations
—
Venue
2023
Year
Challenges the belief that Vision Transformers outperform ConvNets on large datasets by demonstrating that ConvNets, specifically NFNets, achieve comparable performance when pre-trained on a large dataset and fine-tuned on ImageNet.
This paper directly challenges a widely held assumption in computer vision: that Vision Transformers (ViTs) inherently outperform Convolutional Neural Networks (ConvNets) when trained on large datasets. The prevailing narrative has been that ViTs' self-attention mechanisms capture long-range dependencies more effectively than convolutions, leading to superior scaling behavior. By demonstrating that NFNets—a modern ConvNet architecture—can match ViT performance under the same pre-training and fine-tuning conditions, the authors provide a critical counterpoint. This matters because it shifts the focus from architecture choice to training methodology and data scale, suggesting that the advantages of ViTs may be overstated or contingent on specific training recipes.
For AI practitioners, this finding is practically significant: it means that ConvNets remain a viable and potentially more efficient option for large-scale vision tasks, especially in resource-constrained environments where the computational overhead of self-attention may be prohibitive. The paper encourages a more nuanced evaluation of model architectures, emphasizing that performance gains often attributed to architectural innovations may actually stem from differences in training protocols.
The key result is that NFNets achieve performance on ImageNet that is comparable to Vision Transformers when both are pre-trained on a large dataset (e.g., JFT-300M or similar). The paper does not provide specific numerical metrics in the abstract, but the central claim is that the performance gap is negligible under these conditions. This contradicts prior work that reported ViTs outperforming ConvNets by several points on ImageNet after large-scale pre-training.
This paper has broad implications for the AI field. It suggests that the community may have prematurely dismissed ConvNets in favor of Transformers for vision tasks. The findings encourage researchers to revisit ConvNet architectures with modern training techniques, potentially leading to more efficient models that require less computational resources than ViTs. For practitioners, it means that existing ConvNet-based infrastructure and expertise remain valuable, and that the choice between ConvNets and ViTs should be guided by practical considerations (e.g., latency, memory) rather than assumed superiority. The work also underscores the importance of controlled experiments in AI research, where confounding factors like training recipes can obscure true architectural comparisons.
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