Oct, 2024

ELBOing Stein:基于Stein混合推断的变分贝叶斯方法

TL;DR本文解决了Stein变分梯度下降(SVGD)中存在的方差崩溃问题,这一问题导致在小型贝叶斯神经网络(BNNs)中预测不佳。我们提出了一种新的方法Stein混合推断(SMI),通过让每个粒子参数化混合模型中的组件分布,并优化证据的下界(ELBO),显著提高了不确定性估计的准确性,尤其是在数据维度高且稀疏的情况下。