Skip to content

When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay

Sep 2026 · 0 citations · 35 references
Computer Science

TL;DR

Together, these results isolate a single governing quantity for scale-invariant optimization, providing a precise and actionable lens on training dynamics, optimizer behavior, and schedule design in modern deep learning.

Abstract

Normalization renders large parts of neural networks effectively scale invariant, inducing a hidden feedback loop in which learning-rate schedules and weight decay interact through the parameter norm to control the effective step taken by the optimizer. We show that this interaction is governed by an exact discrete-time law: a single scalar quantity captures all schedule and decay forcing, while norm growth induces an opposing geometric self-quenching effect. This yields a sharp boundary that cleanly separates contraction- and expansion-dominated effective learning rate regimes. To understand the underlying mechanism, we provide exact analysis of a fully solved normalized regression model where the dynamics reduce to two dimensions and show that the balance point is intrinsically unstable, implying that constant learning rate with weight decay cannot stably maintain an interior equilibrium and instead produces recurrent behavior driven by discrete-time Jacobian structure. We further extend this perspective across optimizers through unified homogeneous-optimizer framework that reveals a structural dichotomy in self-quenching strength, providing a first-principles explanation for why adaptive methods exhibit systematically weaker stabilization under normalization. Across dynamical systems and neural networks (MLP, CNN, GPT2 / MNIST, CIFAR, wikiText, OpenWebText), the predicted law holds with high precision and enables direct control of training via the identified scalar, with performance peaking sharply at the predicted boundary. Together, these results isolate a single governing quantity for scale-invariant optimization, providing a precise and actionable lens on training dynamics, optimizer behavior, and schedule design in modern deep learning. Code is available in https://github.com/shasanamin/normalized-optimization-dynamics.

View source

Similar papers

Conference Sep 2026

Control Parameter Dynamics in Neural Network Optimization: An Analysis of Learning Rate Strategies for Sequential Data Modeling

The learning rate (LR) is the single most influential hyperparameter governing the convergence and stability of stochastic gradient descent (SGD). While numerous heuristic schedules for LR decay and warmup have been proposed to improve generalization, their underlying mathematical behavior as discrete control parameter...

S. Akhatkulov, Islom Yalgoshev, Dildora Rajabova · 0 citations
#machine learning Preprint Sep 2026

From Spectra to Joint Schedules in LLM Pre-training: 3+3(+2) Scaling-Law Regimes

Power-law learning curves are often treated as fixed properties of a model and its data, although learning-rate and batch-size schedules can change the observed loss. We study this dependence in noisy online SGD with linear random features. Conditional on the representation, an exact Volterra equation separates two res...

Yi-Chen Wang, Fang-Hui Liu, Yu-Dong Chen · 0 citations
#machine learning Preprint Sep 2026

SGD in Multiclass Logistic Regression: Sequential Learning and Scaling Laws

Theoretical scaling laws from linear regression to multiclass classification to multiclass classification are extended, while connecting to empirical scaling laws observed in large-scale neural networks.

Konstantinos Christopher Tsiolis, Denny Wu, Christos Thrampoulidis et al. · 0 citations
#machine learning Preprint Oct 2026

Mind the Drift: Diagonal Linear Networks Under Large Learning Rates

Large learning rates can qualitatively change the trajectory of neural network training, often pushing optimization into regimes far from classical gradient-flow behavior. The Edge of Stability (EoS) offers a valuable lens on the dynamics such learning rates induce. We study corresponding dynamics in diagonal linear ne...

Aniket Sanyal, Tom Jacobs, Rebekka Burkholz · 0 citations
#machine learning Preprint Oct 2026

A Solvable Model of Adaptive Learning Rate Rescaling: Acceleration, Stability&Scaling

A recurring design principle in modern optimizers is to decouple update magnitude from the raw gradient norm, yet its consequences for learning-curve and resource scaling remain unclear. We isolate this mechanism by studying normalized SGD in a random-feature model with power-law teacher and data covariance. Fixed-norm...

Itay Lavie, Clarissa Lauditi, Cengiz Pehlevan · 0 citations

Broken scale symmetries in undercomplete linear autoencoders

Neural network loss landscapes have many symmetries, which are preserved by gradient flow but broken by finite-stepsize stochastic gradient descent (SGD). A canonical example of such a symmetry is scale in homogeneous networks: one can scale up the parameters in one layer and down in the next without changing the netwo...

Farhad Pashakhanloo, Jacob A. Zavatone-Veth · 0 citations

Related blog posts

MIT News · Artificial Intelligence Oct 28, 2026

Making machines that can listen

Professor Paris Smaragdis develops computers with superhuman hearing for applications like speech recognition, music understanding, and environmental monitoring.

MIT News · Artificial Intelligence Oct 23, 2026

Q&A: How AI came to model the “messy” human mind

Assistant Professor Benjamin Lindquist studies how engineers deliberately injected randomness into early AI, giving rise to deep-learning models that sometimes hallucinate.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.