Resampling Base Distributions of Normalizing Flows
Not on the map. Why?
Abstract (arXiv)
Normalizing flows are a popular class of models for approximating probability distributions. However, their invertible nature limits their ability to model target distributions whose support have a complex topological structure, such as Boltzmann distributions. Several procedures have been proposed to solve this problem but many of them sacrifice invertibility and, thereby, tractability of the log-likelihood as well as other desirable properties. To address these limitations, we introduce a base distribution for normalizing flows based on learned rejection sampling, allowing the resulting normalizing flow to model complicated distributions without giving up bijectivity. Furthermore, we develop suitable learning algorithms using both maximizing the log-likelihood and the optimization of the Kullback-Leibler divergence, and apply them to various sample problems, i.e. approximating 2D densities, density estimation of tabular data, image generation, and modeling Boltzmann distributions. In these experiments our method is competitive with or outperforms the baselines.
Key facts. Cited by 42 papers in the map (data through 7 Oct 2026). Builds on 37 of its 54 references that are in the map. First posted on arXiv 29 Oct 2021.
Builds on
37 of 54 references.
References · 37
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2019Analyzing and Improving the Image Quality of StyleGANcited by 3,580
- 2017Improved Training of Wasserstein GANscited by 2,223
- 2015Variational Inference with Normalizing Flowscited by 2,161
- 2016Density estimation using Real NVPcited by 2,029
- 2018Glow: Generative Flow with Invertible 1x1 Convolutionscited by 1,539
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
25 more
- 2020Training Generative Adversarial Networks with Limited Datacited by 1,184
- 2021Variational Diffusion Modelscited by 1,145
- 2014NICE: Non-linear Independent Components Estimationcited by 1,052
- 2018Image Transformercited by 770
- 2017Masked Autoregressive Flow for Density Estimationcited by 680
- 2019Neural Spline Flowscited by 590
- 2018Invertible Residual Networkscited by 364
- 2019Flow++: Improving Flow-Based Generative Models with Variational Dequantization and Architecture Designcited by 258
- 2019Residual Flows for Invertible Generative Modelingcited by 256
- 2017The Reversible Residual Network: Backpropagation Without Storing Activationscited by 215
- 2017Parallel WaveNet: Fast High-Fidelity Speech Synthesiscited by 214
- 2020Stochastic Normalizing Flowscited by 156
- 2017PixelSNAIL: An Improved Autoregressive Generative Modelcited by 137
- 2019Semi-Supervised Learning with Normalizing Flowscited by 76
- 2020Understanding and Mitigating Exploding Inverses in Invertible Neural Networkscited by 71
- 2020SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flowscited by 71
- 2020Targeted free energy estimation via learned mappingscited by 66
- 2020Augmented Normalizing Flows: Bridging the Gap Between Generative Flows and Latent Variable Modelscited by 66
- 2018Resampled Priors for Variational Autoencoderscited by 56
- 2021Smooth Normalizing Flowscited by 52
- 2021Densely connected normalizing flowscited by 49
- 2019MaCow: Masked Convolutional Generative Flowcited by 32
- 2020Stabilizing Invertible Neural Networks Using Mixture Modelscited by 30
- 2018Variational Rejection Samplingcited by 13
- 2020Normalizing Flows with Multi-Scale Autoregressive Priorscited by 11
Built on by
Citing papers per year: 2021: 1; 2022: 2; 2023: 16; 2024: 5; 2025: 13; 2026: 5.
42 papers.
Sort:
By area:
- 2022Flow Annealed Importance Sampling Bootstrapcited by 124
- 2023Improved sampling via learned diffusionscited by 103
- 2023normflows: A PyTorch Package for Normalizing Flowscited by 56
- 2025On the Guidance of Flow Matchingcited by 44
- 2023SE(3) Equivariant Augmented Coupling Flowscited by 37
- 2024On the Challenges and Opportunities in Generative AIcited by 31
- 2024Diffusive Gibbs Samplingcited by 21
- 2025Adjoint Schr"odinger Bridge Samplercited by 20
- 2023Squared Neural Families: A New Class of Tractable Density Modelscited by 12
- 2023Designing losses for data-free training of normalizing flows on Boltzmann distributionscited by 9
- 2023Precision-Recall Divergence Optimization for Generative Modeling with GANs and Normalizing Flowscited by 9
- 2025Learning Boltzmann Generators via Constrained Mass Transportcited by 8
30 more
- 2023Training, Architecture, and Prior for Deterministic Uncertainty Methodscited by 7
- 2023Normalizing flow sampling with Langevin dynamics in the latent spacecited by 7
- 2023Topology-Matching Normalizing Flows for Out-of-Distribution Detection in Robot Learningcited by 7
- 2024AdvNF: Reducing Mode Collapse in Conditional Normalising Flows using Adversarial Learningcited by 6
- 2021Bootstrap Your Flowcited by 5
- 2023Optimal Budgeted Rejection Sampling for Generative Modelscited by 5
- 2026Action Hallucination in Generative Vision-Language-Action Modelscited by 5
- 2023Learning Distributions via Monte-Carlo Marginalizationcited by 4
- 2025Evaluating multiple models using labeled and unlabeled datacited by 4
- 2025Learning Geometry and Topology via Multi-Chart Flowscited by 3
- 2022Nonlinear MCMC for Bayesian Machine Learningcited by 2
- 2023Density-based Feasibility Learning with Normalizing Flows for Introspective Robotic Assemblycited by 2
- 2023Reparameterized Variational Rejection Samplingcited by 2
- 2025Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Datacited by 2
- 2025Generative Machine Learning for Multivariate Angular Simulationcited by 2
- 2025Wavelet Flow For Extragalactic Foreground Simulationscited by 2
- 2026Efficient Training of Boltzmann Generators Using Off-Policy Log-Dispersion Regularizationcited by 2
- 2023Piecewise Normalizing Flowscited by 1
- 2023Variations and Relaxations of Normalizing Flowscited by 1
- 2024Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flowscited by 1
- 2025Normalizing Flow Regression for Bayesian Inference with Offline Likelihood Evaluationscited by 1
- 2023Training Normalizing Flows with the Precision-Recall Divergencecited by 0
- 2024Controlling for discrete unmeasured confounding in nonlinear causal modelscited by 0
- 2025Exploring Image Generation via Mutually Exclusive Probability Spaces and Local Correlation Hypothesiscited by 0
- 2025Adaptive Heterogeneous Mixtures of Normalising Flows for Robust Variational Inferencecited by 0
- 2025Score-Based Density Estimation from Pairwise Comparisonscited by 0
- 2025Adapting Noise to Data: Generative Flows from 1D Processescited by 0
- 2026Correcting Source Mismatch in Flow Matching with Radial-Angular Transportcited by 0
- 2026Implicit Variational Rejection Samplingcited by 0
- 2026Simplifying Flow Matching Transformations with Low-Rank Mixture Modelscited by 0
- 2026Simplifying Flow Matching Transformations with Low-Rank Mixture Models29 Jun 2026
- 2026Implicit Variational Rejection Sampling12 Jun 2026
- 2026Correcting Source Mismatch in Flow Matching with Radial-Angular Transport5 Apr 2026
- 2026Action Hallucination in Generative Vision-Language-Action Models6 Feb 2026
- 2026Efficient Training of Boltzmann Generators Using Off-Policy Log-Dispersion Regularization3 Feb 2026
- 2025Learning Boltzmann Generators via Constrained Mass Transport21 Oct 2025
- 2025Adapting Noise to Data: Generative Flows from 1D Processes14 Oct 2025
- 2025Score-Based Density Estimation from Pairwise Comparisons10 Oct 2025
- 2025Adaptive Heterogeneous Mixtures of Normalising Flows for Robust Variational Inference2 Oct 2025
- 2025Adjoint Schr"odinger Bridge Sampler27 Jun 2025
- 2025Exploring Image Generation via Mutually Exclusive Probability Spaces and Local Correlation Hypothesis26 Jun 2025
- 2025Learning Geometry and Topology via Multi-Chart Flows30 May 2025
30 more
- 2025Wavelet Flow For Extragalactic Foreground Simulations27 May 2025
- 2025Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Data29 Apr 2025
- 2025Generative Machine Learning for Multivariate Angular Simulation28 Apr 2025
- 2025Normalizing Flow Regression for Bayesian Inference with Offline Likelihood Evaluations15 Apr 2025
- 2025On the Guidance of Flow Matching4 Feb 2025
- 2025Evaluating multiple models using labeled and unlabeled data21 Jan 2025
- 2024Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flows28 Oct 2024
- 2024Controlling for discrete unmeasured confounding in nonlinear causal models10 Aug 2024
- 2024On the Challenges and Opportunities in Generative AI28 Feb 2024
- 2024Diffusive Gibbs Sampling5 Feb 2024
- 2024AdvNF: Reducing Mode Collapse in Conditional Normalising Flows using Adversarial Learning29 Jan 2024
- 2023Topology-Matching Normalizing Flows for Out-of-Distribution Detection in Robot Learning11 Nov 2023
- 2023Optimal Budgeted Rejection Sampling for Generative Models1 Nov 2023
- 2023Reparameterized Variational Rejection Sampling26 Sep 2023
- 2023Variations and Relaxations of Normalizing Flows8 Sep 2023
- 2023SE(3) Equivariant Augmented Coupling Flows20 Aug 2023
- 2023Learning Distributions via Monte-Carlo Marginalization11 Aug 2023
- 2023Density-based Feasibility Learning with Normalizing Flows for Introspective Robotic Assembly3 Jul 2023
- 2023Improved sampling via learned diffusions3 Jul 2023
- 2023Precision-Recall Divergence Optimization for Generative Modeling with GANs and Normalizing Flows30 May 2023
- 2023Squared Neural Families: A New Class of Tractable Density Models22 May 2023
- 2023Normalizing flow sampling with Langevin dynamics in the latent space20 May 2023
- 2023Piecewise Normalizing Flows4 May 2023
- 2023Training, Architecture, and Prior for Deterministic Uncertainty Methods10 Mar 2023
- 2023Training Normalizing Flows with the Precision-Recall Divergence1 Feb 2023
- 2023normflows: A PyTorch Package for Normalizing Flows26 Jan 2023
- 2023Designing losses for data-free training of normalizing flows on Boltzmann distributions13 Jan 2023
- 2022Flow Annealed Importance Sampling Bootstrap3 Aug 2022
- 2022Nonlinear MCMC for Bayesian Machine Learning11 Feb 2022
- 2021Bootstrap Your Flow22 Nov 2021
- 2025On the Guidance of Flow Matchingcited by 44
- 2025Adjoint Schr"odinger Bridge Samplercited by 20
- 2026Correcting Source Mismatch in Flow Matching with Radial-Angular Transportcited by 0
- 2026Simplifying Flow Matching Transformations with Low-Rank Mixture Modelscited by 0
- 2026Simplifying Flow Matching Transformations with Low-Rank Mixture Models29 Jun 2026
- 2026Correcting Source Mismatch in Flow Matching with Radial-Angular Transport5 Apr 2026
- 2025Adjoint Schr"odinger Bridge Sampler27 Jun 2025
- 2025On the Guidance of Flow Matching4 Feb 2025
- 2022Flow Annealed Importance Sampling Bootstrapcited by 124
- 2023Improved sampling via learned diffusionscited by 103
- 2023normflows: A PyTorch Package for Normalizing Flowscited by 56
- 2023SE(3) Equivariant Augmented Coupling Flowscited by 37
- 2024On the Challenges and Opportunities in Generative AIcited by 31
- 2024Diffusive Gibbs Samplingcited by 21
- 2023Squared Neural Families: A New Class of Tractable Density Modelscited by 12
- 2023Designing losses for data-free training of normalizing flows on Boltzmann distributionscited by 9
- 2023Precision-Recall Divergence Optimization for Generative Modeling with GANs and Normalizing Flowscited by 9
- 2025Learning Boltzmann Generators via Constrained Mass Transportcited by 8
- 2023Training, Architecture, and Prior for Deterministic Uncertainty Methodscited by 7
- 2023Normalizing flow sampling with Langevin dynamics in the latent spacecited by 7
12 most cited of 37 All 37 (JSON)
- 2026Implicit Variational Rejection Sampling12 Jun 2026
- 2026Efficient Training of Boltzmann Generators Using Off-Policy Log-Dispersion Regularization3 Feb 2026
- 2025Learning Boltzmann Generators via Constrained Mass Transport21 Oct 2025
- 2025Adapting Noise to Data: Generative Flows from 1D Processes14 Oct 2025
- 2025Score-Based Density Estimation from Pairwise Comparisons10 Oct 2025
- 2025Adaptive Heterogeneous Mixtures of Normalising Flows for Robust Variational Inference2 Oct 2025
- 2025Exploring Image Generation via Mutually Exclusive Probability Spaces and Local Correlation Hypothesis26 Jun 2025
- 2025Learning Geometry and Topology via Multi-Chart Flows30 May 2025
- 2025Wavelet Flow For Extragalactic Foreground Simulations27 May 2025
- 2025Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Data29 Apr 2025
- 2025Generative Machine Learning for Multivariate Angular Simulation28 Apr 2025
- 2025Normalizing Flow Regression for Bayesian Inference with Offline Likelihood Evaluations15 Apr 2025
12 newest of 37 All 37 (JSON)