Optimizing the Preconditioner: A Black-box Online-to-Nonconvex Conversion with Static Regret Minimization Oracles
Not on the map. Why?
Abstract (arXiv)
Stochastic nonconvex optimization is central to training deep networks and LLMs in modern machine learning. We give a black-box reduction from stochastic nonconvex optimization to ordinary static regret minimization in online convex optimization (OCO), thereby resolving the open problem posed by Chen and Hazan (2024). Our reduction maintains a predictable gradient tracker, while a black-box online learner A selects a preconditioner that transforms this tracker into the update direction. Given a (β)-smooth function with a range bounded by M and an unbiased gradient oracle with variance bounded by σ2, we bound the expected average squared gradient norm by O(σ√Mβ/T+√MβRegT(A)/T+fracMβT), where RegT(A) is the static regret of A. Thus, any OCO oracle with O(√T) regret recovers the classical O(T-1/2) convergence rate. We further extend the framework to nonsmooth nonconvex objectives, still relying only on ordinary static regret, and attain the optimal convergence rate for Goldstein-type stationarity. Finally, we conduct numerical experiments on nonconvex objectives to illustrate how the reduction exploits online-selected preconditioners while using the same stochastic-oracle budget as stochastic gradient descent.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 37 of its 59 references that are in the map. First posted on arXiv 20 Jul 2026.
Builds on
37 of 59 references.
References · 37
- 2019Introduction to Online Convex Optimizationcited by 793
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2025Muon is Scalable for LLM Trainingcited by 381
- 2019Lower Bounds for Non-Convex Stochastic Optimizationcited by 303
- 2020A Simple Convergence Proof of Adam and Adagradcited by 169
- 2016NEXT: In-Network Nonconvex Optimizationcited by 107
- 2015Strongly Adaptive Online Learningcited by 102
- 2017On the convergence properties of a K-step averaging stochastic gradient descent algorithm for nonconvex optimizationcited by 91
- 2020Dynamic Regret of Convex and Smooth Functionscited by 90
25 more
- 2013Accelerated Gradient Methods for Nonconvex Nonlinear and Stochastic Programmingcited by 74
- 2013Dynamical Models and Tracking Regret in Online Convex Programmingcited by 42
- 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversioncited by 42
- 2021A gradient sampling method with complexity guarantees for Lipschitz functions in high and low dimensionscited by 37
- 2024Memory-Efficient LLM Training with Online Subspace Descentcited by 37
- 2021On the Finite-Time Complexity and Practical Computation of Approximate Stationarity Concepts of Lipschitz Functionscited by 30
- 2023Deterministic Nonsmooth Nonconvex Optimizationcited by 25
- 2024Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguisecited by 23
- 2024Adam with model exponential moving average is effective for nonconvex optimizationcited by 22
- 2024Random Scaling and Momentum for Non-smooth Non-convex Optimizationcited by 18
- 2024General framework for online-to-nonconvex conversion: Schedule-free SGD is also effective for nonconvex optimizationcited by 14
- 2024Gradient Methods with Online Scalingcited by 11
- 2020Fast and Near-Optimal Diagonal Preconditioningcited by 9
- 2023Searching for Optimal Per-Coordinate Step-sizes with Multidimensional Backtrackingcited by 9
- 2023Scalable Approximate Optimal Diagonal Preconditioningcited by 9
- 2026Adaptive Matrix Online Learning through Smoothing with Guarantees for Nonsmooth Nonconvex Optimizationcited by 9
- 2024Private Zeroth-Order Nonsmooth Nonconvex Optimizationcited by 8
- 2017Stochastic Nonconvex Optimization with Large Minibatchescited by 7
- 2025Provable and Practical Online Learning Rate Adaptation with Hypergradient Descentcited by 5
- 2021Robust Online Control with Model Misspecificationcited by 4
- 2019Surrogate Losses for Online Learning of Stepsizes in Stochastic Non-Convex Optimizationcited by 3
- 2025Improving Online-to-Nonconvex Conversion for Smooth Optimization via Double Optimismcited by 3
- 2026Pion: A Spectrum-Preserving Optimizer via Orthogonal Equivalence Transformationcited by 2
- 2025Gradient Methods with Online Scaling Part II. Practical Aspectscited by 1
- 2026OSDN: Improving Delta Rule with Provable Online Preconditioning in Linear Attentioncited by 1
Built on by
Not cited yet.