Motivation – scaling under budget constraints

self.conv = nn.Conv2d(in_channels=3, out_channels=64, kernel_size=3, stride=1, padding=1)
  1. You've all been there:
    You're tuning your CNN — more layers, more channels, or higher resolution?
  2. Industry reality:
    You built a prototype on a tight budget. But how to scale it for production?

Problem – Insight – Method – Result

Problem:
Scaling CNNs by tuning depth, width, or resolution — but always one at a time, which is suboptimal
Insight:
Depth, width, and resolution are interdependent — scaling them together in a fixed ratio results in better accuracy & efficiency
Method:
A compound scaling rule (ϕ) to jointly scale depth, width, and resolution together
Result:
EfficientNet — a family of models (B0–B7) that outperform prior networks at a fraction of the parameters

Importance of resolution

What is it?
Photo: Henrique Ferreira / Unsplash

The CNN Arms Race

2012
AlexNet
ImageNet winner 2012
60M Params
2015
ResNet
ImageNet winner 2015
60M Params
2018
GPipe
ImageNet record 2018
557M Params

Observation 1 – scaling any single dimension improves accuracy

75 76 77 78 79 80 81 0 2 4 6 8 FLOPS (billions) ImageNet Top-1 accuracy (%) w=1.0 w=1.4 w=1.8 w=2.6 w=3.8 w=5.0 width w 75 76 77 78 79 80 81 0 1 2 3 4 FLOPS (billions) d=1.0 d=2.0 d=3.0 d=4.0 d=6.0 d=8.0 depth d 75 76 77 78 79 80 81 0 1 2 3 FLOPS (billions) r=1.0 r=1.3 r=1.5 r=1.7 r=1.9 r=2.2 r=2.5 resolution r
~80% accuracy ceiling
Redrawn from values read off Tan & Le (2019), Figure 3 (EfficientNet-B0 baseline).

Observation 2 – but scaling all dimensions together is better

76 77 78 79 80 81 82 0 5 10 15 20 25 FLOPS (billions) ImageNet Top-1 accuracy (%) d=1.0, r=1.0 d=1.0, r=1.3 d=2.0, r=1.0 d=2.0, r=1.3
Redrawn from values read off Tan & Le (2019), Figure 4 (width scaled at each depth d and resolution r).

Compound scaling rule

d = αϕ, w = βϕ, r = γϕ
subject to   α · β2 · γ2 ≈ 2   with α, β, γ ≥ 1

EfficientNet vs. the Field — same accuracy, fewer resources

ModelTop-1Top-5#ParamsRatio#FLOPsRatio
EfficientNet-B077.1%93.3%5.3M1×0.39B1×
ResNet-5076.0%93.0%26M4.9×4.1B11×
DenseNet-16976.2%93.2%14M2.6×3.5B8.9×
EfficientNet-B179.1%94.4%7.8M1×0.70B1×
ResNet-15277.8%93.8%60M7.6×11B16×
Inception-v378.8%94.4%24M3.0×5.7B8.1×
EfficientNet-B280.1%94.9%9.2M1×1.0B1×
Inception-v480.0%95.0%48M5.2×13B13×
Inception-resnet-v280.1%95.1%56M6.1×13B13×
EfficientNet-B381.6%95.7%12M1×1.8B1×
ResNeXt-10180.9%95.6%84M7.0×32B18×
EfficientNet-B482.9%96.4%19M1×4.2B1×
SENet82.7%96.2%146M7.7×42B10×
NASNet-A82.7%96.2%89M4.7×24B5.7×
AmoebaNet-A82.8%96.1%87M4.6×23B5.5×
EfficientNet-B583.6%96.7%30M1×9.9B1×
AmoebaNet-C83.5%96.5%155M5.2×41B4.1×
EfficientNet-B684.0%96.8%43M1×19B1×
EfficientNet-B784.3%97.0%66M1×37B1×
GPipe84.3%97.0%557M8.4×——
We omit ensemble and multi-crop models, or models pretrained on 3.5B Instagram images.

Transfer Learning – What is it?

Option A – Train from scratch
Hire someone who has never cooked before — train them from scratch
Option B – Transfer Learning
Hire a French chef — teach them sushi
Simon Says Sushi

Transfer Learning – EfficientNet Performance Results

Comparison to best reported results
ModelAcc.#ParamOur ModelAcc.#Param(ratio)
CIFAR-10†GPipe99.0%556MEfficientNet-B798.9%64M (8.7×)
CIFAR-100GPipe91.3%556MEfficientNet-B791.7%64M (8.7×)
BirdsnapGPipe83.6%556MEfficientNet-B784.3%64M (8.7×)
Stanford Cars‡DAT94.8%–EfficientNet-B794.7%–
FlowersDAT97.7%–EfficientNet-B798.8%–
FGVC AircraftDAT92.9%–EfficientNet-B792.9%–
Oxford-IIIT PetsGPipe95.9%556MEfficientNet-B695.4%41M (14×)
Food-101GPipe93.0%556MEfficientNet-B793.0%64M (8.7×)
Geo-Mean(9.6×)
EfficientNet is Gordon Ramsay

Key takeaways

Compound Model Scaling:
single dimension scaling  <<  compound scaling
Key formula: d = αϕ, w = βϕ, r = γϕ ⟹ α = 1.2, β = 1.1, γ = 1.15
EfficientNet:
Achieves higher accuracy with less parameters and FLOPs
EfficientNet learns better general representations - not just better ImageNet performance!

Outlook – what happened since?

Direct successor:
EfficientNetV2 (Tan & Le, 2021)
Vision Transformers dethroned CNNs in 2020 (ViT):
but the scaling insight lives on

Compound scaling rule – constraints

Subject to the constraint:
α · β2 · γ2 ≈ 2      with   α ≥ 1,  β ≥ 1,  γ ≥ 1
FLOPS scale differently per dimension:
  • Doubling depth → FLOPS ×2  (linear)
  • Doubling width → FLOPS ×4  (quadratic)
  • Doubling resolution → FLOPS ×4  (quadratic, pixels scale in 2D)
Each ϕ step → total FLOPS × 2ϕ