Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimization
Not on the map. Why?
Abstract (arXiv)
This paper investigates new families of compositional optimization problems, called underlinebf non-underlinebf smooth underlinebf weakly-underlinebf convex underlinebf finite-sum underlinebf coupled underlinebf compositional underlinebf optimization (NSWC FCCO). There has been a growing interest in FCCO due to its wide-ranging applications in machine learning and AI, as well as its ability to address the shortcomings of stochastic algorithms based on empirical risk minimization. However, current research on FCCO presumes that both the inner and outer functions are smooth, limiting their potential to tackle a more diverse set of problems. Our research expands on this area by examining non-smooth weakly-convex FCCO, where the outer function is weakly convex and non-decreasing, and the inner function is weakly-convex. We analyze a single-loop algorithm and establish its complexity for finding an ε-stationary point of the Moreau envelop of the objective function. Additionally, we also extend the algorithm to solving novel non-smooth weakly-convex tri-level finite-sum coupled compositional optimization problems, which feature a nested arrangement of three functions. Lastly, we explore the applications of our algorithms in deep learning for two-way partial AUC maximization and multi-instance two-way partial AUC maximization, using empirical studies to showcase the effectiveness of the proposed algorithms.
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Builds on 37 of its 42 references that are in the map. First posted on arXiv 5 Oct 2023.
Builds on
37 of 42 references.
References · 37
- 2018How Powerful are Graph Neural Networks?cited by 4,393
- 2018Attention-based Deep Multiple Instance Learningcited by 1,095
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,051
- 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 306
- 2019On Gradient Descent Ascent for Nonconvex-Concave Minimax Problemscited by 270
- 2018Stochastic model-based minimization of weakly convex functionscited by 159
- 2022AUC Maximization in the Era of Big Data and AI: A Surveycited by 100
- 2014Stochastic Compositional Gradient Descent: Algorithms for Minimizing Compositions of Expected-Value Functionscited by 95
- 2019Lung and Colon Cancer Histopathological Image Dataset (LC25000)cited by 89
- 2020Large-scale Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image Classificationcited by 86
- 2018Weakly-Convex Concave Min-Max Optimization: Provable Algorithms and Applications in Machine Learningcited by 63
- 2016Efficiency of minimizing compositions of convex functions and smooth mapscited by 59
25 more
- 2018A Single Time-Scale Stochastic Approximation Method for Nested Stochastic Optimizationcited by 53
- 2020Solving Stochastic Compositional Optimization is Nearly as Easy as Solving Stochastic Optimizationcited by 51
- 2017Stochastic Methods for Composite and Weakly Convex Optimization Problemscited by 48
- 2020Biased Stochastic First-Order Methods for Conditional Stochastic Optimization and Applications in Meta Learningcited by 47
- 2016Accelerating Stochastic Composition Optimizationcited by 46
- 2022Provable Stochastic Optimization for Global Contrastive Learning: Small Batch Does Not Harm Performancecited by 43
- 2020An Online Method for A Class of Distributionally Robust Optimization with Non-Convex Objectivescited by 42
- 2017Proximally Guided Stochastic Subgradient Method for Nonsmooth, Nonconvex Problemscited by 30
- 2022When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guaranteecited by 29
- 2019Multi-Level Composite Stochastic Optimization via Nested Variance Reductioncited by 28
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applicationscited by 27
- 2019A Stochastic Composite Gradient Method with Incremental Variance Reductioncited by 26
- 2020Stochastic Multi-level Composition Optimization Algorithms with Level-Independent Convergence Ratescited by 25
- 2018Multi-Level Stochastic Gradient Methods for Nested Composition Optimizationcited by 24
- 2022SAPD+: An Accelerated Stochastic Method for Nonconvex-Concave Minimax Problemscited by 23
- 2022Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergencecited by 22
- 2014Online and Stochastic Gradient Methods for Non-decomposable Loss Functionscited by 21
- 2020Optimal Algorithms for Convex Nested Stochastic Composite Optimizationcited by 20
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimizationcited by 16
- 2020A Primal-Dual Smoothing Framework for Max-Structured Non-Convex Optimizationcited by 14
- 2021Robust Accelerated Primal-Dual Methods for Computing Saddle Pointscited by 10
- 2020Attentional-Biased Stochastic Gradient Descentcited by 9
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 8
- 2023Blockwise Stochastic Variance-Reduced Methods with Parallel Speedup for Multi-Block Bilevel Optimizationcited by 8
- 2023Provable Multi-instance Deep AUC Maximization with Stochastic Poolingcited by 3
Built on by
Citing papers per year: 2023: 1; 2024: 3; 2025: 6.
10 papers.
- 2025Single-loop Algorithms for Stochastic Non-convex Optimization with Weakly-Convex Constraintscited by 6
- 2024Single-Loop Stochastic Algorithms for Difference of Max-Structured Weakly Convex Functionscited by 4
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimizationcited by 3
- 2024Communication-Efficient Federated Group Distributionally Robust Optimizationcited by 3
- 2024Stability and Generalization for Stochastic Recursive Momentum-based Algorithms for (Strongly-)Convex One to K-Level Stochastic Optimizationscited by 2
- 2025Model Steering: Learning with a Reference Model Improves Generalization Bounds and Scaling Lawscited by 1
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximizationcited by 1
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimizationcited by 0
- 2025Stochastic Weakly Convex Optimization Under Heavy-Tailed Noisescited by 0
- 2025DualHash: A Stochastic Primal-Dual Algorithm with Theoretical Guarantee for Deep Hashingcited by 0
- 2025DualHash: A Stochastic Primal-Dual Algorithm with Theoretical Guarantee for Deep Hashing21 Oct 2025
- 2025Stochastic Weakly Convex Optimization Under Heavy-Tailed Noises17 Jul 2025
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimization3 Jun 2025
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximization28 May 2025
- 2025Model Steering: Learning with a Reference Model Improves Generalization Bounds and Scaling Laws10 May 2025
- 2025Single-loop Algorithms for Stochastic Non-convex Optimization with Weakly-Convex Constraints21 Apr 2025
- 2024Communication-Efficient Federated Group Distributionally Robust Optimization8 Oct 2024
- 2024Stability and Generalization for Stochastic Recursive Momentum-based Algorithms for (Strongly-)Convex One to K-Level Stochastic Optimizations7 Jul 2024
- 2024Single-Loop Stochastic Algorithms for Difference of Max-Structured Weakly Convex Functions28 May 2024
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimization4 Dec 2023