Weibull Weight-Scale Parameter Evolution under AdamW Training Dynamics
Not on the map. Why?
Abstract (arXiv)
Building on a two-parameter Weibull framework for diagnosing transformer weight distributions, we study why the Weibull weight-scale parameter λ grows, overshoots, and then relaxes during AdamW training. We derive a leading-order three-force decomposition of the squared weight norm from the AdamW update: an alignment force measuring the correlation between weights and the adaptive update direction, an injection force from adaptive step magnitude, and a decay force from decoupled weight decay. On self-trained Pythia-70M models with ground-truth optimizer moments, alignment dominates the rise phase, contributing 88-94% of the absolute force budget across four random seeds and remaining robust to super-weight removal. Near saturation, alignment and decay approach balance, explaining the transition from weight-scale growth to relaxation. These force dynamics directly govern the squared-norm component underlying λ(t); the remaining RMS-to-Weibull reconstruction offset is measurable and decomposes into bridge and integration components, totaling approximately 5-6% in densely sampled regions. To extend the analysis to real models where optimizer moments are unavailable, we introduce a spline displacement method that recovers the alignment force from sparse checkpoints with approximately 92-94% accuracy, about twice the naive two-point baseline. We further observe that the peak value of λ(t) varies with training-data coherence in our experiments, suggesting a data-dependent component of weight-scale growth that we leave to a controlled follow-up study. Code and data are available at https://github.com/tiexinding/NPM-Weibull-public.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 8 of its 13 references that are in the map. First posted on arXiv 11 Jun 2026.
Builds on
8 of 13 references.
References · 8
- 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 218
- 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 136
- 2023Rotational Equilibrium: How Weight Decay Balances Learning Across Neural Networkscited by 50
- 2025Weight Decay may matter more than muP for Learning Rate Transfer in Practicecited by 20
- 2024Feature learning as alignment: a structural property of gradient descent in non-linear neural networkscited by 9
- 2025Robust Layerwise Scaling Rules by Proper Weight Decay Tuningcited by 8
- 2024Data Diversity as Implicit Regularization: How Does Diversity Shape the Weight Space of Deep Neural Networks?cited by 4
- 2026A Two-Parameter Weibull Framework for Diagnosing Transformer Weight Distributionscited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.