NeurIPS2020

Stochastic Recursive Gradient Descent Ascent for Stochastic Nonconvex-Strongly-Concave Minimax Problems

Luo Luo, Haishan Ye, Zhichao Huang, Tong Zhang

140 citations

Abstract

We consider nonconvex-concave minimax problems of the form minxmaxyf(x,y)\min_{\bf x}\max_{\bf y} f({\bf x},{\bf y}), where ff is strongly-concave in y\bf y but possibly nonconvex in x\bf x. We focus on the stochastic setting, where we can only access an unbiased stochastic gradient estimate of ff at each iteration. This formulation includes many machine learning applications as special cases such as adversary training and certifying robustness in deep learning. We are interested in finding an O(ε){\mathcal O}(\varepsilon)-stationary point of the function Φ()=maxyf(,y)\Phi(\cdot)=\max_{\bf y} f(\cdot, {\bf y}). The most popular algorithm to solve this problem is stochastic gradient decent ascent, which requires O(κ3ε4)\mathcal O(\kappa^3\varepsilon^{-4}) stochastic gradient evaluations, where κ\kappa is the condition number. In this paper, we propose a novel method called Stochastic Recursive gradiEnt Descent Ascent (SREDA), which estimates gradients more efficiently using variance reduction. This method achieves the best known stochastic gradient complexity of O(κ3ε3){\mathcal O}(\kappa^3\varepsilon^{-3}), and its dependency on ε\varepsilon is optimal for this problem.