Preprint
Machine Learning

Efficient Net V2

April 1, 2024

0

Citations

0

Influential Citations

Venue

2024

Year

Abstract

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.

Analysis

Why This Paper Matters

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.

Technical Contributions

  • Neural Architecture Search (NAS): The authors use a NAS framework to discover a family of models (EfficientNetV2-S, M, L) that balance depth, width, and resolution more effectively than hand-designed or previous NAS-found architectures.
  • Progressive Learning: A training schedule that starts with low-resolution images (e.g., 128x128) and weak regularization, then gradually increases resolution (to 224x224 or 300x300) and regularization strength (dropout, RandAugment). This reduces training time by allowing the model to learn coarse patterns early.
  • Fused-MBConv Blocks: A new building block that combines depthwise convolutions with standard convolutions to improve efficiency on modern hardware (e.g., TPUs, GPUs).
  • Scaling Strategy: The paper introduces a compound scaling method that jointly scales depth, width, and resolution, but with a focus on training speed rather than just accuracy.

Results

  • EfficientNetV2-M achieves 90.0% top-1 accuracy on ImageNet while training 5x faster than EfficientNet-B7.
  • EfficientNetV2-L trains 11x faster than previous state-of-the-art convolutional models on ImageNet.
  • On CIFAR-10/100 and other transfer learning benchmarks, EfficientNetV2 matches or exceeds prior models with significantly fewer training steps.
  • The progressive learning schedule alone reduces training time by 2-3x compared to standard training with fixed resolution.

Significance

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.