用正交函数展开实现高效变分推断,无需迭代优化
EigenVI: score-based variational inference with orthogonal function expansions
- 基于特征值求解,直接构造正交函数展开的变分近似
- 在基准测试中比传统高斯变分推断更准确
- 适合需要稳定、无需调参的复杂分布建模场景
我们提出EigenVI,一种基于特征值的黑箱变分推断方法。该方法通过正交函数展开构建变分近似:在ℝᴰ上,低阶项提供高斯近似,高阶项系统性建模非高斯特性。这些近似灵活且能捕捉多峰、非对称等复杂分布,同时可高效计算低阶矩并采样。通过选择不同正交函数族,EigenVI还可处理非负、有界等变量。其核心是通过最小化目标分布得分函数的随机估计弗里希散度,求得最优近似。关键优势在于该优化转化为最小特征值问题,避免了传统梯度方法依赖学习率、终止条件等超参数的迭代优化。我们在posteriodb中的多个贝叶斯模型上验证,EigenVI在精度上优于现有高斯型黑箱变分推断方法。
原文摘要 · Abstract (English)
We develop EigenVI, an eigenvalue-based approach for black-box variational inference (BBVI). EigenVI constructs its variational approximations from orthogonal function expansions. For distributions over $\mathbb{R}^D$, the lowest order term in these expansions provides a Gaussian variational approximation, while higher-order terms provide a systematic way to model non-Gaussianity. These approximations are flexible enough to model complex distributions (multimodal, asymmetric), but they are simple enough that one can calculate their low-order moments and draw samples from them. EigenVI can also model other types of random variables (e.g., nonnegative, bounded) by constructing variational approximations from different families of orthogonal functions. Within these families, EigenVI computes the variational approximation that best matches the score function of the target distribution by minimizing a stochastic estimate of the Fisher divergence. Notably, this optimization reduces to solving a minimum eigenvalue problem, so that EigenVI effectively sidesteps the iterative gradient-based optimizations that are required for many other BBVI algorithms. (Gradient-based methods can be sensitive to learning rates, termination criteria, and other tunable hyperparameters.) We use EigenVI to approximate a variety of target distributions, including a benchmark suite of Bayesian models from posteriordb. On these distributions, we find that EigenVI is more accurate than existing methods for Gaussian BBVI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。