arXiv:2606.19367cs.LGstat.ML2026-06被引 1

解析AdamW训练中权重尺度参数λ的动态变化机制

Weibull Weight-Scale Parameter Evolution under AdamW Training Dynamics

论文配图:Weibull Weight-Scale Parameter Evolution under AdamW Training Dynamics
图 1 · 摘自论文原文
  • 分解权重范数增长为对齐、注入、衰减三力,对齐主导初期增长
  • λ峰值受训练数据一致性影响,饱和时对齐与衰减力平衡
  • 提出样条位移法从稀疏检查点重建对齐力,精度达92-94%

基于双参数威布尔框架分析Transformer权重分布,研究AdamW训练中权重尺度参数λ先增长、超调后松弛的原因。推导出平方权重范数的三力分解:衡量权重与自适应更新方向相关性的对齐力,来自自适应步长的注入力,以及解耦权重衰减带来的衰减力。在自训练的Pythia-70M模型上,对齐力主导上升期,贡献88-94%的绝对力预算,且对超大权重移除具有鲁棒性。接近饱和时,对齐力与衰减力趋于平衡,解释了尺度参数从增长到松弛的转变。这些力动态直接控制λ(t)的平方范数成分;剩余的RMS到威布尔重构偏移可测量,由桥接与积分分量构成,密集采样区域总计约5-6%。为扩展至真实模型(无优化器时刻信息),提出样条位移法,从稀疏检查点恢复对齐力,精度约92-94%,约为朴素两点基线的两倍。进一步观察到λ的峰值随训练数据一致性变化,提示权重尺度增长存在数据依赖成分,留待后续控制实验研究。代码与数据见https://github.com/tiexinding/NPM-Weibull-public。

原文摘要 · Abstract (English)

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.

权重分析优化器动力学威布尔分布

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。