Functional Bayesian Neural Networks for Model Uncertainty Quantification

Nanyang Ye, Zhanxing Zhu

Abstract: In this paper, we extend the Bayesian neural network to functional Bayesian neural network with functional Monte Carlo methods that use the samples of functionals instead of samples of networks' parameters for inference to overcome the curse of dimensionality for uncertainty quantification. Based on the previous work on Riemannian Langevin dynamics, we propose the stochastic gradient functional Riemannian dynamics for training functional Bayesian neural network. We show the effectiveness and efficiency of our proposed approach with various experiments.

Data: [MNIST](https://paperswithcode.com/dataset/mnist)