arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.02437stat.COstat.ME

利用稀疏性提升分层贝叶斯模型的No-U-Turn采样效率

Leveraging Sparsity to Improve No-U-Turn Sampling Efficiency for Hierarchical Bayesian Models

Cole C. Monnahan, Kasper Kristensen, James T. Thorson, Bob Carpenter

AI总结:

SNUTS通过稀疏精度矩阵提升分层贝叶斯模型的采样效率,适用于高维、稀疏和高相关性场景。

摘要:

分析人员经常使用贝叶斯分层模型来理解自然过程。No-U-Turn采样器(NUTS)是用于采样高维连续可微模型最广泛使用的算法。但NUTS在高维情况下受到高相关性的限制,限制了应用分析的复杂性。本文引入了稀疏NUTS(SNUTS),通过使用稀疏精度矩阵(Q)对后验进行预条件(去相关和去缩放)。我们使用模板模型构建器(TMB)高效地从拉普拉斯近似边际后验的模式计算Q,然后通过贝叶斯软件Stan将预处理后的后验传递给NUTS进行采样。我们对十七个不同的案例研究应用SNUTS,证明了使用Q进行预处理比Stan的行业标准对角或密集预处理器快一个到两个数量级。SNUTS在使用Pathfinder变分推断估计协方差的逆进行预处理时也表现更优。SNUTS在曲率高度变化的模型中(如funnels、宽尾或多个模式)不提高采样效率。SNUTS在高维性、稀疏性和高相关性存在的情况下最有效,这些在应用统计学中都很普遍。SNUTS的开源实现提供在R包SparseNUTS中。

英文摘要:

Analysts routinely use Bayesian hierarchical models to understand natural processes. The no-U-turn sampler (NUTS) is the most widely used algorithm to sample high-dimensional, continuously differentiable models. But NUTS is slowed by high correlations, especially in high dimensions, limiting the complexity of applied analyses. Here we introduce Sparse NUTS (SNUTS), which preconditions (decorrelates and descales) posteriors using a sparse precision matrix ($Q$). We use Template Model Builder (TMB) to efficiently compute $Q$ from the mode of the Laplace approximation to the marginal posterior, then pass the preconditioned posterior to NUTS through the Bayesian software Stan for sampling. We apply SNUTS to seventeen diverse case studies to demonstrate that preconditioning with $Q$ converges one to two orders of magnitude faster than Stan's industry standard diagonal or dense preconditioners. SNUTS also outperforms preconditioning with the inverse of the covariance estimated with Pathfinder variational inference. SNUTS does not improve sampling efficiency for models with the highly varying curvature found in funnels, wide tails, or multiple modes. SNUTS is most advantageous, and can be scaled beyond $10^4$ parameters, in the presence of high dimensionality, sparseness, and high correlations, all of which are widespread in applied statistics. An open-source implementation of SNUTS is provided in the R package SparseNUTS.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑