Augmented Normalizing Flows: Bridging the Gap Between Generative Flows and Latent Variable Models
Not on the map. Why?
Abstract (arXiv)
In this work, we propose a new family of generative flows on an augmented data space, with an aim to improve expressivity without drastically increasing the computational cost of sampling and evaluation of a lower bound on the likelihood. Theoretically, we prove the proposed flow can approximate a Hamiltonian ODE as a universal transport map. Empirically, we demonstrate state-of-the-art performance on standard benchmarks of flow-based generative modeling.
Key facts. Cited by 66 papers in the map (data through 14 Oct 2026). First posted on arXiv 17 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 20; 2022: 12; 2023: 17; 2024: 9; 2025: 4; 2026: 4.
66 papers.
Sort:
By area:
- 2021Score-based Generative Modeling in Latent Spacecited by 636
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 252
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusioncited by 233
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 219
- 2023Two for One: Diffusion Models and Force Fields for Coarse-Grained Molecular Dynamicscited by 72
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamicscited by 66
- 2021Densely connected normalizing flowscited by 49
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connectionscited by 47
- 2022CANF-VC: Conditional Augmented Normalizing Flows for Video Compressioncited by 46
- 2023Learning Interpolations between Boltzmann Densitiescited by 43
- 2021Resampling Base Distributions of Normalizing Flowscited by 42
- 2023SE(3) Equivariant Augmented Coupling Flowscited by 37
38 more
- 2022Flow-matching -- efficient coarse-graining of molecular dynamics without forcescited by 36
- 2023Where to Diffuse, How to Diffuse, and How to Get Back: Automated Learning for Multivariate Diffusionscited by 29
- 2021Implicit Normalizing Flowscited by 28
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2023Rigid Body Flows for Sampling Molecular Crystal Structurescited by 27
- 2022Efficient-VDVAE: Less is morecited by 26
- 2023Free-form Flows: Make Any Architecture a Normalizing Flowcited by 26
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamicscited by 23
- 2021ANFIC: Image Compression Using Augmented Normalizing Flowscited by 20
- 2022B-CANF: Adaptive B-frame Coding with Conditional Augmented Normalizing Flowscited by 18
- 2023A Review of Change of Variable Formulas for Generative Modelingcited by 13
- 2021Density estimation on low-dimensional manifolds: an inflation-deflation approachcited by 11
- 2022Semi-Discrete Normalizing Flows through Differentiable Tessellationcited by 11
- 2023Spatial Attention Kinetic Networks with E(n)-Equivariancecited by 11
- 2025Scalable Boltzmann Generators for equilibrium sampling of large-scale materialscited by 11
- 2021Latent Space Refinement for Deep Generative Modelscited by 10
- 2021Automatic variational inference with cascading flowscited by 9
- 2023Designing losses for data-free training of normalizing flows on Boltzmann distributionscited by 9
- 2024Universal Functional Regression with Neural Operator Flowscited by 9
- 2025VFP: Variational Flow-Matching Policy for Multi-Modal Robot Manipulationcited by 9
- 2021Jacobian Determinant of Normalizing Flowscited by 8
- 2021Universal Approximation for Log-concave Distributions using Well-conditioned Normalizing Flowscited by 7
- 2023Normalizing flow sampling with Langevin dynamics in the latent spacecited by 7
- 2022Invertible Monotone Operators for Normalizing Flowscited by 6
- 2022Whitening Convergence Rate of Coupling-based Normalizing Flowscited by 6
- 2024Stable Training of Normalizing Flows for High-dimensional Variational Inferencecited by 6
- 2021Spatial Dependency Networks: Neural Layers for Improved Generative Image Modelingcited by 5
- 2022Flowification: Everything is a Normalizing Flowcited by 5
- 2022Distilling Style from Image Pairs for Global Forward and Inverse Tone Mappingcited by 5
- 2023ELSA -- Enhanced latent spaces for improved collider simulationscited by 5
- 2024Fast and Unified Path Gradient Estimators for Normalizing Flowscited by 5
- 2024Entropy-Informed Weighting Channel Normalizing Flow for Deep Generative Modelscited by 5
- 2021Multi-Resolution Continuous Normalizing Flowscited by 4
- 2021Funnels: Exact maximum likelihood with dimensionality reductioncited by 4
- 2022Deterministic training of generative autoencoders using invertible layerscited by 4
- 2024A Non-negative VAE:the Generalized Gamma Belief Networkcited by 4
- 2025Fine-tuning Flow Matching Generative Models with Intermediate Feedbackcited by 4
- 2024TRADE: Transfer of Distributions between External Conditions with Normalizing Flowscited by 3
50 most cited of 66 All 66 (JSON)
- 2026CrossFlow: One-Step Generation Across Latent and Pixel Spaces18 Jun 2026
- 2026Robust Conditional Conformal Prediction via Branched Normalizing Flow3 May 2026
- 2026Noise Titration: Exact Distributional Benchmarking for Probabilistic Time Series Forecasting23 Mar 2026
- 2026Generalizing Score-based generative models for Heavy-tailed Distributions28 Feb 2026
- 2025Fine-tuning Flow Matching Generative Models with Intermediate Feedback20 Oct 2025
- 2025Scalable Boltzmann Generators for equilibrium sampling of large-scale materials29 Sep 2025
- 2025VFP: Variational Flow-Matching Policy for Multi-Modal Robot Manipulation3 Aug 2025
- 2025Ellipsoidal Time Series Forecasting23 May 2025
- 2024TRADE: Transfer of Distributions between External Conditions with Normalizing Flows25 Oct 2024
- 2024Approximately Invertible Neural Network for Learned Image Compression30 Aug 2024
- 2024A Non-negative VAE:the Generalized Gamma Belief Network6 Aug 2024
- 2024Entropy-Informed Weighting Channel Normalizing Flow for Deep Generative Models6 Jul 2024
38 more
- 2024Universal Functional Regression with Neural Operator Flows3 Apr 2024
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connections3 Apr 2024
- 2024Fast and Unified Path Gradient Estimators for Normalizing Flows23 Mar 2024
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applications1 Mar 2024
- 2024Stable Training of Normalizing Flows for High-dimensional Variational Inference26 Feb 2024
- 2023Free-form Flows: Make Any Architecture a Normalizing Flow25 Oct 2023
- 2023Variations and Relaxations of Normalizing Flows8 Sep 2023
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamics3 Sep 2023
- 2023Mixed Variational Flows for Discrete Variables29 Aug 2023
- 2023SE(3) Equivariant Augmented Coupling Flows20 Aug 2023
- 2023A Review of Change of Variable Formulas for Generative Modeling4 Aug 2023
- 2023Normalizing flow sampling with Langevin dynamics in the latent space20 May 2023
- 2023ELSA -- Enhanced latent spaces for improved collider simulations12 May 2023
- 2023Multiscale Augmented Normalizing Flows for Image Compression9 May 2023
- 2023Invertible Convolution with Symmetric Paddings30 Mar 2023
- 2023Where to Diffuse, How to Diffuse, and How to Get Back: Automated Learning for Multivariate Diffusions14 Feb 2023
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamics2 Feb 2023
- 2023Two for One: Diffusion Models and Force Fields for Coarse-Grained Molecular Dynamics1 Feb 2023
- 2023Rigid Body Flows for Sampling Molecular Crystal Structures26 Jan 2023
- 2023Spatial Attention Kinetic Networks with E(n)-Equivariance21 Jan 2023
- 2023Learning Interpolations between Boltzmann Densities18 Jan 2023
- 2023Designing losses for data-free training of normalizing flows on Boltzmann distributions13 Jan 2023
- 2022Whitening Convergence Rate of Coupling-based Normalizing Flows25 Oct 2022
- 2022Invertible Monotone Operators for Normalizing Flows15 Oct 2022
- 2022Distilling Style from Image Pairs for Global Forward and Inverse Tone Mapping30 Sep 2022
- 2022B-CANF: Adaptive B-frame Coding with Conditional Augmented Normalizing Flows5 Sep 2022
- 2022CANF-VC: Conditional Augmented Normalizing Flows for Video Compression12 Jul 2022
- 2022Flowification: Everything is a Normalizing Flow30 May 2022
- 2022Deterministic training of generative autoencoders using invertible layers19 May 2022
- 2022Discretely Indexed Flows4 Apr 2022
- 2022Efficient-VDVAE: Less is more25 Mar 2022
- 2022Flow-matching -- efficient coarse-graining of molecular dynamics without forces21 Mar 2022
- 2022Semi-Discrete Normalizing Flows through Differentiable Tessellation14 Mar 2022
- 2022Generative Modeling for Low Dimensional Speech Attributes with Neural Spline Flows3 Mar 2022
- 2021Funnels: Exact maximum likelihood with dimensionality reduction15 Dec 2021
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusion14 Dec 2021
- 2021ELF: Exact-Lipschitz Based Universal Density Approximator Flow13 Dec 2021
- 2021Characteristic Neural Ordinary Differential Equations25 Nov 2021
50 newest of 66 All 66 (JSON)
- 2021Score-based Generative Modeling in Latent Spacecited by 636
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusioncited by 233
- 2023Two for One: Diffusion Models and Force Fields for Coarse-Grained Molecular Dynamicscited by 72
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connectionscited by 47
- 2022Flow-matching -- efficient coarse-graining of molecular dynamics without forcescited by 36
- 2023Where to Diffuse, How to Diffuse, and How to Get Back: Automated Learning for Multivariate Diffusionscited by 29
- 2025VFP: Variational Flow-Matching Policy for Multi-Modal Robot Manipulationcited by 9
- 2025Fine-tuning Flow Matching Generative Models with Intermediate Feedbackcited by 4
- 2026Generalizing Score-based generative models for Heavy-tailed Distributionscited by 3
- 2026CrossFlow: One-Step Generation Across Latent and Pixel Spacescited by 2
- 2026CrossFlow: One-Step Generation Across Latent and Pixel Spaces18 Jun 2026
- 2026Generalizing Score-based generative models for Heavy-tailed Distributions28 Feb 2026
- 2025Fine-tuning Flow Matching Generative Models with Intermediate Feedback20 Oct 2025
- 2025VFP: Variational Flow-Matching Policy for Multi-Modal Robot Manipulation3 Aug 2025
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connections3 Apr 2024
- 2023Where to Diffuse, How to Diffuse, and How to Get Back: Automated Learning for Multivariate Diffusions14 Feb 2023
- 2023Two for One: Diffusion Models and Force Fields for Coarse-Grained Molecular Dynamics1 Feb 2023
- 2022Flow-matching -- efficient coarse-graining of molecular dynamics without forces21 Mar 2022
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusion14 Dec 2021
- 2021Score-based Generative Modeling in Latent Space10 Jun 2021
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 252
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 219
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamicscited by 66
- 2021Densely connected normalizing flowscited by 49
- 2022CANF-VC: Conditional Augmented Normalizing Flows for Video Compressioncited by 46
- 2023Learning Interpolations between Boltzmann Densitiescited by 43
- 2021Resampling Base Distributions of Normalizing Flowscited by 42
- 2023SE(3) Equivariant Augmented Coupling Flowscited by 37
- 2021Implicit Normalizing Flowscited by 28
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2023Rigid Body Flows for Sampling Molecular Crystal Structurescited by 27
- 2022Efficient-VDVAE: Less is morecited by 26
12 most cited of 56 All 56 (JSON)
- 2026Robust Conditional Conformal Prediction via Branched Normalizing Flow3 May 2026
- 2026Noise Titration: Exact Distributional Benchmarking for Probabilistic Time Series Forecasting23 Mar 2026
- 2025Scalable Boltzmann Generators for equilibrium sampling of large-scale materials29 Sep 2025
- 2025Ellipsoidal Time Series Forecasting23 May 2025
- 2024TRADE: Transfer of Distributions between External Conditions with Normalizing Flows25 Oct 2024
- 2024Approximately Invertible Neural Network for Learned Image Compression30 Aug 2024
- 2024A Non-negative VAE:the Generalized Gamma Belief Network6 Aug 2024
- 2024Entropy-Informed Weighting Channel Normalizing Flow for Deep Generative Models6 Jul 2024
- 2024Universal Functional Regression with Neural Operator Flows3 Apr 2024
- 2024Fast and Unified Path Gradient Estimators for Normalizing Flows23 Mar 2024
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applications1 Mar 2024
- 2024Stable Training of Normalizing Flows for High-dimensional Variational Inference26 Feb 2024
12 newest of 56 All 56 (JSON)