AIMech AI

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

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 →
CONCEPT KIND Algorithms AIMech:ScheduleFreeOptimizationViaIterateAveragingAndMomentumInterpolation Schedule-Free... 4 claims · 3 sources AIO GROUNDING AIO:ArtificialNeuralNetwork (2)

Curation Provenance

Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T11:08:00Z
Updated: 2026-10-19T11:08:00Z
Source File: data/curated/ScheduleFreeOptimizationViaIterateAveragingAndMomentumInterpolation.yaml · Schema: curated_concept.yaml

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