对比了贝叶斯神经网络中不同推理方法的精度与不确定性量化表现。
Understanding the Trade-offs in Accuracy and Uncertainty Quantification: Architecture and Inference Choices in Bayesian Neural Networks
- 比较了变分推断与马尔可夫链蒙特卡洛在模型复杂度下的表现
- 变分推断在不确定性估计上优于MCMC,且计算成本更低
- 集成变分近似能接近MCMC精度,适合实际部署
随着现代神经网络日益复杂,如何构建兼具高预测性能和可靠不确定性估计的模型成为挑战。尽管贝叶斯神经网络的后验预测分布有理论支持,但常用近似后验的性质常受质疑。计算负担和不可解析的后验导致贝叶斯神经网络出现精度差和不确定性估计不可靠的问题。近似贝叶斯推断通过用更简单可行的分布替代未知的难解后验分布来应对。现代深度模型的高维性和非可识别性使得马尔可夫链蒙特卡洛(MCMC)计算代价极高,难以充分探索多模态后验;而变分推断虽计算效率更高,但缺乏基于采样的渐近保证,易集中于单一模式。两种方法的表现高度依赖架构选择。本文通过分析大宽度模型和样本外数据等场景下的计算成本、准确性和不确定性量化,揭示其权衡关系。研究了多种模型平均与集成技术对后验探索的提升作用。实验表明,变分推断总体上比MCMC提供更好的不确定性量化,且变分近似的集成与堆叠方法在大幅降低计算成本的同时,达到了与MCMC相当的精度。
原文摘要 · Abstract (English)
As modern neural networks get more complex, specifying a model with high predictive performance and sound uncertainty quantification becomes a more challenging task. Despite some promising theoretical results on the true posterior predictive distribution of Bayesian neural networks, the properties of even the most commonly used posterior approximations are often questioned. Computational burdens and intractable posteriors expose miscalibrated Bayesian neural networks to poor accuracy and unreliable uncertainty estimates. Approximate Bayesian inference aims to replace unknown and intractable posterior distributions with some simpler but feasible distributions. The dimensions of modern deep models, coupled with the lack of identifiability, make Markov chain Monte Carlo (MCMC) tremendously expensive and unable to fully explore the multimodal posterior. On the other hand, variational inference benefits from improved computational complexity but lacks the asymptotical guarantees of sampling-based inference and tends to concentrate around a single mode. The performance of both approaches heavily depends on architectural choices; this paper aims to shed some light on this by considering the computational costs, accuracy and uncertainty quantification in different scenarios including large width and out-of-sample data. To improve posterior exploration, different model averaging and ensembling techniques are studied, along with their benefits on predictive performance. In our experiments, variational inference overall provided better uncertainty quantification than MCMC; further, stacking and ensembles of variational approximations provided comparable accuracy to MCMC at a much-reduced cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。