Bayesian neural networks via MCMC: a Python-based tutorial

Abstract

Bayesian inference provides a methodology for parameter estimation and uncertainty quantification in machine learning and deep learning methods. Variational inference and Markov Chain Monte-Carlo (MCMC) sampling methods are used to implement Bayesian inference. In the past three decades, MCMC sampling methods have faced some challenges in being adapted to larger models (such as in deep learning) and big data problems. Advanced proposal distributions that incorporate gradients, such as a Langevin proposal distribution, provide a means to address some of the limitations of MCMC sampling for Bayesian neural networks. Furthermore, MCMC methods have typically been constrained to statisticians and currently not well-known among deep learning researchers. We present a tutorial for MCMC methods that covers simple Bayesian linear and logistic models, and Bayesian neural networks. The aim of this tutorial is to bridge the gap between theory and implementation via coding, given a general sparsity of libraries and tutorials to this end. This tutorial provides code in Python with data and instructions that enable their use and extension. We provide results for some benchmark problems showing the strengths and weaknesses of implementing the respective Bayesian models via MCMC. We highlight the challenges in sampling multi-modal posterior distributions for the case of Bayesian neural networks and the need for further improvement of convergence diagnosis methods.

IEEE Access (2024)

References in corpus (11)

• Array Programming with NumPy
• ADADELTA: An Adaptive Learning Rate Method
• Evaluation of deep learning models for multi-step ahead time series prediction
• Deep Learning for Physical Processes: Incorporating Prior Scientific Knowledge
• Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
• A Comprehensive guide to Bayesian Convolutional Neural Network with Variational Inference
• Markov Chain Monte Carlo Method without Detailed Balance
• Sequentially interacting Markov chain Monte Carlo methods
• Bayesian graph convolutional neural networks via tempered MCMC
• Revisiting Bayesian Autoencoders with MCMC
• Post-Processing of MCMC