arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.14532cs.LGstat.ML

一种轻量级贝叶斯神经网络异方差不确定性的变分推断框架

A Framework for Variational Inference of Lightweight Bayesian Neural Networks with Heteroscedastic Uncertainties

Teledyne Scientific & Imaging(泰勒-扬科学与成像)

David J. Schodt, Ryan Brown, Michael Merritt, Samuel Park, Delsin Menolascino, Mark A. Peot

AI总结:

本文提出一种轻量级贝叶斯神经网络异方差不确定性变分推断框架,通过将异方差aleatoric和epistemic方差嵌入到学习参数的方差中,提升预测性能。

摘要:

从贝叶斯神经网络(BNN)中获得异方差预测不确定性对于许多应用至关重要。通常,异方差aleatoric不确定性作为BNN的输出与预测均值一同学习,但这样做可能需要向网络添加更多可学习参数。在本工作中,我们证明了异方差aleatoric和epistemic方差可以嵌入到学习BNN参数的方差中,从而提高轻量级网络的预测性能。通过将此方法与动量传播方法相结合,我们引入了一种相对简单的采样-free变分推断框架,适用于轻量级BNNs。

英文摘要:

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 in addition to the predictive means, however doing so may necessitate adding more learnable parameters to the network. In this work, we demonstrate that both the heteroscedastic aleatoric and epistemic variance can be embedded into the variances of learned BNN parameters, improving predictive performance for lightweight networks. By complementing this approach with a moment propagation approach to inference, we introduce a relatively simple framework for sampling-free variational inference suitable for lightweight BNNs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑