Efficient Mirror Descent Ascent Methods for Nonsmooth Minimax Problems
Feihu Huang, Xidong Wu, Heng Huang
Abstract
In the paper, we propose a class of efficient mirror descent ascent methods to solve the nonsmooth nonconvex-strongly-concave minimax problems by using dynamic mirror functions, and introduce a convergence analysis framework to conduct rigorous theoretical analysis for our mirror descent ascent methods. For our stochastic algorithms, we first prove that the mini-batch stochastic mirror descent ascent (SMDA) method obtains a gradient complexity of O(κ 3 -4 ) for finding an -stationary point, where κ denotes the condition number. Further, we propose an accelerated stochastic mirror descent ascent (VR-SMDA) method based on the variance reduced technique. We prove that our VR-SMDA method achieves a lower gradient complexity of O(κ 3 -3 ). For our deterministic algorithm, we prove that our deterministic mirror descent ascent (MDA) achieves a lower gradient complexity of O( √ κ -2 ) under mild conditions, which matches the best known complexity in solving smooth nonconvex-strongly-concave minimax optimization. We conduct the experiments on fair classifier and robust neural network training tasks to demonstrate the efficiency of our new algorithms.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext de478b61-7737-459d-a94b-85affdefc4a1Cited by top-tier papers17
- Faster Adaptive Federated LearningXidong Wu, Feihu Huang, Zhengmian Hu, Heng HuangAAAI 2023 · 99 citations
- A Faster Decentralized Algorithm for Nonconvex Minimax ProblemsWenhan Xian, Feihu Huang, Yanfu Zhang, Heng HuangNeurIPS 2021 · 72 citations
- SAPD+: An Accelerated Stochastic Method for Nonconvex-Concave Minimax ProblemsXuan Zhang, Necdet Serhat Aybat, Mert GürbüzbalabanNeurIPS 2022 · 55 citations
- Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax OptimizationJunchi Yang, Xiang Li, Niao HeNeurIPS 2022 · 29 citations
- Solving a Class of Non-Convex Minimax Optimization in Federated LearningXidong Wu, Jianhui Sun, Zhengmian Hu, Aidong Zhang et al.NeurIPS 2023 · 26 citations
Builds on10
- On Gradient Descent Ascent for Nonconvex-Concave Minimax ProblemsTianyi Lin, Chi Jin, Michael I. JordanICML 2020 · 587 citations
- Distributionally Robust Federated AveragingYuyang Deng, Mohammad Mahdi Kamani, Mehrdad MahdaviNeurIPS 2020 · 176 citations
- Stochastic Recursive Gradient Descent Ascent for Stochastic Nonconvex-Strongly-Concave Minimax ProblemsLuo Luo, Haishan Ye, Zhichao Huang, Tong ZhangNeurIPS 2020 · 152 citations
- Accelerated Algorithms for Smooth Convex-Concave Minimax Problems with O(1/k^2) Rate on Squared Gradient NormTaeho Yoon, Ernest K. RyuICML 2021 · 138 citations
- Global Convergence and Variance Reduction for a Class of Nonconvex-Nonconcave Minimax ProblemsJunchi Yang, Negar Kiyavash, Niao HeNeurIPS 2020 · 136 citations
Related papers
- A Single-Loop Smoothed Gradient Descent-Ascent Algorithm for Nonconvex-Concave Min-Max ProblemsJiawei Zhang, Peijun Xiao, Ruoyu Sun, Zhi-Quan LuoNeurIPS 2020 · 130 citations
- Hybrid Variance-Reduced SGD Algorithms For Minimax Problems with Nonconvex-Linear FunctionQuoc Tran-Dinh, Deyi Liu, Lam M. NguyenNeurIPS 2020 · 28 citations
- Tight Analysis of Extra-gradient and Optimistic Gradient Methods For Nonconvex Minimax ProblemsPouria Mahdavinia, Yuyang Deng, Haochuan Li, Mehrdad MahdaviNeurIPS 2022 · 24 citations
- TiAda: A Time-scale Adaptive Algorithm for Nonconvex Minimax OptimizationXiang Li, Junchi Yang, Niao HeICLR 2023
- Stability and Generalization of Stochastic Gradient Methods for Minimax ProblemsYunwen Lei, Zhenhuan Yang, Tianbao Yang, Yiming YingICML 2021 · 57 citations
