Asymptotically exact variational inference via measure-preserving dynamical systems

Variational inference (VI) approximates a target distribution with a surrogate distribution within a pre-specified variational family designed to allow \iid sampling and density evaluation. The approximation is obtained by minimizing a divergence to the target, so its max obtainable quality is fundamentally determined by the expressiveness of the family. However, greater flexibility does not guarantee better approximations: the optimization problem is typically highly non-convex, making the theoretical optimum rarely attainable in practice. As a result, VI lacks the asymptotic exactness of Markov chain Monte Carlo methods (MCMC)---the guarantee of arbitrarily accurate inference results given sufficient computation regardless of tuning. This thesis addresses this limitation by introducing mixed variational flows (MixFlows), a framework for constructing practical, asymptotically exact variational families using measure-preserving dynamical systems. We develop both homogeneous MixFlows, derived from ergodic deterministic dynamics, and more general MixFlows, based on measure-preserving stochastic dynamics obtained from involutive MCMC kernels. We establish rigorous guarantees showing that every distribution in a MixFlow family converges in total variation to the target as the flow length increases, while retaining tractable density evaluation and i.i.d. sampling. A range of numerical examples illustrates these theoretical and methodological contributions. Along the way, we develop tools to analyze how probabilistic error propagates through inexact flow dynamics. Such inexactness may arise from floating-point computation or discretization of continuous-time flows. While the error of a single flow map may be negligible, composing many maps can cause severe accumulation. Surprisingly, we show empirically that flow-based inference often remains accurate despite instability. Drawing on shadowing theory from chaotic dynamical systems, we prove that errors in sampling, density evaluation, and evidence lower bound estimation grow far more slowly than trajectory errors themselves, which often grow exponentially with flow length.

Authors

Publication Details

Journal
Open Collections
Published
2026-01-01
DOI
https://doi.org/10.14288/1.0451101
Primary Topic
Markov Chains and Monte Carlo Methods
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Asymptotically exact variational inference via measure-preserving dynamical systems

Zuheng Xu
Open Collections
Markov Chains and Monte Carlo Methods
article

Asymptotically exact variational inference via measure-preserving dynamical systems

Zuheng Xu
article en

Abstract

Variational inference (VI) approximates a target distribution with a surrogate distribution within a pre-specified variational family designed to allow \iid sampling and density evaluation. The approximation is obtained by minimizing a divergence to the target, so its max obtainable quality is fundamentally determined by the expressiveness of the family. However, greater flexibility does not guarantee better approximations: the optimization problem is typically highly non-convex, making the theoretical optimum rarely attainable in practice. As a result, VI lacks the asymptotic exactness of Markov chain Monte Carlo methods (MCMC)---the guarantee of arbitrarily accurate inference results given sufficient computation regardless of tuning. This thesis addresses this limitation by introducing mixed variational flows (MixFlows), a framework for constructing practical, asymptotically exact variational families using measure-preserving dynamical systems. We develop both homogeneous MixFlows, derived from ergodic deterministic dynamics, and more general MixFlows, based on measure-preserving stochastic dynamics obtained from involutive MCMC kernels. We establish rigorous guarantees showing that every distribution in a MixFlow family converges in total variation to the target as the flow length increases, while retaining tractable density evaluation and i.i.d. sampling. A range of numerical examples illustrates these theoretical and methodological contributions. Along the way, we develop tools to analyze how probabilistic error propagates through inexact flow dynamics. Such inexactness may arise from floating-point computation or discretization of continuous-time flows. While the error of a single flow map may be negligible, composing many maps can cause severe accumulation. Surprisingly, we show empirically that flow-based inference often remains accurate despite instability. Drawing on shadowing theory from chaotic dynamical systems, we prove that errors in sampling, density evaluation, and evidence lower bound estimation grow far more slowly than trajectory errors themselves, which often grow exponentially with flow length.

Open Collections
Peace, Justice and strong institutions
Openalex Percentile: Top 93%
Markov Chains and Monte Carlo Methods
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.