Search papers, labs, and topics across Lattice.
This paper introduces ST-FS-EB, a function-space empirical Bayes regularization framework for Bayesian deep learning that utilizes Student's t priors in both parameter and function spaces to better capture the heavy-tailed nature of neural network outputs. The authors approximate the posterior distribution using variational inference with Monte Carlo dropout to derive an ELBO objective. Experiments demonstrate that ST-FS-EB achieves robust performance in in-distribution prediction, out-of-distribution detection, and handling distribution shifts compared to baselines using Gaussian priors.
Student's t priors in function-space Bayesian regularization unlock more robust uncertainty estimates and better handle distribution shifts compared to Gaussian priors.
Bayesian deep learning (BDL) has emerged as a principled approach to produce reliable uncertainty estimates by integrating deep neural networks with Bayesian inference, and the selection of informative prior distributions remains a significant challenge. Various function-space variational inference (FSVI) regularisation methods have been presented, assigning meaningful priors over model predictions. However, these methods typically rely on a Gaussian prior, which fails to capture the heavy-tailed statistical characteristics inherent in neural network outputs. By contrast, this work proposes a novel function-space empirical Bayes regularisation framework -- termed ST-FS-EB -- which employs heavy-tailed Student's $t$ priors in both parameter and function spaces. Also, we approximate the posterior distribution through variational inference (VI), inducing an evidence lower bound (ELBO) objective based on Monte Carlo (MC) dropout. Furthermore, the proposed method is evaluated against various VI-based BDL baselines, and the results demonstrate its robust performance in in-distribution prediction, out-of-distribution (OOD) detection and handling distribution shifts.