Adaptive Exponential Integration for Stable Gaussian Mixture Black-Box Variational Inference

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Che, Baojun, Chen, Yifan, Huang, Daniel Zhengyu, Mao, Xinying, Wang, Weijie
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914610783715328
author Che, Baojun
Chen, Yifan
Huang, Daniel Zhengyu
Mao, Xinying
Wang, Weijie
author_facet Che, Baojun
Chen, Yifan
Huang, Daniel Zhengyu
Mao, Xinying
Wang, Weijie
contents Black-box variational inference (BBVI) with Gaussian mixture families offers a flexible approach for approximating complex posterior distributions without requiring gradients of the target density. However, standard numerical optimization methods often suffer from instability and inefficiency. We develop a stable and efficient framework that combines three key components: (1) affine-invariant preconditioning via natural gradient formulations, (2) an exponential integrator that unconditionally preserves the positive definiteness of covariance matrices, and (3) adaptive time stepping to ensure stability and to accommodate distinct warm-up and convergence phases. The proposed approach has natural connections to manifold optimization and mirror descent. For Gaussian posteriors, we prove exponential convergence in the noise-free setting and almost-sure convergence under Monte Carlo estimation, rigorously justifying the necessity of adaptive time stepping. Numerical experiments on multimodal distributions, Neal's multiscale funnel, and a PDE-based Bayesian inverse problem for Darcy flow demonstrate the effectiveness of the proposed method.
format Preprint
id arxiv_https___arxiv_org_abs_2601_14855
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Adaptive Exponential Integration for Stable Gaussian Mixture Black-Box Variational Inference
Che, Baojun
Chen, Yifan
Huang, Daniel Zhengyu
Mao, Xinying
Wang, Weijie
Machine Learning
Black-box variational inference (BBVI) with Gaussian mixture families offers a flexible approach for approximating complex posterior distributions without requiring gradients of the target density. However, standard numerical optimization methods often suffer from instability and inefficiency. We develop a stable and efficient framework that combines three key components: (1) affine-invariant preconditioning via natural gradient formulations, (2) an exponential integrator that unconditionally preserves the positive definiteness of covariance matrices, and (3) adaptive time stepping to ensure stability and to accommodate distinct warm-up and convergence phases. The proposed approach has natural connections to manifold optimization and mirror descent. For Gaussian posteriors, we prove exponential convergence in the noise-free setting and almost-sure convergence under Monte Carlo estimation, rigorously justifying the necessity of adaptive time stepping. Numerical experiments on multimodal distributions, Neal's multiscale funnel, and a PDE-based Bayesian inverse problem for Darcy flow demonstrate the effectiveness of the proposed method.
title Adaptive Exponential Integration for Stable Gaussian Mixture Black-Box Variational Inference
topic Machine Learning
url https://arxiv.org/abs/2601.14855