Dense Weak Hiding: Closing Complexity Gaps in Nonconvex and PL Finite-Sum Optimization under Individual Smoothness
Organizations: Institute of Artificial Intelligence and Future Networks, Faculty of Arts and Sciences, Beijing Normal University, Zhuhai 519087, China · Guangdong Key Lab of AI and Multi-Modal Data Processing, BNU-HKBU United International College, Zhuhai 519087, China
Abstract
Under individual smoothness, known algorithms for nonconvex finite-sum optimization use incremental first-order oracle (IFO) calls, while existing lower bounds miss the factor in the second term. We prove the matching lower bound for randomized IFO algorithms, allowing component indices and query points to depend on the complete preceding history. Thus PAGE and SPIDER are minimax optimal up to universal constants under individual and mean-squared smoothness. Under the global Polyak-Lojasiewicz (PL) condition, prior lower bounds did not capture the correct dependence on under individual smoothness, and the regime of small remained open for randomized IFO algorithms. We prove an lower bound for , and an lower bound for . Restarted PAGE attains matching upper bounds under mean-squared smoothness in terms of . Our lower bounds use dense weak hiding. Each hidden direction is spread across all components, making individual queries weakly informative while preserving the direction in the average. A sufficiently large dimension makes premature activation of future stages unlikely within the query budget. Balancing the cost of revealing a stage against the number of stages permitted by individual smoothness yields the missing factor.
Figures & tables
| Nonconvex stationarity | |||
|---|---|---|---|
| Result | Oracle complexity | Algorithm class | Reference |
| Upper | PAGE / SPIDER | Fang et al.,2018 ; Li et al.,2021 | |
| Lower | randomized methods; arbitrary queries | Emmenegger et al.,2022 | |
| Lower | linear-span IFO / PIFO | Zhou and Gu,2019 ; Han et al.,2024 | |
| New [-2pt] lower | randomized IFO algorithms | Theorem 3.1 | |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Class | Defining regularity | Representative sources |
|---|---|---|
| Finite-sum lower bounds and analyses of component gradient updates or stored gradients ( Zhou and Gu, 2019 ; Han et al., 2024 ; Emmenegger et al., 2022 ; Defazio et al., 2014 ; Kavis et al., 2022 ) ; this work. | ||
| PAGE and SPIDER ( Fang et al., 2018 ; Li et al., 2021 ) ; Zhou and Gu (2019, Definition 3.2) ; PL lower bounds in Bai et al. (2024) . | ||
| Lower bounds for stationary points of the full objective ( Carmon et al., 2020 ; Carmon et al., 2021 ) . |
Oracle Complexity
Oracle complexity studies how many queries to limited-information procedures algorithms need to solve optimization, learning, sampling, and related problems.
See all 30 papers