Analysis of Schedule-Free Nonconvex Optimization
Not on the map. Why?
Abstract (arXiv)
First-order methods underpin most large-scale learning algorithms, yet their classical convergence guarantees hinge on carefully scheduled step-sizes that depend on the total horizon T, which is rarely known in advance. The Schedule-Free (SF) method promises optimal performance with hyperparameters that are independent of T by interpolating between Polyak--Ruppert averaging and momentum, but nonconvex analysis of SF has been limited or reliant on strong global assumptions. We introduce a robust Lyapunov framework that, under only L-smoothness and lower-boundedness, reduces SF analysis to a single-step descent inequality. This yields horizon-agnostic bounds in the nonconvex setting: O(1/log T) for constant step + PR averaging, O(log T/T) for a linearly growing step-size, and a continuum of O(T-(1-α)) rates for polynomial averaging. We complement these proofs with Performance Estimation Problem (PEP) experiments that numerically validate our rates and suggest that our O(1/log T) bound on the original nonconvex SF algorithm may tighten to O(1/T). Our work extends SF's horizon-free guarantees to smooth nonconvex optimization and charts future directions for optimal nonconvex rates.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 10 of its 21 references that are in the map. First posted on arXiv 8 Aug 2025.
Builds on
10 of 21 references.
References · 10
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
- 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 170
- 2024The Road Less Scheduledcited by 163
- 2020An Improved Analysis of Stochastic Gradient Descent with Momentumcited by 119
- 2014Global convergence of the Heavy-ball method for convex optimizationcited by 61
- 2015Smooth Strongly Convex Interpolation and Exact Worst-case Performance of First-order Methodscited by 37
- 2020Momentum via Primal Averaging: Theoretical Insights and Learning Rate Schedules for Non-Convex Optimizationcited by 21
- 2015Exact Worst-case Performance of First-order Methods for Composite Convex Optimizationcited by 16
- 2024General framework for online-to-nonconvex conversion: Schedule-free SGD is also effective for nonconvex optimizationcited by 15
- 2021First-Order Methods for Convex Optimizationcited by 4
Built on by
Citing papers per year: 2025: 1.
1 paper.
- 2025Taking the Road Less Scheduled with Adaptive Polyak Steps11 Nov 2025