Select All
16874

SGD Learns One-Layer Networks in WGANs

APA

(2020). SGD Learns One-Layer Networks in WGANs. The Simons Institute for the Theory of Computing. https://simons.berkeley.edu/talks/sgd-learns-one-layer-networks-wgans

Qi Lei (Princeton University)
Talk number16874
Source RepositorySimons Institute

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 min-max optimization problem to global optimality but are in practice successfully trained using stochastic gradient descent-ascent. In this talk, 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.