Forourensemble,weused6networks. Eachwasbased
onBN-x30,modiﬁedviasomeofthefollowing: increased
initial weights in the convolutionallayers; using Dropout
(with the Dropout probability of 5% or 10%, vs. 40%
for the original Inception); and using non-convolutional,
per-activation Batch Normalization with last hidden layers of the model. Each network achieved its maximum
accuracyafter about 6·106training steps. The ensemble
prediction was based on the arithmetic average of class
probabilities predicted by the constituent networks. The
detailsofensembleandmulticropinferencearesimilarto
(Szegedyet al., 2014).
We demonstrate in Fig. 4 that batch normalization allowsusto set new state-of-the-artby a healthymarginon
theImageNetclassiﬁcationchallengebenchmarks.
5 Conclusion
We have presented a novel mechanism for dramatically
accelerating the training of deep networks. It is based on
the premise that covariate shift, which is known to complicate the trainingof machine learning systems, also ap7
Model Resolution Crops Models Top-1error Top-5error
GoogLeNetensemble 224 144 7 - 6.67%
Training Deep Neural Networks is complicated by the fact that the
distribution of each layer's inputs changes during training, as the parameters
of the previous layers change. This slows down the training by requiring lower
learning rates and careful parameter initialization, and makes it notoriously
hard to train models with saturating nonlinearities. We refer to this
phenomenon as internal covariate shift, and address the problem by normalizing
layer inputs. Our method draws its strength from making normalization a part of
the model architecture and performing the normalization for each training
mini-batch. Batch Normalization allows us to use much higher learning rates and
be less careful about initialization. It also acts as a regularizer, in some
cases eliminating the need for Dropout. Applied to a state-of-the-art image
classification model, Batch Normalization achieves the same accuracy with 14
times fewer training steps, and beats the original model by a significant
margin. Using an ensemble of batch-normalized networks, we improve upon the
best published result on ImageNet classification: reaching 4.9% top-5
validation error (and 4.8% test error), exceeding the accuracy of human raters.
Sergey Ioffe Christian Szegedy