风格试刊约 13 分钟

噪声、方向与实验预算

Muon 的 batch-size 效应 · Scaling law 的主动实验设计

从方向相关的梯度噪声,到发现 scaling law 所需的实验预算。本期展开两篇研究,附两条相关动态;另精读一篇经典:S4。

MuonScaling laws
每日精读 · 2022S4
主读 01已读相关正文与作者解读

Batch size 与方向相关的优化动力学

The Best Optimizer Depends on Batch Size
Xingyu Dang · Kaiyue Wen · Sadhika Malladi
首发 10.06 · v2 10.12

固定样本预算 TT,增大 batch size BB 会降低梯度噪声,同时减少更新次数。论文用 noisy quadratic model 区分这两个因素。[1]

模型
f(w)=12∑ihiwi2,g^i=hiwi+ciB zi,K=TB.f(w)=\frac12\sum_i h_iw_i^2,\qquad \widehat g_i=h_iw_i+\sqrt{\frac{c_i}{B}}\,z_i,\qquad K=\frac{T}{B}.

hi>0h_i>0:曲率;ci>0c_i>0:单样本梯度噪声方差;zi∼N(0,1)z_i\sim\mathcal N(0,1)。方向相关的曲率—噪声比为 CNRi=hi2/ci\mathrm{CNR}_i=h_i^2/c_i。

理想 Muon 在 1×11\times1 矩阵上对应 sign 操作,因此先考察带 momentum 的 SignSGD:

更新
mt+1=μmt+(1−μ)g^t,wt+1=wt−η sign⁡(mt+1),0≤μ<1.m_{t+1}=\mu m_t+(1-\mu)\widehat g_t,\qquad w_{t+1}=w_t-\eta\,\operatorname{sign}(m_{t+1}),\quad 0\le\mu<1.

局部机制。 固定 wiw_i,令 momentum 达到稳态。在独立高斯噪声下,

响应
E[sign⁡(mi)]=erf⁡(qi),qi=wiB(1+μ)hi22(1−μ)ci.\mathbb E[\operatorname{sign}(m_i)]=\operatorname{erf}(q_i),\qquad q_i=w_i\sqrt{\frac{B(1+\mu)h_i^2}{2(1-\mu)c_i}}.

每处理一个样本的平均位移与 η erf⁡(qi)/B\eta\,\operatorname{erf}(q_i)/B 成正比。保持这一局部位移给出:

两种区间
η(B)∝{B,∣qi∣≪1noise-dominated,B,∣qi∣≫1signal-dominated.\eta(B)\propto \begin{cases}\sqrt B,& |q_i|\ll1 \quad \text{noise-dominated},\\[2pt] B,& |q_i|\gg1 \quad \text{signal-dominated}.\end{cases}

方向干预。 从同一 checkpoint 出发,将 batch 从 128K128\mathrm K 增至 2M2\mathrm M,仅在选定方向保留小 batch 更新。局部 loss 差距恢复比例为

指标
R=Llarge−LmixedLlarge−Lsmall.R=\frac{L_{\rm large}-L_{\rm mixed}}{L_{\rm large}-L_{\rm small}}.
保留小 batch 更新的方向局部差距恢复
最尖锐的 768 个方向,约占 0.001%0.001\%最高 59.5%
同维数随机子空间变化幅度至多约 3%
同一数据流上的局部曲率诊断;sharp directions 来自 Muon 预条件后的曲率近似。[2]
相关动态基于摘要

噪声可以平滑 sign,也可能改变特征学习顺序

From Parallel to Sequential: Small-Batch Noise Restores Simplicity Bias of Muon
Tom Jacobs · Rebekka Burkholz
arXiv 10.12 · OpenReview 已有 10.01 记录

作者报告:小 batch 噪声能恢复 full-batch spectral descent 中缺失的顺序特征学习,并在深线性模型中形式化这一机制。[3][4]

下面是用于理解“噪声平滑”的标量示意推导。取 z∼N(0,1)z\sim\mathcal N(0,1)、σ>0\sigma>0,则

标量直觉
E[sign⁡(g+σz)]=2Φ(g/σ)−1=2πgσ+O ⁣((g/σ)3),∣g∣≪σ.\begin{aligned}\mathbb E[\operatorname{sign}(g+\sigma z)]&=2\Phi(g/\sigma)-1\\&=\sqrt{\frac2\pi}\frac g\sigma+O\!\left((g/\sigma)^3\right),\qquad |g|\ll\sigma.\end{aligned}
主读 02已读相关正文

发现 scaling law,本身也有算力预算

必要背景

Chinchilla(2022)通过小规模训练实验研究模型与数据的算力分配;SLDBench(2025,ICLR 2026)进一步让 agent 自动发现预测规律。新的问题是:获得拟合数据本身也要付出训练成本。[6][7]

候选规律
L(N,D)=E+aN−α+bD−β.L(N,D)=E+aN^{-\alpha}+bD^{-\beta}.

上式只是候选形式。Agent 既不知道正确函数形式,也不能免费获得全部观测;它拿到的是带价格的实验目录。[5]

实验目录
U={(xi,ci)}i=1n,ci>0.\mathcal U=\{(x_i,c_i)\}_{i=1}^{n},\qquad c_i>0.

xix_i 是训练配置,cic_i 通常为训练 FLOPs;购买后才看到结果 yiy_i。略去分组记号,每轮购买 AtA_t,在时刻 τ\tau 停止:

支出
St+1=St∪At,ρ=∑i∈Sτci∑i∈Uci.S_{t+1}=S_t\cup A_t,\qquad \rho=\frac{\sum_{i\in S_\tau}c_i}{\sum_{i\in\mathcal U}c_i}.

评价同时记录 (更大规模上的外推误差, ρ)(\text{更大规模上的外推误差},\,\rho)。因此需要联合决定买什么实验、何时停止、拟合什么规律。测试配置比候选实验更大,无法直接购买。[5]

设置各系统的中位 R2R^2 范围实验成本
Passive · 全部观测0.723–0.753100%
Active · 自行购买0.235–0.570各系统平均约 8%–44%
30 个筛选出的困难任务,8 个 agent 系统;R² 范围跨系统,不是置信区间。[5]
简讯基于摘要

Monarch-Muon:将大矩阵正交化拆成小块

Monarch-Muon: Structure-Aware Muon for Fast and Memory-Efficient Pretraining
Alexander Miasnikov 等
arXiv 10.12 · OpenReview 已有 10.01 记录

Monarch 因子对权重施加结构,减少权重、梯度和优化器状态,并产生小块更新子问题。摘要中的 6.89B 系统测量为:[8][9]

测量原设置Monarch-Muon
峰值显存68.9 GB52.9 GB
Muon optimizer step1288 ms304 ms
每日精读 · 经典 001已读方法、理论与实验相关章节

S4:一个状态空间,三种计算方式

Efficiently Modeling Long Sequences with Structured State Spaces
Albert Gu · Karan Goel · Christopher Ré
首发 2021.10 · ICLR 2022 · 本次阅读 v3(2022.08)
ICLR 2022 Oral · Outstanding Paper Honorable Mention ↗
从哪里来

长序列既需要保留历史,也需要高效训练。HiPPO(2020)用多项式投影设计连续记忆;S4(2022)进一步寻找可快速计算、数值稳定的状态空间参数化。[10][13]

先看一个输入、输出均为标量的通道,隐藏状态 x(t)∈RNx(t)\in\mathbb R^N。AA 控制记忆演化,BB 写入输入,CC 读出状态;略去可单独计算的 skip term。[12]

连续系统
x˙(t)=Ax(t)+Bu(t),y(t)=Cx(t).\dot x(t)=Ax(t)+Bu(t),\qquad y(t)=Cx(t).

以步长 Δ>0\Delta>0 作双线性离散化;假设逆矩阵存在。Δ\Delta 是采样时间尺度,可作为模型参数学习。[12]

离散化
Aˉ=(I−Δ2A)−1(I+Δ2A),Bˉ=(I−Δ2A)−1ΔB.\bar A=\left(I-\frac{\Delta}{2}A\right)^{-1}\left(I+\frac{\Delta}{2}A\right),\qquad \bar B=\left(I-\frac{\Delta}{2}A\right)^{-1}\Delta B.

零初始状态 x−1=0x_{-1}=0,固定参数时,逐步递推与整段卷积完全等价:[12]

RNN ↔ 卷积
xk=Aˉxk−1+Bˉuk,yk=∑j=0kKjuk−j,Kj=CAˉjBˉ.\begin{aligned}x_k&=\bar A x_{k-1}+\bar B u_k,\\ y_k&=\sum_{j=0}^{k}K_j u_{k-j},\qquad K_j=C\bar A^j\bar B.\end{aligned}

真正的瓶颈是计算核。 直接对 HiPPO 矩阵做特征分解可能严重病态。S4 的关键是“正规矩阵 + 低秩修正”,再用酉变换得到 DPLR:[10]

结构化参数化
A=V(Λ−PQ∗)V∗,V∗V=I,P,Q∈CN×r,r≪N.A=V(\Lambda-PQ^*)V^*,\qquad V^*V=I,\qquad P,Q\in\mathbb C^{N\times r},\quad r\ll N.

Λ\Lambda 为对角矩阵。令 A^=Λ−PQ∗\widehat A=\Lambda-PQ^*、R(ω)=(ωI−Λ)−1R(\omega)=(\omega I-\Lambda)^{-1},在相关逆存在时,Woodbury 恒等式给出:[10]

低秩修正
(ωI−A^)−1=R−RP(I+Q∗RP)−1Q∗R.(\omega I-\widehat A)^{-1}=R-RP(I+Q^*RP)^{-1}Q^*R.

高维求逆由对角求逆与 r×rr\times r 修正替代。论文再把有限长卷积核的生成函数化为 Cauchy 计算并作逆 FFT:固定低秩时,理论核构造为 O~(N+L)\widetilde O(N+L),每步递推为 O(N)O(N)。这是单个 SSM 的复杂度,快速 Cauchy 算法是前提。[10]

Path-X:长度 16,384 的连通性分类准确率
随机猜测50.00%
S4 original88.10%
采用 v3 附录 Table 10 的 original 行;后续改进结果另列为 updated。原始 Path-X 配置:6 层、256 通道、每通道状态维度 64,单张 A100 40GB。[10]

READ THE ORIGINALS

原始来源

  1. The Best Optimizer Depends on Batch Size · v2
  2. The Best Optimizer Depends on Batch Size · 作者解读
  3. From Parallel to Sequential · arXiv
  4. Small Batch Noise Restores Simplicity Bias of Muon · OpenReview
  5. SLDBench v2 · 正文
  6. Training Compute-Optimal Large Language Models · 2022
  7. Can Language Models Discover Scaling Laws? · 2025/2026
  8. Monarch-Muon · arXiv
  9. Monarch-Muon · OpenReview
  10. S4 · arXiv v3,§2–3、Appendix D.2/Table 10
  11. ICLR 2022 · Oral 与 Outstanding Paper Honorable Mention 官方名单
  12. S4 作者解读 · 连续时间、递推与卷积
  13. HiPPO (2020) · Recurrent Memory with Optimal Polynomial Projections
检索窗口

2026-10-12 — 2026-10-14 · America/New_York

READ · THINK · DISCUSS

讨论与点赞

交流推导、补充文献,或推荐下一篇精读。

在下方直接评论、回复或点赞。首次参与需要 GitHub 登录授权;评论公开可见。

正在加载讨论区…

在 GitHub 打开讨论 ↗

← 返回日报归档