GANs as Gradient Flows that Converge
Not on the map. Why?
Abstract (arXiv)
This paper approaches the unsupervised learning problem by gradient descent in the space of probability density functions. A main result shows that along the gradient flow induced by a distribution-dependent ordinary differential equation (ODE), the unknown data distribution emerges as the long-time limit. That is, one can uncover the data distribution by simulating the distribution-dependent ODE. Intriguingly, the simulation of the ODE is shown equivalent to the training of generative adversarial networks (GANs). This equivalence provides a new "cooperative" view of GANs and, more importantly, sheds new light on the divergence of GANs. In particular, it reveals that the GAN algorithm implicitly minimizes the mean squared error (MSE) between two sets of samples, and this MSE fitting alone can cause GANs to diverge. To construct a solution to the distribution-dependent ODE, we first show that the associated nonlinear Fokker-Planck equation has a unique weak solution, by the Crandall-Liggett theorem for differential equations in Banach spaces. Based on this solution to the Fokker-Planck equation, we construct a unique solution to the ODE, using Trevisan's superposition principle. The convergence of the induced gradient flow to the data distribution is obtained by analyzing the Fokker-Planck equation.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 20 of its 47 references that are in the map. First posted on arXiv 5 May 2022.
Builds on
20 of 47 references.
References · 20
- 2022Generative Adversarial Networkscited by 5,627
- 2016Improved Techniques for Training GANscited by 2,970
- 2017Improved Training of Wasserstein GANscited by 2,235
- 2016Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Networkcited by 1,908
- 2018Spectral Normalization for Generative Adversarial Networkscited by 1,636
- 2016Generative Adversarial Text to Image Synthesiscited by 735
- 2017Towards Principled Methods for Training Generative Adversarial Networkscited by 464
- 2016Generating Videos with Scene Dynamicscited by 376
- 2016NIPS 2016 Tutorial: Generative Adversarial Networkscited by 285
- 2017MMD GAN: Towards Deeper Understanding of Moment Matching Networkcited by 254
- 2015Deep Generative Image Models using a Laplacian Pyramid of Adversarial Networkscited by 214
- 2016Unrolled Generative Adversarial Networkscited by 208
8 more
- 2019Quant GANs: Deep Generation of Financial Time Seriescited by 108
- 2019Mean-Field Langevin Dynamics and Energy Landscape of Neural Networkscited by 50
- 2020Refining Deep Generative Models via Discriminator Gradient Flowcited by 45
- 2019Deep Generative Learning via Variational Gradient Flowcited by 32
- 2019A Characteristic Function Approach to Deep Implicit Generative Modelingcited by 26
- 2020On the Convergence of Gradient Descent in GANs: MMD GAN As a Gradient Flowcited by 21
- 2019Deconstructing Generative Adversarial Networkscited by 15
- 2020Learning Implicit Generative Models with Theoretical Guaranteescited by 1
Built on by
Citing papers per year: 2024: 2; 2025: 3.
5 papers.
- 2025Training Latent Diffusion Models with Interacting Particle Algorithmscited by 4
- 2024Generative Modeling by Minimizing the Wasserstein-2 Losscited by 3
- 2024A Differential Equation Approach for Wasserstein GANs and Beyondcited by 0
- 2025Mean-Field Langevin Diffusions with Density-dependent Temperaturecited by 0
- 2025On optimal solutions of classical and sliced Wasserstein GANs with non-Gaussian datacited by 0
- 2025On optimal solutions of classical and sliced Wasserstein GANs with non-Gaussian data8 Sep 2025
- 2025Mean-Field Langevin Diffusions with Density-dependent Temperature28 Jul 2025
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2024Generative Modeling by Minimizing the Wasserstein-2 Loss19 Jun 2024
- 2024A Differential Equation Approach for Wasserstein GANs and Beyond25 May 2024
Similar, not cited
Close in wording, no citation either way
- 2026On Interpolation Formulas Describing Neural Network Generalization
- 2024Deep MMD Gradient Flow without adversarial training
- 2024A Survey on Statistical Theory of Deep Learning: Approximation, Training Dynamics, and Generative Models
- 2026Introduction to Stochastic Differential Equations for Generative Machine Learning: A Variational Perspective
- 2022From Points to Functions: Infinite-dimensional Representations in Diffusion Models
- 2023DiffFlow: A Unified SDE Framework for Score-Based Diffusion Models and Generative Adversarial Networks
- 2023Unifying GANs and Score-Based Diffusion as Generative Particle Models
- 2023SMaRt: Improving GANs with Score Matching Regularity