Variance-reduced accelerated methods for decentralized stochastic double-regularized nonconvex strongly-concave minimax problems
Not on the map. Why?
Abstract (arXiv)
In this paper, we consider the decentralized, stochastic nonconvex strongly-concave (NCSC) minimax problem with nonsmooth regularization terms on both primal and dual variables, wherein a network of m computing agents collaborate via peer-to-peer communications. We consider when the coupling function is in expectation or finite-sum form and the double regularizers are convex functions, applied separately to the primal and dual variables. Our algorithmic framework introduces a Lagrangian multiplier to eliminate the consensus constraint on the dual variable. Coupling this with variance-reduction (VR) techniques, our proposed method, entitled VRLM, by a single neighbor communication per iteration, is able to achieve an O(kappa3varepsilon-3) sample complexity under the general stochastic setting, with either a big-batch or small-batch VR option, where kappa is the condition number of the problem and varepsilon is the desired solution accuracy. With a big-batch VR, we can additionally achieve O(kappa2varepsilon-2) communication complexity. Under the special finite-sum setting, our method with a big-batch VR can achieve an O(n + √n kappa2varepsilon-2) sample complexity and O(kappa2varepsilon-2) communication complexity, where n is the number of components in the finite sum. All complexity results match the best-known results achieved by a few existing methods for solving special cases of the problem we consider. To the best of our knowledge, this is the first work which provides convergence guarantees for NCSC minimax problems with general convex nonsmooth regularizers applied to both the primal and dual variables in the decentralized stochastic setting. Numerical experiments are conducted on two machine learning problems. Our code is downloadable from https://github.com/RPI-OPT/VRLM.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 47 of its 64 references that are in the map. First posted on arXiv 14 Jul 2023.
Builds on
47 of 64 references.
References · 47
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2014Striving for Simplicity: The All Convolutional Netcited by 983
- 2017Can Decentralized Algorithms Outperform Centralized Algorithms? A Case Study for Decentralized Parallel Stochastic Gradient Descentcited by 496
- 2019Agnostic Federated Learningcited by 451
- 2014SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectivescited by 401
- 2019Lower Bounds for Non-Convex Stochastic Optimizationcited by 303
- 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 302
- 2018SPIDER: Near-Optimal Non-Convex Optimization via Stochastic Path Integrated Differential Estimatorcited by 283
- 2019On Gradient Descent Ascent for Nonconvex-Concave Minimax Problemscited by 266
- 2017SARAH: A Novel Method for Machine Learning Problems Using Stochastic Recursive Gradientcited by 218
35 more
- 2019A Survey on Distributed Machine Learningcited by 187
- 2014EXTRA: An Exact First-Order Algorithm for Decentralized Consensus Optimizationcited by 160
- 2016Achieving Geometric Convergence for Distributed Optimization over Time-Varying Graphscited by 149
- 2019Solving a Class of Non-Convex Min-Max Games Using Iterative First Order Methodscited by 148
- 2018D2: Decentralized Training over Decentralized Datacited by 140
- 2020Near-Optimal Algorithms for Minimax Optimizationcited by 119
- 2020PAGE: A Simple and Optimal Probabilistic Gradient Estimator for Nonconvex Optimizationcited by 116
- 2016NEXT: In-Network Nonconvex Optimizationcited by 107
- 2019Efficient Algorithms for Smooth Minimax Optimizationcited by 75
- 2020Stochastic Recursive Gradient Descent Ascent for Stochastic Nonconvex-Strongly-Concave Minimax Problemscited by 75
- 2022An Improved Analysis of Gradient Tracking for Decentralized Machine Learningcited by 68
- 2016Error bounds, quadratic growth, and linear convergence of proximal methodscited by 59
- 2020Optimal Complexity in Decentralized Trainingcited by 59
- 2020A Unified Single-loop Alternating Gradient Projection Algorithm for Nonconvex-Concave and Convex-Nonconcave Minimax Problemscited by 56
- 2021Faster Single-loop Algorithms for Minimax Optimization without Strong Concavitycited by 47
- 2019A Decentralized Parallel Algorithm for Training Generative Adversarial Netscited by 43
- 2020Efficient Search of First-Order Nash Equilibria in Nonconvex-Concave Smooth Min-Max Problemscited by 39
- 2020Accelerated Zeroth-Order and First-Order Momentum Methods from Mini to Minimax Optimizationcited by 39
- 2020SQuARM-SGD: Communication-Efficient Momentum SGD for Decentralized Optimizationcited by 31
- 2021ZeroSARAH: Efficient Nonconvex Finite-Sum Optimization with Zero Full Gradient Computationcited by 28
- 2018Distributed Nonconvex Constrained Optimization over Time-Varying Digraphscited by 23
- 2022SAPD+: An Accelerated Stochastic Method for Nonconvex-Concave Minimax Problemscited by 23
- 2021A Hybrid Variance-Reduced Method for Decentralized Stochastic Non-Convex Optimizationcited by 21
- 2020Momentum-based variance-reduced proximal stochastic gradient method for composite nonconvex stochastic optimizationcited by 14
- 2020Variance Reduced EXTRA and DIGing and Their Optimal Acceleration for Strongly Convex Decentralized Optimizationcited by 13
- 2022Nonsmooth Nonconvex-Nonconcave Minimax Optimization: Primal-Dual Balancing and Iteration Complexity Analysiscited by 12
- 2021A Stochastic Proximal Gradient Framework for Decentralized Non-Convex Composite Optimization: Topology-Independent Sample Complexity and Communication Efficiencycited by 11
- 2022Decentralized Stochastic Gradient Descent Ascent for Finite-Sum Minimax Problemscited by 11
- 2022Proximal Stochastic Recursive Momentum Methods for Nonconvex Composite Decentralized Optimizationcited by 10
- 2022An Efficient Stochastic Algorithm for Decentralized Nonconvex-Strongly-Concave Minimax Optimizationcited by 6
- 2023Decentralized gradient descent maximization method for composite nonconvex strongly-concave minimax problemscited by 6
- 2021A Decentralized Primal-Dual Framework for Non-convex Smooth Consensus Optimizationcited by 5
- 2023PRECISION: Decentralized Constrained Min-Max Learning with Low Communication and Sample Complexitiescited by 5
- 2021Accelerated Proximal Alternating Gradient-Descent-Ascent for Nonconvex Minimax Machine Learningcited by 4
- 2022Gradient Tracking: A Unified Approach to Smooth Distributed Optimizationcited by 1
Built on by
Citing papers per year: 2024: 2.
2 papers.