mlsanityData to 7 Oct 2026

Weibull Weight-Scale Parameter Evolution under AdamW Training Dynamics

arXiv 2606.19367 ↗First posted 11 Jun 2026cs.LG stat.MLcited by 2

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

  1. 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 218
  2. 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 136
  3. 2023Rotational Equilibrium: How Weight Decay Balances Learning Across Neural Networkscited by 50
  4. 2025Weight Decay may matter more than muP for Learning Rate Transfer in Practicecited by 20
  5. 2024Feature learning as alignment: a structural property of gradient descent in non-linear neural networkscited by 9
  6. 2025Robust Layerwise Scaling Rules by Proper Weight Decay Tuningcited by 8
  7. 2024Data Diversity as Implicit Regularization: How Does Diversity Shape the Weight Space of Deep Neural Networks?cited by 4
  8. 2026A Two-Parameter Weibull Framework for Diagnosing Transformer Weight Distributionscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.