Timezone: »
We study a variant of a recently introduced min-max optimization framework where the max-player is constrained to update its parameters in a greedy manner until it reaches a first-order stationary point. Our equilibrium definition for this framework depends on a proposal distribution which the min-player uses to choose directions in which to update its parameters. We show that, given a smooth and bounded nonconvex-nonconcave objective function, access to any proposal distribution for the min-player’s updates, and stochastic gradient oracle for the max-player, our algorithm converges to the aforementioned approximate local equilibrium in a number of iterations that does not depend on the dimension. The equilibrium point found by our algorithm depends on the proposal distribution, and when applying our algorithm to train GANs we choose the proposal distribution to be a distribution of stochastic gradients. We empirically evaluate our algorithm on challenging nonconvex-nonconcave test-functions and loss functions arising in GAN training. Our algorithm converges on these test functions and, when used to train GANs, trains stably on synthetic and real-world datasets and avoids mode collapse.
Author Information
Vijay Keswani (Yale University)
Oren Mangoubi (WPI)
Sushant Sachdeva (University of Toronto)
Nisheeth K. Vishnoi (Yale University)
Related Events (a corresponding poster, oral, or spotlight)
-
2022 Poster: A Convergent and Dimension-Independent Min-Max Optimization Algorithm »
Thu. Jul 21st through Fri the 22nd Room Hall E #1215
More from the Same Authors
-
2023 Poster: Subset Selection Based On Multiple Rankings in the Presence of Bias: Effectiveness of Fairness Constraints for Multiwinner Voting Score Functions »
Niclas Boehmer · L. Elisa Celis · Lingxiao Huang · Anay Mehrotra · Nisheeth K. Vishnoi -
2021 Poster: Fair Classification with Noisy Protected Attributes: A Framework with Provable Guarantees »
L. Elisa Celis · Lingxiao Huang · Vijay Keswani · Nisheeth K. Vishnoi -
2021 Spotlight: Fair Classification with Noisy Protected Attributes: A Framework with Provable Guarantees »
L. Elisa Celis · Lingxiao Huang · Vijay Keswani · Nisheeth K. Vishnoi -
2020 Poster: Data preprocessing to mitigate bias: A maximum entropy based approach »
L. Elisa Celis · Vijay Keswani · Nisheeth K. Vishnoi