Multivariate Bayesian Last Layer for Regression with Uncertainty Quantification and Decomposition
Machine Learning
2026-02-02 v2 Machine Learning
Abstract
We present new Bayesian Last Layer neural network models in the setting of multivariate regression under heteroscedastic noise, and propose EM algorithms for parameter learning. Bayesian modeling of a neural network's final layer has the attractive property of uncertainty quantification with a single forward pass. The proposed framework is capable of disentangling the aleatoric and epistemic uncertainty, and can be used to enhance a canonically trained deep neural network with uncertainty-aware capabilities.
Cite
@article{arxiv.2405.01761,
title = {Multivariate Bayesian Last Layer for Regression with Uncertainty Quantification and Decomposition},
author = {Han Wang and Eiji Kawasaki and Guillaume Damblin and Geoffrey Daniel},
journal= {arXiv preprint arXiv:2405.01761},
year = {2026}
}