Practical Variational Inference for Neural Networks
Proposes a stochastic variational inference method with a diagonal Gaussian posterior that scales practical Bayesian learning, regularisation, and weight pruning to general differentiable neural network architectures.
Complex neural networks often struggle with overfitting and high computational demands, while traditional Bayesian methods that quantify uncertainty have remained mathematically intractable for all but the simplest architectures. This article develops and evaluates a practical, stochastic variational inference method that can be applied to virtually any standard neural network trained with gradient descent, framing the learning problem through a Minimum Description Length lens.
The approach replaces intractable analytical derivations with numerical sampling over a diagonal Gaussian weight distribution, allowing simultaneous learning of weight means and variances. The article evaluated this framework alongside an information-theoretic pruning rule on a 15-layer recurrent neural network using the benchmark TIMIT speech dataset (3,696 training utterances).
Key findings demonstrate that adaptive weight noise achieved a 23.8% phoneme error rate on the test set, noticeably outperforming standard maximum likelihood training (27.1%) and fixed regularizers like weight decay (27.4%). Because the network actively compressed the data, it proved resistant to overfitting, eliminating the need to sacrifice training data for early stopping validation. Furthermore, applying the proposed pruning rule allowed removing between 55% and 78% of the network weights while slightly reducing the error rate further to 23.3% after retraining.
These results provide a straightforward method for practitioners to train compact, robust neural models that generalize better without requiring custom Bayesian derivations or complex tuning. The method directly reduces hardware memory overhead and inferencing costs by eliminating unneeded weights.
Organizations deploying large neural architectures should consider adopting adaptive weight distributions to improve generalization and using the signal-to-noise pruning heuristic to streamline model size. Next steps should explore validating the method across additional domains, though teams should note that training with stochastic derivatives increases training times and introduces noisy optimization curves.
- Paper: Keeping Neural Networks Simple by Minimizing the Description Length of the Weights, Geoffrey E. Hinton et al. (1993). This seminal paper introduced Minimum Description Length (MDL) and variational weight penalties to keep neural network weights simple, which the source directly builds upon and generalises to arbitrary architectures.
- Paper: Optimal Brain Damage, Yann LeCun et al. (1989). This foundational work establishes second-order saliency pruning heuristics for neural network compression, motivating the variational pruning criteria developed in the source.
- Paper: A Simple Weight Decay Can Improve Generalization, A. Krogh et al. (1991). This paper provides the classical understanding of weight decay regularization, which the source re-interprets and analyses through a variational lens.
- Paper: An Introduction to Variational Methods for Graphical Models, MICHAEL I. JORDAN et al. (1999). This comprehensive overview establishes the foundational mathematical principles of variational inference and lower bounding intractable distributions in graphical and neural models.
- Paper: Weight Uncertainty in Neural Network, C. Blundell et al. (2015). This work introduces Bayes by Backprop, which refines and modernises variational inference over neural network weights using the reparameterization trick and backpropagation.
- Paper: Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning, Yarin Gal et al. (2016). This paper connects dropout directly to variational inference in deep neural networks, providing an alternative practical framework for weight uncertainty.
- Paper: Stochastic Backpropagation and Approximate Inference in Deep Generative Models, Danilo Jimenez Rezende et al. (2014). This paper extends stochastic variational methods and backpropagation through stochastic variables to train deep latent Gaussian models.
- Paper: An Introduction to Variational Autoencoders, Diederik P. Kingma et al. (2019). This comprehensive introduction details the broader framework of amortized stochastic variational inference that emerged from stochastic backpropagation in deep neural networks.
- Paper: Variational Inference: A Review for Statisticians, David M. Blei et al. (2016). This review provides a systematic perspective on modern variational inference techniques, connecting classical mean-field algorithms with stochastic scalable methods in machine learning.
