Schedule-Free Optimization (Schedule-Free SGD and Schedule-Free AdamW)
Algorithms AIMech:ScheduleFreeOptimizationViaIterateAveragingAndMomentumInterpolation DRAFT
Aliases: Schedule-Free Optimization, Schedule-Free SGD, Schedule-Free AdamW, The Road Less Scheduled Optimizer
A family of anytime stochastic optimization algorithms (Schedule-Free SGD, Equations 3-5, and Schedule-Free AdamW, Algorithm 1) and an online-to-batch conversion framework (Theorem 2 and Theorem 5) that eliminate preset learning rate decay schedules and knowledge of the stopping horizon T without adding extra hyperparameters or memory buffers over standard momentum optimizers. At step t, Schedule-Free optimization forms the gradient evaluation location y_t = (1 - beta) z_t + beta x_t as a convex interpolation between the base optimizer sequence z_t and the online weighted evaluation average x_t, evaluates the stochastic subgradient g_t in partial f(y_t, zeta_t) at y_t, updates the base sequence z_{t+1} (e.g., z_{t+1} = z_t - gamma g_t for SGD or z_{t+1} = z_t - gamma_t g_t / (sqrt{v_t} + epsilon) - gamma_t lambda y_t for AdamW), and updates the evaluation iterate via x_{t+1} = (1 - c_{t+1}) x_t + c_{t+1} z_{t+1} where c_{t+1} = w_{t+1} / sum_{i=1}^{t+1} w_i (such as c_{t+1} = 1/(t+1) or warmup-squared weights c_{t+1} = gamma_t^2 / sum_{i=1}^t gamma_i^2).
Identity Notes & Concept Scope
Distinct from Polyak-Ruppert averaging (beta = 0, which evaluates gradients purely at the fast base sequence y_t = z_t and suffers from instability and poor deep learning performance) and from Primal averaging (beta = 1, which evaluates gradients purely at the slow running average y_t = x_t and converges too slowly in early stages). By setting beta in (0, 1) (typically beta in [0.9, 0.98]), Schedule-Free interpolation immediately injects (1 - beta) Delta_t into y_{t+1} while gradually incorporating the remainder via x_{t+1}, achieving worst-case optimal O(DG/sqrt{T}) non-smooth convex convergence for every beta in [0, 1].
Claims & Grounded Evidence
4 Claims
Mechanism
#unified-bregman-online-to-batch-conversion-and-momentum-optimality
Theorem 2 (Equation 12, Appendix A) and Theorem 5 (Appendix C, Equations 27-30) prove the exact step identity w_{1:t} F(x_t) - w_{1:t-1} F(x_{t-1}) - w_t F(x_*) = w_t langle nabla F(y_t), z_t - x_* rangle - (w_t / alpha_t) B_F(y_t, x_t) - (w_t(1 - alpha_t)/alpha_t) B_F(x_t, y_t) - w_{1:t-1} B_F(x_{t-1}, x_t) - w_t B_F(x_*, y_t) for any convex function F and any sequence beta_t = 1 - alpha_t in [0, 1], establishing E[F(x_T) - F(x_*)] <= E[sum_{t=1}^T w_t langle g_t, z_t - x_* rangle] / sum_{t=1}^T w_t and proving that Schedule-Free SGD achieves the exact minimax optimal rate E[F(x_T) - F(x_*)] <= D G / sqrt{T} for every momentum parameter beta in [0, 1] (Theorem 1).
Mechanism
#exact-unification-of-polyak-averaging-primal-averaging-and-linear-decay-schedules
Section 2.1 and Appendix B (Theorem 4 and Section B.1) show that Theorem 2 recovers classical Polyak-Ruppert online-to-batch conversion at beta_t = 0, Primal Averaging / Heavy-Ball at beta_t = 1, and last-iterate linear-decay scheduling at beta_t = t/T (alpha_t = (T - t)/T with w_t = 1, where the momentum coefficient 1 - alpha_t + t(alpha_{t+1} - alpha_t) vanishes identically so y_{t+1} - y_t = ((T - t)/T)(z_{t+1} - z_t) and y_T = x_T).
Application
#schedule-free-adamw-with-two-buffer-in-place-memory-and-warmup-squared-weighting
Schedule-Free AdamW (Algorithm 1, Section 4.4) eliminates the traditional momentum buffer m_t by storing only z_t, v_t, and the primary parameter buffer (flipping in-place between y_t during training steps and x_t during evaluation passes to match standard AdamW's memory footprint) while weighting the online average by c_{t+1} = gamma_t^2 / sum_{i=1}^t gamma_i^2 where gamma_t = gamma sqrt{1 - beta_2^t} min(1, t / T_warmup) (Equation 23).
Limitation
#batch-normalization-running-statistics-mismatch-between-y-and-x-sequences
Because training forward passes evaluate activations at the interpolated sequence y_t whereas inference and validation evaluate the model at the averaged sequence x_t, networks containing Batch Normalization layers require recomputing BatchNorm running mean and variance statistics on x_t over a small number of training batches prior to evaluation.
⚠️ Curated Limitations & Negative Results
- Requires updating BatchNorm running mean and variance buffers on the evaluation sequence x_t before model evaluation.
- Optimal base learning rate and weight decay values differ from those tuned for cosine or linear decay schedules.
Artificial Intelligence Ontology (AIO) Grounding
Mapped AIO Classes
| AIO Target Class | Mapping Relation | Mapping Rationale |
|---|---|---|
AIO:ArtificialNeuralNetwork
Artificial Neural Network
|
related_match | Schedule-Free SGD and Schedule-Free AdamW are general-purpose optimizers for training deep artificial neural networks without learning rate schedules. |
AIO:LogisticRegression
Logistic Regression
|
related_match | Provides minimax-optimal anytime convergence guarantees and empirical speedups on convex stochastic logistic regression problems. |
Evidence Sources & Primary Literature
3 Sources| Source ID | Publication / Document | Type / Access | Reference & Checksum |
|---|---|---|---|
r585-primary-pdf
|
Aaron Defazio, Xingyu Alice Yang, Harsh Mehta, Konstantin Mishchenko, Ahmed Khaled, Ashok Cutkosky
|
primary paper
full_text
|
ID: https://dblp.org/search/publ/api?q=The+Road+Less+Scheduled+Defazio&format=json
sha256:1aef42351ac4…
|
r585-arxiv-abs
|
Aaron Defazio, Xingyu Alice Yang, Harsh Mehta, Konstantin Mishchenko, Ahmed Khaled, Ashok Cutkosky
|
primary paper
abstract_only
|
ID: https://arxiv.org/abs/2405.15682
sha256:f32c531ae811…
|
r585-metadata-cr
|
Aaron Defazio, Xingyu Alice Yang, Harsh Mehta, Konstantin Mishchenko, Ahmed Khaled, Ashok Cutkosky
|
primary paper
abstract_only
|
ID: https://dblp.org/search/publ/api?q=The+Road+Less+Scheduled+Defazio&format=json
sha256:454943ae8c85…
|
Ontology & Grounding Topology
Open in interactive network graph →Curation Provenance
Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T11:08:00Z
Updated: 2026-10-19T11:08:00Z
Explore across X-Mech
Related computational & biological Mechs:
Embedding Map
Network Graph
ProteinTraitsMech (ESM-2 pLM)
DUFMech (Domain pLM)
TraitMech
CultureMech
All 14 X-Mech Repositories