English

Bayesian Uncertainty Estimation for Batch Normalized Deep Networks

Machine Learning 2018-07-17 v2

Abstract

We show that training a deep network using batch normalization is equivalent to approximate inference in Bayesian models. We further demonstrate that this finding allows us to make meaningful estimates of the model uncertainty using conventional architectures, without modifications to the network or the training procedure. Our approach is thoroughly validated by measuring the quality of uncertainty in a series of empirical experiments on different tasks. It outperforms baselines with strong statistical significance, and displays competitive performance with recent Bayesian approaches.

Keywords

Cite

@article{arxiv.1802.06455,
  title  = {Bayesian Uncertainty Estimation for Batch Normalized Deep Networks},
  author = {Mattias Teye and Hossein Azizpour and Kevin Smith},
  journal= {arXiv preprint arXiv:1802.06455},
  year   = {2018}
}

Comments

ICML 2018

R2 v1 2026-06-23T00:25:54.595Z