Theory and computation for structured variational inference

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sheng, Shunan, Wu, Bohan, Zhu, Bennett, Chewi, Sinho, Pooladian, Aram-Alexandre
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912706419752960
author Sheng, Shunan
Wu, Bohan
Zhu, Bennett
Chewi, Sinho
Pooladian, Aram-Alexandre
author_facet Sheng, Shunan
Wu, Bohan
Zhu, Bennett
Chewi, Sinho
Pooladian, Aram-Alexandre
contents Structured variational inference constitutes a core methodology in modern statistical applications. Unlike mean-field variational inference, the approximate posterior is assumed to have interdependent structure. We consider the natural setting of star-structured variational inference, where a root variable impacts all the other ones. We prove the first results for existence, uniqueness, and self-consistency of the variational approximation. In turn, we derive quantitative approximation error bounds for the variational approximation to the posterior, extending prior work from the mean-field setting to the star-structured setting. We also develop a gradient-based algorithm with provable guarantees for computing the variational approximation using ideas from optimal transport theory. We explore the implications of our results for Gaussian measures and hierarchical Bayesian models, including generalized linear models with location family priors and spike-and-slab priors with one-dimensional debiasing. As a by-product of our analysis, we develop new stability results for star-separable transport maps which might be of independent interest.
format Preprint
id arxiv_https___arxiv_org_abs_2511_09897
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Theory and computation for structured variational inference
Sheng, Shunan
Wu, Bohan
Zhu, Bennett
Chewi, Sinho
Pooladian, Aram-Alexandre
Machine Learning
Structured variational inference constitutes a core methodology in modern statistical applications. Unlike mean-field variational inference, the approximate posterior is assumed to have interdependent structure. We consider the natural setting of star-structured variational inference, where a root variable impacts all the other ones. We prove the first results for existence, uniqueness, and self-consistency of the variational approximation. In turn, we derive quantitative approximation error bounds for the variational approximation to the posterior, extending prior work from the mean-field setting to the star-structured setting. We also develop a gradient-based algorithm with provable guarantees for computing the variational approximation using ideas from optimal transport theory. We explore the implications of our results for Gaussian measures and hierarchical Bayesian models, including generalized linear models with location family priors and spike-and-slab priors with one-dimensional debiasing. As a by-product of our analysis, we develop new stability results for star-separable transport maps which might be of independent interest.
title Theory and computation for structured variational inference
topic Machine Learning
url https://arxiv.org/abs/2511.09897