Preprint
Large Language Models

ConvNets Match Vision Transformers at Scale

October 1, 2023

0

Citations

0

Influential Citations

Venue

2023

Year

Abstract

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.

Analysis

Why This Paper Matters

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.

Technical Contributions

  • Empirical comparison under controlled conditions: The paper ensures that both ConvNets and ViTs are pre-trained on the same large dataset and fine-tuned on ImageNet, isolating architecture as the primary variable.
  • Focus on NFNets: NFNets are a specific ConvNet variant that uses normalization-free training and adaptive gradient clipping, which may contribute to their strong scaling properties.
  • Challenging the scaling narrative: By showing that ConvNets can match ViTs at scale, the paper reframes the debate from "which architecture is best" to "how to best train any architecture at scale."

Results

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.

Significance

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.