Dense Weak Hiding: Closing Complexity Gaps in Nonconvex and PL Finite-Sum Optimization under Individual Smoothness
Abstract
Under individual smoothness, the optimal incremental first-order oracle (IFO) complexity of nonconvex finite-sum optimization is still open. Known algorithms use calls, but the best lower bounds miss a factor of on the term. We prove the matching lower bound . Thus PAGE and SPIDER are minimax optimal up to universal constants, under both individual and mean-squared smoothness. Unlike earlier bounds for linear-span algorithms, our result applies to the broader class of randomized IFO algorithms, allowing arbitrary choices of component indices and query points based on the complete preceding history. Under the global Polyak–Lojasiewicz (PL) condition, we use a similar idea to obtain an lower bound for large . We also note that existing PL lower bounds have explored only relatively large values of . To fill this gap, we study the regime and obtain a new lower-bound rate, . The new lower bound motivates us to propose Restarted PAGE. Its upper bound matches the new rate for small and recovers the standard PAGE rate for large , so both lower bounds are nearly tight. Our new lower bounds are based on our proposed dense weak hiding construction. By spreading each hidden direction across all components, the construction makes every single IFO query weakly informative while preserving the direction in the full row average. Because unrevealed stages remain inactive even for arbitrary query points, an algorithm must spend many calls to expose a stage before it can make substantial progress. We choose the bias to balance this revelation cost against the number of stages permitted by individual smoothness, and this balance produces the missing factor.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.