Timezone: »

Regularizing by the Variance of the Activations' Sample-Variances
Etai Littwin · Lior Wolf

Tue Dec 04 07:45 AM -- 09:45 AM (PST) @ Room 210 #62

Normalization techniques play an important role in supporting efficient and often more effective training of deep neural networks. While conventional methods explicitly normalize the activations, we suggest to add a loss term instead. This new loss term encourages the variance of the activations to be stable and not vary from one random mini-batch to the next. As we prove, this encourages the activations to be distributed around a few distinct modes. We also show that if the inputs are from a mixture of two Gaussians, the new loss would either join the two together, or separate between them optimally in the LDA sense, depending on the prior probabilities. Finally, we are able to link the new regularization term to the batchnorm method, which provides it with a regularization perspective. Our experiments demonstrate an improvement in accuracy over the batchnorm technique for both CNNs and fully connected networks.

Author Information

Etai Littwin (Apple)
Lior Wolf (Facebook AI Research)

More from the Same Authors