mlsanityData to 7 Oct 2026

Optimizing the Preconditioner: A Black-box Online-to-Nonconvex Conversion with Static Regret Minimization Oracles

arXiv 2607.17607 ↗First posted 20 Jul 2026cs.LG math.OC stat.MLcited by 0

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

  1. 2019Introduction to Online Convex Optimizationcited by 793
  2. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
  3. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  4. 2023Symbolic Discovery of Optimization Algorithmscited by 423
  5. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
  6. 2025Muon is Scalable for LLM Trainingcited by 381
  7. 2019Lower Bounds for Non-Convex Stochastic Optimizationcited by 303
  8. 2020A Simple Convergence Proof of Adam and Adagradcited by 169
  9. 2016NEXT: In-Network Nonconvex Optimizationcited by 107
  10. 2015Strongly Adaptive Online Learningcited by 102
  11. 2017On the convergence properties of a K-step averaging stochastic gradient descent algorithm for nonconvex optimizationcited by 91
  12. 2020Dynamic Regret of Convex and Smooth Functionscited by 90
25 more
  1. 2013Accelerated Gradient Methods for Nonconvex Nonlinear and Stochastic Programmingcited by 74
  2. 2013Dynamical Models and Tracking Regret in Online Convex Programmingcited by 42
  3. 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversioncited by 42
  4. 2021A gradient sampling method with complexity guarantees for Lipschitz functions in high and low dimensionscited by 37
  5. 2024Memory-Efficient LLM Training with Online Subspace Descentcited by 37
  6. 2021On the Finite-Time Complexity and Practical Computation of Approximate Stationarity Concepts of Lipschitz Functionscited by 30
  7. 2023Deterministic Nonsmooth Nonconvex Optimizationcited by 25
  8. 2024Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguisecited by 23
  9. 2024Adam with model exponential moving average is effective for nonconvex optimizationcited by 22
  10. 2024Random Scaling and Momentum for Non-smooth Non-convex Optimizationcited by 18
  11. 2024General framework for online-to-nonconvex conversion: Schedule-free SGD is also effective for nonconvex optimizationcited by 14
  12. 2024Gradient Methods with Online Scalingcited by 11
  13. 2020Fast and Near-Optimal Diagonal Preconditioningcited by 9
  14. 2023Searching for Optimal Per-Coordinate Step-sizes with Multidimensional Backtrackingcited by 9
  15. 2023Scalable Approximate Optimal Diagonal Preconditioningcited by 9
  16. 2026Adaptive Matrix Online Learning through Smoothing with Guarantees for Nonsmooth Nonconvex Optimizationcited by 9
  17. 2024Private Zeroth-Order Nonsmooth Nonconvex Optimizationcited by 8
  18. 2017Stochastic Nonconvex Optimization with Large Minibatchescited by 7
  19. 2025Provable and Practical Online Learning Rate Adaptation with Hypergradient Descentcited by 5
  20. 2021Robust Online Control with Model Misspecificationcited by 4
  21. 2019Surrogate Losses for Online Learning of Stepsizes in Stochastic Non-Convex Optimizationcited by 3
  22. 2025Improving Online-to-Nonconvex Conversion for Smooth Optimization via Double Optimismcited by 3
  23. 2026Pion: A Spectrum-Preserving Optimizer via Orthogonal Equivalence Transformationcited by 2
  24. 2025Gradient Methods with Online Scaling Part II. Practical Aspectscited by 1
  25. 2026OSDN: Improving Delta Rule with Provable Online Preconditioning in Linear Attentioncited by 1

Built on by

Not cited yet.