跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

AdamW训练动态下Weibull权重尺度参数的演化

文章摘要

本文基于两参数Weibull框架,研究AdamW训练中权重尺度参数λ先增长、超调后松弛的现象,提出三力分解(对齐力、注入力、衰减力),并在Pythia-70M上验证对齐力主导上升阶段,贡献88-94%的力预算;同时提出样条位移法从稀疏检查点恢复对齐力,精度达92-94%。

来源arXiv Machine Learning作者: Tiexin Ding
AdamW训练动态下Weibull权重尺度参数的演化
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

在一篇新提交于 arXiv 的机器学习论文中,作者 Tiexin Ding 在既有的双参数 Weibull 框架基础上,进一步研究了 Transformer 权重分布在 AdamW 训练过程中为何会呈现出“先增长、再过冲、最后回落”的演化模式。该论文题为《Weibull Weight-Scale Parameter Evolution under AdamW Training Dynamics》,于 2026 年 6 月 11 日提交,属于 cs.LG(机器学习)分类,全文共 21 页、含 14 幅图,代码与数据已在 GitHub 上公开(仓库名为 NPM-Weibull-public)。

研究的核心对象是 Weibull 权重尺度参数 λ。此前的工作已用双参数 Weibull 分布来刻画 Transformer 权重分布,并以此作为诊断工具;而这项工作要回答的问题是:在 AdamW 优化器驱动下,λ 为什么会上升、为何会暂时超过平衡点,随后又如何回落。为解释这一过程,作者从 AdamW 更新公式出发,对权重平方范数做了领头阶的“三力分解”,把影响权重范数变化的因素拆成三个部分:第一是对齐力(alignment force),衡量权重与自适应更新方向之间的相关性;第二是注入力(injection force),来自自适应步长的大小;第三是衰减力(decay force),来自解耦的权重衰减项。

作者在自行训练的 Pythia-70M 模型上进行了实验,这些模型拥有真实的优化器动量信息(ground-truth optimizer moments),因此可以精确计算上述三种力。结果显示,在权重尺度上升阶段,对齐力占据主导地位:在四个随机种子下,对齐力贡献了绝对力预算的 88% 至 94%,并且在移除“超级权重”(super-weight)后依然保持稳健。到了接近饱和的阶段,对齐力与衰减力逐渐趋于平衡,这正好解释了权重尺度从增长转向回落的转折。这些力的动态直接决定了 λ(t) 背后的平方范数分量;至于 RMS 到 Weibull 重建过程中剩余的偏差,则是可测量的,并可进一步拆分为桥接分量(bridge)与积分分量(integration),在密集采样区域内合计约为 5% 至 6%。

为了让分析能够推广到无法获得优化器动量的真实模型,论文提出了一种“样条位移法”(spline displacement method),可以从稀疏的检查点中恢复对齐力,准确率约为 92% 至 94%,大约是朴素两点基线方法的两倍。此外,作者在实验中观察到 λ(t) 的峰值会随训练数据的一致性(coherence)而变化,这暗示权重尺度增长中存在依赖数据的成分;不过作者对这一现象持谨慎态度,将其留待后续受控研究进一步验证,并未在本文中给出定论。总体而言,该工作把优化器动力学与权重分布诊断联系起来,为理解 AdamW 训练中权重尺度的演化提供了可量化的力学解释,同时也明确指出了现有分析的边界与待解问题。

展开要点与分析

文章情报

工程师进阶

要点

  • 从AdamW更新推导出平方权重范数的三力分解:对齐力、注入力、衰减力。
  • 在Pythia-70M上,对齐力在上升阶段贡献88-94%的绝对力预算,且对超权重移除稳健。
  • 接近饱和时,对齐力与衰减力趋于平衡,解释权重尺度从增长转向松弛。
  • 提出样条位移法,从稀疏检查点恢复对齐力,精度约92-94%,是朴素两点基线的两倍。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。