A Framework for Variational Inference of Lightweight Bayesian Neural Networks with Heteroscedastic Uncertainties
arXiv:2402.14532v2 Announce Type: replace
Abstract: Obtaining heteroscedastic predictive uncertainties from a Bayesian Neural Network (BNN) is vital to many applications. Often, heteroscedastic aleatoric uncertainties are learned as outputs of the BNN…