ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
0
Citations
0
Influential Citations
—
Venue
2024
Year
A new family of convolutional networks, achieves faster training speed and better parameter efficiency than previous models through neural architecture search and scaling, with progressive learning allowing for improved accuracy on various datasets while training up to 11x faster.
EfficientNet V2 addresses a critical bottleneck in deep learning: the high computational cost of training state-of-the-art convolutional neural networks. While previous EfficientNet models achieved excellent parameter efficiency, they still required long training times. This paper shows that by jointly optimizing architecture and training strategy, one can achieve up to 11x faster training without sacrificing accuracy. This is particularly significant for AI practitioners who need to iterate quickly or deploy models in resource-constrained environments.
The progressive learning approach is a key insight—by starting with small images and low regularization, the model learns coarse features quickly, then gradually increases difficulty. This mimics human learning and stabilizes training, allowing the use of larger batch sizes and higher learning rates. The combination of neural architecture search (NAS) with progressive learning is novel and sets a new standard for efficient training.
EfficientNet V2 has broad implications for the AI field. By making high-accuracy models trainable in hours instead of days, it lowers the barrier to entry for researchers and startups. The progressive learning technique is model-agnostic and could be applied to other architectures (e.g., transformers). This work also reinforces the importance of co-designing architecture and training algorithms—a trend that is likely to grow as compute costs remain a primary concern. For practitioners, EfficientNetV2 offers a practical drop-in replacement for older EfficientNets and many ResNet variants, providing both speed and accuracy gains.
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