vix.ing · top · new · best · stats

Weight Uncertainty in Neural Networks

2015/05/20 by Charles Blundell, Blundell, Charles, Julien Cornebise +5 · 1,284 citations
Computer Science · Decision Sciences · Mathematics · #Advanced Bandit Algorithms Research #Adversarial Robustness in Machine Learning #Artificial intelligence #Artificial neural network #Backpropagation #Bayes' theorem #Bayesian probability #Computer science #Dropout (neural networks) #Energy (signal processing) #FOS: Computer and information sciences #MNIST database #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Machine learning #Mathematical optimization #Mathematics #Reinforcement Learning in Robotics #Reinforcement learning #Statistics #Upper and lower bounds #cs.LG #stat.ML

paper · pdf · doi:10.48550/arxiv.1505.05424

published in arXiv (Cornell University) (Cornell University) · In Proceedings of the 32nd International Conference on Machine Learning (ICML 2015)

openalex publication_date 2015/05/20 · arxiv created 2015/05/21 · arxiv updated 2015/05/22 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

We introduce a new, efficient, principled and backpropagation-compatible algorithm for learning a probability distribution on the weights of a neural network, called Bayes by Backprop. It regularises the weights by minimising a compression cost, known as the variational free energy or the expected lower bound on the marginal likelihood. We show that this principled kind of regularisation yields comparable performance to dropout on MNIST classification. We then demonstrate how the learnt uncertainty in the weights can be used to improve generalisation in non-linear regression problems, and how this weight uncertainty can be used to drive the exploration-exploitation trade-off in reinforcement learning.

Citations

Cited by

Related