arXiv:2607.25376cs.LGcs.AI2026-07

用学生t分布替代高斯分布,显著提升贝叶斯神经网络性能

Rethinking Likelihood distributions: Student's t Likelihood Boosts Bayesian Neural Network Performance

论文配图:Rethinking Likelihood distributions: Student's t Likelihood Boosts Bayesian Neural Network Performance
图 1 · 摘自论文原文
  • 改用学生t分布作为似然函数,更适应数据中的异常值
  • 在多种数据集和网络结构上,预测精度均优于高斯分布
  • 实现简单且部分场景下训练更快,适合追求鲁棒性的研究者

在贝叶斯神经网络中,变分推断是建模不确定性的常用框架,其标准目标函数为证据下界(ELBO)。传统做法通常假设先验、近似后验和似然分布均为高斯分布,因其计算简便、支持重参数化梯度且损失函数闭合。然而,这一假设可能不成立,存在模型误设风险。已有研究针对先验分布提出替代方案,但对似然分布的选择影响尚未探索。本文系统比较了偏斜或重尾分布等替代假设,涵盖人工与真实世界回归任务,使用标准多层感知机(MLP)。结果表明,学生t分布的似然函数在不同数据分布及网络架构(深度与宽度)下均表现更优,且在某些情况下可缩短训练时间,同时保持实现简便。

原文摘要 · Abstract (English)

In Bayesian neural networks (BNNs), variational inference is a widely adopted framework for modeling uncertainty in a distributional way, with the evidence lower bound (ELBO) serving as the standard objective function. Several distributions contribute to the ELBO loss, such as the prior, approximated posterior, and likelihood distribution. Typically, these distributions are all approximated by a Gaussian distribution, since it is easy to compute, allows for reparameterized gradients, and provides a closed-form loss for training. However, several works have highlighted that this assumption may not generally hold, posing the risk of model misspecification. Alternative distributions have been proposed for the prior specifically, while the effect of distribution choice on the likelihood distribution remains unexplored. In this work, our aim is to close this gap by investigating whether alternative assumptions for the likelihood distribution can outperform the commonly used Gaussian. We compare several likelihood distribution assumptions, such as skewed or heavy-tailed, across regression tasks on both artificial and real-world datasets using standard multilayer perceptrons (MLPs). Our findings demonstrate that Student's t yields better predictive performance than a Gaussian likelihood distribution, independent of the data distribution and MLP architecture (depth and width). In some cases, Student's t can also lead to shorter training times, while still being easy to implement.

贝叶斯神经网络不确定性建模学生t分布

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。