DL0073 EfficientNet Compound Scaling

How does EfficientNet scale networks, and what is compound scaling?

Answer

EfficientNet starts from a small baseline, EfficientNet-B0, produced by a multi-objective architecture search that rewards accuracy and FLOPs together, and then grows that fixed topology into the family B1 through B7 by scaling three dimensions simultaneously: depth (number of layers), width (number of channels), and input resolution. The empirical observation behind the method is that scaling any single dimension saturates: beyond a point, extra layers, extra channels, or extra pixels buy almost no accuracy while still costing compute. Compound scaling ties the three together through one user-chosen coefficient \phi and three fixed exponents \alpha, \beta, \gamma obtained by a small grid search on B0, subject to \alpha \cdot \beta^{2} \cdot \gamma^{2} \approx 2 so that each unit of \phi roughly doubles the FLOPs budget. With \alpha = 1.2, \beta = 1.1, \gamma = 1.15, the family climbs from 77.1% ImageNet top-1 at 0.39B FLOPs (B0) to 84.3% at 37B FLOPs (B7), matching the best accuracy of its era with about 8.4x fewer parameters than GPipe.

(1) The Baseline Is a Prerequisite: compound scaling only multiplies an existing topology, so a weak baseline yields a weak family; B0 is itself a search result built from MBConv blocks with squeeze-and-excitation, and the same scaling rule applied to MobileNet or ResNet gives smaller gains.
(2) Single-Dimension Scaling Saturates: very deep networks hit optimization and degradation limits, very wide shallow networks capture fine-grained patterns but few high-level ones, and resolution alone raises cost quadratically for shrinking returns.
(3) The Balance Rule: a larger input needs more layers to grow the receptive field and more channels to encode the finer patterns those extra pixels expose, which is why the three factors should move in a fixed ratio rather than one at a time.
(4) Two-Step Search: fix \phi = 1 and grid-search \alpha, \beta, \gamma once on the cheap baseline, then freeze them and sweep \phi to get B1 through B7, which avoids re-searching the architecture at every model size.

Line chart of ImageNet top-1 accuracy against FLOPs on a log axis for depth-only, width-only, resolution-only, and compound scaling, with the single-dimension curves flattening near 80 percent while compound scaling keeps rising past 81 percent

Figure 1: Illustrative accuracy-versus-compute curves from the same B0 baseline: depth-only, width-only, and resolution-only scaling flatten near 80% top-1, while compound scaling keeps converting FLOPs into accuracy.

The constraint has a direct cost interpretation. A standard convolution’s FLOPs scale linearly with the number of layers and quadratically with both channel count and spatial size, so total compute grows like d \cdot w^{2} \cdot r^{2}. Forcing \alpha \cdot \beta^{2} \cdot \gamma^{2} \approx 2 therefore makes \phi a clean compute dial: each additional unit costs about 2x the FLOPs, and the exponents decide how that doubled budget is split across the three dimensions. The exponents are searched once, on a model cheap enough that a small grid over \alpha, \beta, \gamma is affordable.

Mathematical Formulation:
d = \alpha^{\phi}
w = \beta^{\phi}
r = \gamma^{\phi}
\alpha \cdot \beta^{2} \cdot \gamma^{2} \approx 2
\mathrm{FLOPs}(\phi) \approx 2^{\phi} \cdot \mathrm{FLOPs}(0)

Where:

  • d, w, and r are the multipliers applied to the baseline’s layer count per stage, channel count per layer, and input side length.
  • \phi is the user-chosen compound coefficient that sets the resource budget; \phi = 0 recovers the baseline B0.
  • \alpha, \beta, \gamma are constants from a small grid search on B0 with \alpha \geq 1, \beta \geq 1, \gamma \geq 1; the published values are 1.2, 1.1, and 1.15.
  • Convolution cost scales as d \cdot w^{2} \cdot r^{2}, so the product constraint is what turns \phi into an approximate doubling of FLOPs per unit.
Schematic comparing the B0 baseline, drawn as a small input square feeding four short blocks, with the scaled B7 network, drawn as a larger input square feeding six taller blocks

Figure 2: The same topology at two budgets: a bigger input square (resolution), taller blocks (width), and more blocks (depth) all grow together instead of one dimension racing ahead.

ModelDepthWidthResolutionFLOPsImageNet Top-1
B01.0x1.0x2240.39B77.1%
B31.4x1.2x3001.8B81.6%
B52.2x1.6x4569.9B83.6%
B73.1x2.0x60037B84.3%

Two caveats matter in practice. The released coefficients are rounded rather than exact powers of a single \phi, so treat the formula as the design principle and the published table as the shipped configuration. More importantly, the objective is FLOPs, not latency or memory: depthwise separable convolutions have low arithmetic intensity and underuse GPU and TPU matrix units, and activation memory grows with r^{2}, so the largest variants train slowly and can exhaust device memory. EfficientNetV2 addressed exactly this by replacing early MBConv stages with Fused-MBConv, capping the maximum image size, and adding training-aware search plus progressive resizing.


Login to view more content


Log in to track your progress

Comments

Leave a Reply

Your email address will not be published. Required fields are marked *