多速率方法提升贝叶斯采样稳定性与效率
Multirate Stein Variational Gradient Descent for Efficient Bayesian Sampling
- 分时更新吸引与排斥项,适应不同演化速度
- 在六类基准上均优于传统SVGD,尤其在复杂分布中
- 适合高维、各向异性或多重模态的贝叶斯推断任务
许多基于粒子的贝叶斯推断方法对所有更新部分使用单一全局步长。在Stein变分梯度下降(SVGD)中,每次更新结合了两个定性不同的效应:向高后验区域吸引和保持粒子多样性的排斥。这些效应在高维、各向异性或层次化后验中可能以不同速率演化,导致单一步长在某些区域不稳定或效率低下。本文推导出一种多速率版本的SVGD,使这两个组件在不同时间尺度上更新。该框架衍生出实用算法,包括对称分裂法、固定多速率方法(MR-SVGD)和具有局部误差控制的自适应多速率方法(Adapt-MR-SVGD)。我们在涵盖六类问题的广泛严格基准测试中评估这些方法:50维高斯目标、多个二维合成目标、UCI贝叶斯逻辑回归、多重模态高斯混合、贝叶斯神经网络及大规模层次逻辑回归。评估包含后验匹配指标、预测性能、校准质量、混合能力及显式计算成本核算。在六类基准中,多速率SVGD变体在鲁棒性和质量-成本权衡上均优于原始SVGD。最强增益出现在刚性层次化、强各向异性和多重模态目标上,其中自适应多速率SVGD通常表现最优,而固定多速率SVGD则以更低成本提供更稳健的替代方案。
原文摘要 · Abstract (English)
Many particle-based Bayesian inference methods use a single global step size for all parts of the update. In Stein variational gradient descent (SVGD), however, each update combines two qualitatively different effects: attraction toward high-posterior regions and repulsion that preserves particle diversity. These effects can evolve at different rates, especially in high-dimensional, anisotropic, or hierarchical posteriors, so one step size can be unstable in some regions and inefficient in others. We derive a multirate version of SVGD that updates these components on different time scales. The framework yields practical algorithms, including a symmetric split method, a fixed multirate method (MR-SVGD), and an adaptive multirate method (Adapt-MR-SVGD) with local error control. We evaluate the methods in a broad and rigorous benchmark suite covering six problem families: a 50D Gaussian target, multiple 2D synthetic targets, UCI Bayesian logistic regression, multimodal Gaussian mixtures, Bayesian neural networks, and large-scale hierarchical logistic regression. Evaluation includes posterior-matching metrics, predictive performance, calibration quality, mixing, and explicit computational cost accounting. Across these six benchmark families, multirate SVGD variants improve robustness and quality-cost tradeoffs relative to vanilla SVGD. The strongest gains appear on stiff hierarchical, strongly anisotropic, and multimodal targets, where adaptive multirate SVGD is usually the strongest variant and fixed multirate SVGD provides a simpler robust alternative at lower cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。