16
7

Decentralized Stochastic Variance Reduced Extragradient Method

Abstract

This paper studies decentralized convex-concave minimax optimization problems of the form minxmaxyf(x,y)1mi=1mfi(x,y)\min_x\max_y f(x,y) \triangleq\frac{1}{m}\sum_{i=1}^m f_i(x,y), where mm is the number of agents and each local function can be written as fi(x,y)=1nj=1nfi,j(x,y)f_i(x,y)=\frac{1}{n}\sum_{j=1}^n f_{i,j}(x,y). We propose a novel decentralized optimization algorithm, called multi-consensus stochastic variance reduced extragradient, which achieves the best known stochastic first-order oracle (SFO) complexity for this problem. Specifically, each agent requires O((n+κn)log(1/ε))\mathcal O((n+\kappa\sqrt{n})\log(1/\varepsilon)) SFO calls for strongly-convex-strongly-concave problem and O((n+nL/ε)log(1/ε))\mathcal O((n+\sqrt{n}L/\varepsilon)\log(1/\varepsilon)) SFO call for general convex-concave problem to achieve ε\varepsilon-accurate solution in expectation, where κ\kappa is the condition number and LL is the smoothness parameter. The numerical experiments show the proposed method performs better than baselines.

View on arXiv
Comments on this paper