Sep, 2023

贝叶斯神经网络中的摊还推断

TL;DR在本文中,我们提出了一种更数据高效的概率元学习方法,通过对贝叶斯神经网络的推理进行逐数据点摊销,引入了摊销伪观测变分推理贝叶斯神经网络(APOVI-BNN)。我们展示了在我们的摊销方案下获得的近似后验与传统变分推理获得的近似后验在质量上是相似或更好的,尽管摊销推理是在单个前向传递中执行的。然后,我们讨论了如何将 APOVI-BNN 看作神经过程家族的新成员,以此作为激励使用神经过程训练目标,以实现在复杂问题上更好的预测性能。最后,我们评估了 APOVI-BNN 在一维回归问题和更复杂的图像完成设置中与其他概率元模型的预测性能,在训练数据有限的情况下,我们的模型在同类模型中表现最好。