arXiv:2409.16632cs.LG2024-09被引 2

提出函数空间的随机梯度MCMC,解决贝叶斯神经网络后验推断难题

Functional Stochastic Gradient MCMC for Bayesian Neural Networks

  • 设计新型函数空间扩散动力学,支持更优先验建模
  • 理论证明新方法的平稳分布即目标函数后验
  • 在预测精度和不确定性量化上优于参数空间MCMC与变分方法

经典参数空间贝叶斯推断在贝叶斯神经网络中存在知识编码不可行及深层网络异常行为等问题,导致后验推断不恰当。尽管函数空间贝叶斯推断通过函数先验(如功能性变分推断)被提出,但现有随机梯度马尔可夫链蒙特卡洛(MCMC)方法仍局限于参数空间,继承了未解决的先验问题。本文提出新型函数空间MCMC方案,包括其随机梯度版本,基于新设计的扩散动力学,能融合更具信息量的函数先验。我们证明这些函数动力学的平稳测度即为目标函数后验。实验表明,该方法在多个任务上的预测准确率与不确定性量化均优于传统的参数空间MCMC和功能性变分推断。

原文摘要 · Abstract (English)

Classical parameter-space Bayesian inference for Bayesian neural networks (BNNs) suffers from several unresolved prior issues, such as knowledge encoding intractability and pathological behaviours in deep networks, which can lead to improper posterior inference. To address these issues, functional Bayesian inference has recently been proposed leveraging functional priors, such as the emerging functional variational inference. In addition to variational methods, stochastic gradient Markov Chain Monte Carlo (MCMC) is another scalable and effective inference method for BNNs to asymptotically generate samples from the true posterior by simulating continuous dynamics. However, existing MCMC methods perform solely in parameter space and inherit the unresolved prior issues, while extending these dynamics to function space is a non-trivial undertaking. In this paper, we introduce novel functional MCMC schemes, including stochastic gradient versions, based on newly designed diffusion dynamics that can incorporate more informative functional priors. Moreover, we prove that the stationary measure of these functional dynamics is the target posterior over functions. Our functional MCMC schemes demonstrate improved performance in both predictive accuracy and uncertainty quantification on several tasks compared to naive parameter-space MCMC and functional variational inference.

贝叶斯神经网络MCMC函数空间不确定性

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