Decentralized Stochastic Gradient Descent Ascent for Finite-Sum Minimax Problems

12/06/2022
by   Hongchang Gao, et al.
0

Minimax optimization problems have attracted significant attention in recent years due to their widespread application in numerous machine learning models. To solve the minimax optimization problem, a wide variety of stochastic optimization methods have been proposed. However, most of them ignore the distributed setting where the training data is distributed on multiple workers. In this paper, we developed a novel decentralized stochastic gradient descent ascent method for the finite-sum minimax optimization problem. In particular, by employing the variance-reduced gradient, our method can achieve O(√(n)κ^3/(1-λ)^2ϵ^2) sample complexity and O(κ^3/(1-λ)^2ϵ^2) communication complexity for the nonconvex-strongly-concave minimax optimization problem. As far as we know, our work is the first one to achieve such theoretical complexities for this kind of problem. At last, we apply our method to optimize the AUC maximization problem and the experimental results confirm the effectiveness of our method.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset