SGD Learns One-Layer Networks in WGANs
Qi Lei, Jason D. Lee, Alex Dimakis, Constantinos Daskalakis
2020Year
36Citations
8Top-tier citations
Abstract
Generative adversarial networks (GANs) are a widely used framework for learning generative models. Wasserstein GANs (WGANs), one of the most successful variants of GANs, require solving a minmax optimization problem to global optimality, but are in practice successfully trained using stochastic gradient descent-ascent. In this paper, we show that, when the generator is a one-layer network, stochastic gradient descent-ascent converges to a global solution with polynomial time and sample complexity.
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.
Cited by top-tier papers8
- Do GANs always have Nash equilibria?Farzan Farnia, Asuman E. OzdaglarICML 2020 · 93 citations
- A mean-field analysis of two-player zero-sum gamesCarles Domingo-Enrich, Samy Jelassi, Arthur Mensch, Grant M. Rotskoff et al.NeurIPS 2020 · 56 citations
- Towards a Better Global Loss Landscape of GANsRuoyu Sun, Tiantian Fang, Alexander G. SchwingNeurIPS 2020 · 39 citations
- Solving Min-Max Optimization with Hidden Structure via Gradient Descent AscentEmmanouil V. Vlatakis-Gkaragkounis, Lampros Flokas, Georgios PiliourasNeurIPS 2021 · 16 citations
- Forward Super-Resolution: How Can GANs Learn Hierarchical Generative Models for Real-World DistributionsZeyuan Allen-Zhu, Yuanzhi LiICLR 2023 · 4 citations
Builds on1
Related papers
- Hidden Convexity of Wasserstein GANs: Interpretable Generative Models with Closed-Form SolutionsArda Sahiner, Tolga Ergen, Batu Ozturkler, Burak Bartan et al.ICLR 2022 · 23 citations
- WGAN with an Infinitely Wide Generator Has No Spurious Stationary PointsAlbert No, Taeho Yoon, Sehyun Kwon, Ernest K. RyuICML 2021 · 2 citations
- Understanding Over-parameterization in Generative Adversarial NetworksYogesh Balaji, Mohammadmahdi Sajedi, Neha Mukund Kalibhat, Mucong Ding et al.ICLR 2021
- On Convergence of Gradient Descent Ascent: A Tight Local AnalysisHaochuan Li, Farzan Farnia, Subhro Das, Ali JadbabaieICML 2022 · 12 citations
- Stability and Generalization of Stochastic Gradient Methods for Minimax ProblemsYunwen Lei, Zhenhuan Yang, Tianbao Yang, Yiming YingICML 2021 · 57 citations
