Monge-Amp`ere Flow for Generative Modeling
Not on the map. Why?
Abstract (arXiv)
We present a deep generative model, named Monge-Amp`ere flow, which builds on continuous-time gradient flow arising from the Monge-Amp`ere equation in optimal transport theory. The generative map from the latent space to the data space follows a dynamical system, where a learnable potential function guides a compressible fluid to flow towards the target density distribution. Training of the model amounts to solving an optimal control problem. The Monge-Amp`ere flow has tractable likelihoods and supports efficient sampling and inference. One can easily impose symmetry constraints in the generative model by designing suitable scalar potential functions. We apply the approach to unsupervised density estimation of the MNIST dataset and variational calculation of the two-dimensional Ising model at the critical point. This approach brings insights and techniques from Monge-Amp`ere equation, optimal transport, and fluid dynamics into reversible flow-based generative models.
Key facts. Cited by 33 papers in the map (data through 7 Oct 2026). First posted on arXiv 26 Sep 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 8; 2023: 4; 2024: 3; 2025: 5; 2026: 5.
33 papers.
Sort:
By area:
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 251
- 2021An Introduction to Deep Generative Modelingcited by 104
- 2024How Discrete and Continuous Diffusion Meet: Comprehensive Analysis of Discrete Diffusion Models via a Stochastic Integral Frameworkcited by 48
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithmscited by 47
- 2024Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time Complexitycited by 42
- 2022Normalizing flow neural networks by JKO schemecited by 39
- 2021Adaptive deep density approximation for Fokker-Planck equationscited by 28
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCDcited by 26
- 2025DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Modelscited by 21
- 2023Efficient Neural Network Approaches for Conditional Optimal Transport with Applications in Bayesian Inferencecited by 19
- 2025Solving Inverse Problems via Diffusion-Based Priors: An Approximation-Free Ensemble Sampling Approachcited by 18
21 more
- 2022Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO Schemecited by 17
- 2023Gaussian Interpolation Flowscited by 17
- 2021Ab-initio study of interacting fermions at finite temperature with neural canonical transformationcited by 16
- 2021Jacobian Determinant of Normalizing Flowscited by 8
- 2022A Mathematical Framework for Learning Probability Distributionscited by 7
- 2022Deep Variational Free Energy Approach to Dense Hydrogencited by 6
- 2022Turning Normalizing Flows into Monge Maps with Geodesic Gaussian Preserving Flowscited by 6
- 2021Augmented KRnet for density estimation and approximationcited by 4
- 2022High-dimensional density estimation with tensorizing flowcited by 4
- 2021EBMs Trained with Maximum Likelihood are Generator Models Trained with a Self-adverserial Losscited by 2
- 2026Simple Approximation and Derivative Free Inference-Time Scaling for Diffusion Models via Sequential Monte Carlo on Path Measurescited by 2
- 2021Characteristic Neural Ordinary Differential Equationscited by 1
- 2023PINF: Continuous Normalizing Flows for Physics-Constrained Deep Learningcited by 1
- 2026FluxLite: Inference-Time Proposal Control for Discrete Diffusion Modelscited by 1
- 2022Multisymplectic Formulation of Deep Learning Using Mean--Field Type Control and Nonlinear Stability of Training Algorithmcited by 0
- 2024Variational Optimization for Quantum Problems using Deep Generative Networkscited by 0
- 2025Accurate and thermodynamically consistent hydrogen equation of state for planetary modeling with flow matchingcited by 0
- 2025Estimation of Stochastic Optimal Transport Mapscited by 0
- 2026Complex normalizing flows can almost be information K"ahler-Ricci flowscited by 0
- 2026Quantum Dynamics via Score Matching on Bohmian Trajectoriescited by 0
- 2026FES-FM: Free Energy Surface Sampling via Reduced Flow Matchingcited by 0
- 2026FluxLite: Inference-Time Proposal Control for Discrete Diffusion Models28 Sep 2026
- 2026Simple Approximation and Derivative Free Inference-Time Scaling for Diffusion Models via Sequential Monte Carlo on Path Measures18 May 2026
- 2026FES-FM: Free Energy Surface Sampling via Reduced Flow Matching1 May 2026
- 2026Quantum Dynamics via Score Matching on Bohmian Trajectories28 Apr 2026
- 2026Complex normalizing flows can almost be information K"ahler-Ricci flows20 Apr 2026
- 2025Estimation of Stochastic Optimal Transport Maps10 Dec 2025
- 2025DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Models25 Sep 2025
- 2025Solving Inverse Problems via Diffusion-Based Priors: An Approximation-Free Ensemble Sampling Approach4 Jun 2025
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithms1 Feb 2025
- 2025Accurate and thermodynamically consistent hydrogen equation of state for planetary modeling with flow matching17 Jan 2025
- 2024How Discrete and Continuous Diffusion Meet: Comprehensive Analysis of Discrete Diffusion Models via a Stochastic Integral Framework4 Oct 2024
- 2024Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time Complexity24 May 2024
21 more
- 2024Variational Optimization for Quantum Problems using Deep Generative Networks28 Apr 2024
- 2023Gaussian Interpolation Flows20 Nov 2023
- 2023Efficient Neural Network Approaches for Conditional Optimal Transport with Applications in Bayesian Inference25 Oct 2023
- 2023PINF: Continuous Normalizing Flows for Physics-Constrained Deep Learning26 Sep 2023
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusions15 Mar 2023
- 2022Normalizing flow neural networks by JKO scheme29 Dec 2022
- 2022A Mathematical Framework for Learning Probability Distributions22 Dec 2022
- 2022High-dimensional density estimation with tensorizing flow1 Dec 2022
- 2022Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO Scheme30 Nov 2022
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCD14 Nov 2022
- 2022Turning Normalizing Flows into Monge Maps with Geodesic Gaussian Preserving Flows22 Sep 2022
- 2022Deep Variational Free Energy Approach to Dense Hydrogen13 Sep 2022
- 2022Multisymplectic Formulation of Deep Learning Using Mean--Field Type Control and Nonlinear Stability of Training Algorithm7 Jul 2022
- 2021Characteristic Neural Ordinary Differential Equations25 Nov 2021
- 2021Augmented KRnet for density estimation and approximation26 May 2021
- 2021Ab-initio study of interacting fermions at finite temperature with neural canonical transformation18 May 2021
- 2021Adaptive deep density approximation for Fokker-Planck equations20 Mar 2021
- 2021An Introduction to Deep Generative Modeling9 Mar 2021
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Models8 Mar 2021
- 2021EBMs Trained with Maximum Likelihood are Generator Models Trained with a Self-adverserial Loss23 Feb 2021
- 2021Jacobian Determinant of Normalizing Flows12 Feb 2021
- 2024How Discrete and Continuous Diffusion Meet: Comprehensive Analysis of Discrete Diffusion Models via a Stochastic Integral Frameworkcited by 48
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithmscited by 47
- 2024Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time Complexitycited by 42
- 2025Solving Inverse Problems via Diffusion-Based Priors: An Approximation-Free Ensemble Sampling Approachcited by 18
- 2026FluxLite: Inference-Time Proposal Control for Discrete Diffusion Modelscited by 1
- 2026FES-FM: Free Energy Surface Sampling via Reduced Flow Matchingcited by 0
- 2026FluxLite: Inference-Time Proposal Control for Discrete Diffusion Models28 Sep 2026
- 2026FES-FM: Free Energy Surface Sampling via Reduced Flow Matching1 May 2026
- 2025Solving Inverse Problems via Diffusion-Based Priors: An Approximation-Free Ensemble Sampling Approach4 Jun 2025
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithms1 Feb 2025
- 2024How Discrete and Continuous Diffusion Meet: Comprehensive Analysis of Discrete Diffusion Models via a Stochastic Integral Framework4 Oct 2024
- 2024Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time Complexity24 May 2024
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 251
- 2021An Introduction to Deep Generative Modelingcited by 104
- 2022Normalizing flow neural networks by JKO schemecited by 39
- 2021Adaptive deep density approximation for Fokker-Planck equationscited by 28
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCDcited by 26
- 2023Efficient Neural Network Approaches for Conditional Optimal Transport with Applications in Bayesian Inferencecited by 19
- 2022Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO Schemecited by 17
- 2023Gaussian Interpolation Flowscited by 17
- 2021Ab-initio study of interacting fermions at finite temperature with neural canonical transformationcited by 16
- 2021Jacobian Determinant of Normalizing Flowscited by 8
- 2022A Mathematical Framework for Learning Probability Distributionscited by 7
12 most cited of 25 All 25 (JSON)
- 2026Quantum Dynamics via Score Matching on Bohmian Trajectories28 Apr 2026
- 2026Complex normalizing flows can almost be information K"ahler-Ricci flows20 Apr 2026
- 2025Estimation of Stochastic Optimal Transport Maps10 Dec 2025
- 2025Accurate and thermodynamically consistent hydrogen equation of state for planetary modeling with flow matching17 Jan 2025
- 2024Variational Optimization for Quantum Problems using Deep Generative Networks28 Apr 2024
- 2023Gaussian Interpolation Flows20 Nov 2023
- 2023Efficient Neural Network Approaches for Conditional Optimal Transport with Applications in Bayesian Inference25 Oct 2023
- 2023PINF: Continuous Normalizing Flows for Physics-Constrained Deep Learning26 Sep 2023
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusions15 Mar 2023
- 2022Normalizing flow neural networks by JKO scheme29 Dec 2022
- 2022A Mathematical Framework for Learning Probability Distributions22 Dec 2022
- 2022High-dimensional density estimation with tensorizing flow1 Dec 2022
12 newest of 25 All 25 (JSON)