Jun, 2024

贝叶斯神经网络中用于明确定义函数空间变分推断的正则化KL散度

TL;DRBayesian神经网络以贝叶斯理念结合了神经网络的预测性能和对安全关键系统和决策制定至关重要的原则性不确定性建模。但是后验不确定性的估计取决于先验的选择,而在权重空间中找到信息量丰富的先验证明非常困难。为了解决这个问题,我们使用了一种基于广义VI的方法结合正则化的KL散度,可以被认为是BNN中具有高斯过程先验的函数空间推断的首个良定义变分目标。实验证明,我们的方法在合成数据和小型现实世界数据集上具备GP先验指定的特性,并与基于函数和权重空间先验的BNN基线相比,在回归、分类和外分布检测方面提供了有竞争力的不确定性估计。