Path-Gradient Estimators for Continuous Normalizing Flows
Lorenz Vaitl, Kim Andrea Nicoli, Shinichi Nakajima, Pan Kessel
Abstract
Recent work has established a path-gradient estimator for simple variational Gaussian distributions and has argued that the path-gradient is particularly beneficial in the regime in which the variational distribution approaches the exact target distribution. In many applications, this regime can however not be reached by a simple Gaussian variational distribution. In this work, we overcome this crucial limitation by proposing a path-gradient estimator for the considerably more expressive variational family of continuous normalizing flows. We outline an efficient algorithm to calculate this estimator and establish its superior performance empirically.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext feaa3e83-00bb-4cec-86db-910df2fb3855Cited by top-tier papers4
- Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationDinghuai Zhang, Ricky T. Q. Chen, Cheng-Hao Liu, Aaron C. Courville et al.ICLR 2024 · 64 citations
- Fast and unified path gradient estimators for normalizing flowsLorenz Vaitl, Ludwig Winkler, Lorenz Richter, Pan KesselICLR 2024 · 6 citations
- SESaMo: Symmetry-Enforcing Stochastic Modulation for Normalizing FlowsJanik Kreit, Dominic Schuh, Kim Andrea Nicoli, Lena FunckeICLR 2026 · 3 citations
- Path Gradients after Flow MatchingLorenz Vaitl, Leon KleinNeurIPS 2025 · 3 citations
Builds on4
- Equivariant Flows: Exact Likelihood Generative Learning for Symmetric DensitiesJonas Köhler, Leon Klein, Frank NoéICML 2020 · 330 citations
- Advances in Black-Box VI: Normalizing Flows, Importance Weighting, and OptimizationAbhinav Agrawal, Daniel Sheldon, Justin DomkeNeurIPS 2020 · 49 citations
- On the difficulty of unbiased alpha divergence minimizationTomas Geffner, Justin DomkeICML 2021 · 20 citations
- Generalized Doubly Reparameterized Gradient EstimatorsMatthias Bauer, Andriy MnihICML 2021 · 15 citations
Related papers
- Learning Continuous Normalizing Flows For Faster Convergence To Target Distribution via Ascent RegularizationsShuangshuang Chen, Sihao Ding, Yiannis Karayiannidis, Mårten BjörkmanICLR 2023
- Mutual Information Estimation via Normalizing FlowsIvan Butakov, Aleksander Tolmachev, Sofia Malanchuk, Anna Neopryatnaya et al.NeurIPS 2024 · 30 citations
- Accelerating Continuous Normalizing Flow with Trajectory Polynomial RegularizationHan-Hsien Huang, Mi-Yen YehAAAI 2021 · 8 citations
- Flexible Tails for Normalizing FlowsTennessee Hickling, Dennis PrangleICML 2025
- Neural Posterior Estimation with Latent Basis ExpansionsDeclan McNamara, Yicun Duan, Jeffrey RegierICLR 2026
