Rethinking the Inception Architecture for Computer Vision

2015 Multi-GPU dense training 565 citations
Download PDF

Rethinking the Inception Architecture for Computer Vision

Metadata

  • Slug: inception_v3_2015
  • Year: 2015
  • Venue: CVPR
  • Authors: Christian Szegedy et al.
  • Reading status: read complete
  • Compute regime: Multi-GPU dense training
  • Primary sources: PDF, extracted text

Compute Setup

The paper explicitly lists TensorFlow distributed training with 50 replicas, each running on an NVIDIA Kepler GPU. It uses batch size 32 per replica and trains for 100 epochs, with learning rate decayed every two epochs at exponential rate 0.94. This puts Inception v3 in the multi-GPU dense CNN regime, where architecture is judged by ImageNet accuracy, computation, and parameter storage.

The paper also gives a concrete inference budget: 21.2% top-1 and 5.6% top-5 single-frame error with less than 25M parameters and about 5B multiply-adds per inference. The ensemble result uses four models and multi-crop evaluation, so the best leaderboard-style number spends far more inference compute than the single-frame model.

Bottleneck

The bottleneck is naive CNN scaling. Larger filters, wider layers, and denser stacks improve accuracy but multiply computation and parameter count. The paper points to GoogLeNet's 5M parameters as a 12x reduction relative to AlexNet, while VGGNet used about 3x more parameters than AlexNet. That contrast frames the design problem: VGG-style monolithic convolution is accurate but expensive, while Inception-style factorization tries to keep representations expressive under a fixed compute budget.

The design principles target memory and compute: avoid representational bottlenecks, balance width and depth, factorize large convolutions into cheaper sequences, and reduce grid size without spending most computation on the larger grid. The bottleneck is the whole CNN graph, not one kernel.

Method Adaptation

Inception v3 adapts the CNN graph through factorization. A 5 x 5 convolution with the same input and output channel counts costs 25/9 = 2.78 times a 3 x 3 convolution; replacing it with two 3 x 3 layers gives a reported net gain of about 28%. It also factorizes n x n convolutions into 1 x n followed by n x 1; for 3 x 3 this is 33% cheaper when input and output filter counts are equal. The chosen architecture uses 1 x 7 followed by 7 x 1 on 17 x 17 grids.

Grid-size reduction is another compute adaptation. The paper uses parallel pooling and strided convolution branches whose outputs are concatenated, reducing spatial resolution while increasing channel count. This avoids bottlenecks without paying the full cost of large convolutions on unreduced grids.

The training regularizers also fit the distributed setup. Label smoothing replaces hard one-hot targets with a mixture using uniform probability, reducing overconfidence and improving generalization. Batch-normalized auxiliary classifiers are used as regularizers rather than primarily as gradient shortcuts; the paper says auxiliary classifiers did not improve early convergence but helped near the end of training.

Evidence

The core compute-quality result is the single-frame model: 21.2% top-1 and 5.6% top-5 error on ILSVRC 2012 validation, with about 5B multiply-adds and fewer than 25M parameters. The conclusion says this is a 2.5x compute increase over the BatchNorm-Inception-style predecessor but still much less than denser networks. It also states that the model outperforms results from He et al. while being six times cheaper computationally and using at least five times fewer parameters.

The multi-crop results show how inference budget changes accuracy. A single Inception-v3 with 12 crops reaches 19.47% top-1 and 4.48% top-5 error; with 144 crops it reaches 18.77% top-1 and 4.2% top-5. The ensemble table reports four Inception-v3 models with 144 crops at 17.2% top-1 and 3.58% top-5, with a validation top-5 of 3.46%. These numbers are higher-compute evaluation modes and should not be confused with the 5B multiply-add single-frame budget.

The ablations support the changes. Label smoothing improves top-5 and top-1 errors by around 0.2 percentage points. Factorized 7 x 7 convolutions and batch-normalized auxiliary classifiers are reported as cumulative modifications, and lower-resolution receptive fields can still reach high quality.

Historical Effect

Inception v3 codified compute-aware CNN design beyond the Inception family: factorize large spatial filters, avoid narrow graph cuts, balance width and depth, use grid reduction modules instead of blunt downsampling, and treat label smoothing as a simple regularizer.

Historically, the paper connects GoogLeNet's hand-designed efficiency to later architectures that systematically decomposed convolution. It also provides a clear example of multi-GPU architecture engineering before automated architecture search became widespread: the design is evaluated under a real distributed TensorFlow setup and a stated per-inference multiply-add budget.

Limits

The design principles are partly heuristic. The paper explicitly frames some guidance as useful principles rather than formal guarantees, and the resulting architecture remains complex compared with later cleaner blocks. Its best numbers also mix different inference budgets: single-frame, 12-crop, 144-crop, and ensemble results are not interchangeable from a compute perspective.

The reported setup assumes TensorFlow distributed training with 50 Kepler GPU replicas. The paper does not provide a single-GPU training-time equivalent, power budget, or detailed input pipeline measurements. Finally, the local extracted text is interleaved by the two-column PDF layout, so table values are recoverable but some prose around the experimental tables is fragmented.

Links