On the Convergence of Stochastic Smoothed Multi-Level Compositional Gradient Descent Ascent
Xinwen Zhang, Hongchang Gao
Abstract
Multi-level compositional optimization is a fundamental framework in machine learning with broad applications. While recent advances have addressed compositional minimization problems, the stochastic multi-level compositional minimax problem introduces significant new challenges-most notably, the biased nature of stochastic gradients for both the primal and dual variables. In this work, we address this gap by proposing a novel stochastic multi-level compositional gradient descentascent algorithm, incorporating a smoothing technique under the nonconvex-PL condition. We establish a convergence rate to an (ϵ, ϵ/ √ κ)-stationary point with improved dependence on the condition number at O(κ 3/2 ), where ϵ denotes the solution accuracy and κ represents the condition number. Moreover, we design a novel stage-wise algorithm with variance reduction to address the biased gradient issue under the two-sided PL condition. This algorithm successfully enables a translation from and (ϵ, ϵ/ √ κ)-stationary point to an ϵ-stationary point. Finally, extensive experiments validate the effectiveness of our 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.
Builds on9
- Large-scale Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image ClassificationZhuoning Yuan, Yan Yan, Milan Sonka, Tianbao YangICCV 2021 · 147 citations
- Compositional Training for End-to-End Deep AUC MaximizationZhuoning Yuan, Zhishuai Guo, Nitesh V. Chawla, Tianbao YangICLR 2022 · 34 citations
- Optimal Algorithms for Stochastic Multi-Level Compositional OptimizationWei Jiang, Bokun Wang, Yibo Wang, Lijun Zhang et al.ICML 2022 · 25 citations
- Faster Stochastic Algorithms for Minimax Optimization under Polyak-ojasiewicz ConditionLesi Chen, Boyuan Yao, Luo LuoNeurIPS 2022 · 24 citations
- On the Convergence of Local Stochastic Compositional Gradient Descent with MomentumHongchang Gao, Junyi Li, Heng HuangICML 2022 · 18 citations
Related papers
- LSMC-GDA: A Smoothed Algorithm for Federated Stochastic Multi-Level Compositional Minimax OptimizationXinwen Zhang, Richard Souvenir, Hongchang GaoICML 2026
- Hybrid Variance-Reduced SGD Algorithms For Minimax Problems with Nonconvex-Linear FunctionQuoc Tran-Dinh, Deyi Liu, Lam M. NguyenNeurIPS 2020 · 28 citations
- A Doubly Recursive Stochastic Compositional Gradient Descent Method for Federated Multi-Level Compositional OptimizationHongchang GaoICML 2024 · 1 citation
- Fast Training Method for Stochastic Compositional Optimization ProblemsHongchang Gao, Heng HuangNeurIPS 2021 · 17 citations
- A Projection-free Algorithm for Constrained Stochastic Multi-level Composition OptimizationTesi Xiao, Krishnakumar Balasubramanian, Saeed GhadimiNeurIPS 2022 · 9 citations
