Composite Online-to-Nonconvex Conversion with Optimal Oracle Complexity
Organizations: The University of Tokyo · The University of Tokyo, The University of Osaka, and RIKEN
Abstract
We consider stochastic nonsmooth nonconvex composite optimization, which includes several important problems such as constrained optimization and the regularized training of neural networks. The objective is the sum of a possibly nonsmooth nonconvex Lipschitz function and a convex regularizer, and the function is accessed through stochastic gradients or function values. The goal is to find a point that satisfies a Goldstein-type stationarity condition designed for composite objectives. To our knowledge, no oracle complexity bound for this setting is known under first-order access, and existing complexities under zeroth-order access are suboptimal. To handle this issue, we employ the framework of online-to-nonconvex conversion, which chooses update directions by an online learner and is known to achieve optimal rates for noncomposite problems. We extend the framework to our composite scenario by introducing new losses for the learner, which contain the regularizer itself rather than its linearization and for which a variant of online mirror descent achieves low regret. We show that the resulting algorithm finds such a point with stochastic gradient queries or function-value queries, where is the Goldstein radius, is the stationarity tolerance, and is the dimension. These rates match the optimal ones for noncomposite nonsmooth nonconvex optimization, demonstrating that the additional convex regularizer does not worsen the oracle complexity. We also give rates for the smooth case and present numerical experiments.
Figures & tables
| Reference | Problem class | Oracle | Complexity |
| Cutkosky et al. (2023) | First-order | ||
| Kornowski and Shamir (2024) | Zeroth-order | ||
| Chen et al. (2025, Theorem 2) | Composite | Zeroth-order (minibatch) | |
| Chen et al. (2025, Theorem 3) | Composite | Zeroth-order (variance reduction) | |
| Pougkakiotis and Kalogerias (2026) | Composite | Zeroth-order | |
| This work ( Theorem 4.4 ) | Composite | First-order |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Reference | Oracle | Access | Assumption on the oracle | Complexity |
|---|---|---|---|---|
| Nesterov (2013) ; Ghadimi et al. (2016) | Deterministic | First-order | – | |
| Ghadimi et al. (2016) | Stochastic | First-order | Bounded variance | |
| Wang et al. (2019) ; Pham et al. (2020) ; Tran-Dinh et al. (2022) | Stochastic | First-order | Bounded variance, mean-squared smoothness | |
| Ghadimi et al. (2016) | Stochastic | Zeroth-order | Each is smooth | |
| This work ( Theorem 5.3 ) | Deterministic | First-order | – | |
| This work ( Theorem 5.3 ) | Stochastic | First-order | Bounded variance |