精读笔记
Problem Setting
论文标题：Scaling Limits of Constant-Stepsize SGD at Flat Minima（arXiv preprint / 2026-07-21）。
这篇论文实际解决的是常步长 SGD 的“last iterate steady-state law”在 flat minima 附近如何缩放、极限是什么，而不是传统意义上的 convergence-to-optimum。对象是固定 alpha 下的 Markov chain invariant law，然后研究 alpha -> 0 的稳态极限。
关键矛盾是：强凸情形依赖极小点附近非退化 Hessian，曲率给出 1-c alpha 收缩和 sqrt(alpha) fluctuation；但 flat minimum 中 Hessian 在 invariant law 集中的区域趋零，欧氏收缩恰好在最重要区域失效。另一方面，若目标函数尾部只有 subquadratic drift，二次 Lyapunov 又过强，不能自然控制 excursions。Markovian noise 进一步使当前 iterate 与未来噪声相关，不能直接套 iid martingale argument。
以前方法主要卡在两个地方：一是固定 alpha 的 invariant measure existence/uniqueness 需要全局稳定性和局部收缩同时成立；二是 scaling limit 需要在非线性局部漂移和相关噪声下正确识别 generator。强凸 OU 近似在 m>2 时给错尺度，subquadratic SGD 的既有分析多停留在 m=2。
Motivation
已有路线默认局部二次结构是主导因素，所以自然得到 Gaussian stationary approximation。但 quantile、CVaR 等统计目标在分布函数高阶穿越时，population gradient 可能以 |x|^{m-1} 消失；这不是 nonsmooth artifact，而是局部识别强度本身变弱。此时 sqrt(alpha) 标准化会过强，真实稳态波动更大。
作者的核心观察是需要把 local flatness exponent m 和 tail exponent beta 分开：m 决定 invariant law 的小步长尺度和非 Gaussian 极限；beta 只决定远离 minimizer 时的 Lyapunov weight。这个区分是论文的动机核心。很多已有工作把“非强凸/鲁棒损失/尾部次二次”混在一起看，但这里指出真正改变 stationary scaling 的是局部漂移阶数，不是尾部增长。
关键缺口是 rigorous invariant-law scaling limit：之前 flat case 多是数值展示、Gibbs approximation conjecture 或一维条件性论证；本文试图给出多维、Markovian noise 下的无条件极限理论。
Core Idea
核心思想可以概括为：在 stationarity 下，常步长 SGD 的噪声注入和 deterministic restoring force 达到平衡；若 h(x) 约为 ||x||^{m-2}x，则平衡尺度不是 sqrt(alpha)，而是 alpha^{1/m}。把 X_infty 缩放为 Y_alpha = alpha^{-1/m} X_infty 后，局部 homogeneous drift h0 留下来，离散噪声通过 central-limit scaling 变成 Brownian forcing，得到 dY_t = -h0(Y_t)dt + Sigma^{1/2}dB_t。
这相当于把 constant-step SGD 的稳态建模从“linearized OU around minimizer”改成“homogeneous nonlinear diffusion around flat minimizer”。本质区别不在于换了证明技术，而是拒绝把局部 landscape 线性化；对于 m>2，线性化 drift 为零，继续走 OU 路线没有信息。
另一个关键思想是固定 alpha 的 ergodicity 不能用普通 Euclidean contraction 证明，而要在 alpha-dependent induced metric 中恢复 contraction。这个 metric 同时编码远端 tail control 和近端 flat-region correction，使得链在全局有唯一稳态，然后才能合法讨论稳态下 alpha -> 0。
Method
1. 固定 alpha 的 invariant law：论文采用 Qu-Blanchet-Glynn 类型的 weight-induced Wasserstein metric，但没有直接验证其原始条件。原因是 stochastic-gradient Jacobian 可能 samplewise 只有 nonexpansive，甚至 rank deficient，不能逐样本给严格收缩。作者改成 directional kernel estimate，在 expectation 中证明收缩。
2. alpha-dependent weight：尾部权重 T_beta 在 beta<2 时取 exp(kappa ||x||^{2-beta})，对应 deterministic flow 在远端的 |x|^{2-beta} 线性下降；近 minimizer 的 omega correction 只在 m>2 时启用，用很小的 delta_alpha 幅度制造 flat core 中的有效收缩。这个设计的作用是把局部退化和全局弱恢复力放在同一个 metric 中处理。
3. 收缩率：得到 1 - c alpha^{m-1}。这不是普通 strongly convex 的 1-c alpha，而是曲率在 typical scale alpha^{1/m} 上约为 alpha^{(m-2)/m} 后形成的更慢混合。这里的速率反映 flatness，而不是 proof artifact。
4. Markovian noise 的 Poisson decomposition：通过 chi_x - Q chi_x = Q g_x，把 correlated noise 写成 martingale difference D_x 加 telescoping term。generator limit 中 telescoping 部分低阶消失，quadratic variation 给出 Green-Kubo covariance Sigma，而不是 marginal covariance。
5. scaling limit：先用 stationary moment estimates 证明 Y_alpha tight，再用 perturbed test function / generator identity 识别所有 subsequential limits，最后用 limiting nonlinear diffusion 的 invariant-law uniqueness 关闭证明。
Key Insight / Why It Works
最核心贡献是尺度归因：flat minimum 下 steady-state fluctuation 的尺度由 local drift order 决定，而不是由全局 convexity 或 tail behavior 决定。m 是 local identification strength；beta 是 excursion-control exponent。这个分离非常干净，也最值得迁移。
为什么 alpha^{1/m} 成立：在 stationarity 下，二阶噪声项贡献约 alpha^2 Var(g)，漂移耗散项约 alpha <X,h(X)>；若 <x,h(x)> 约 ||x||^m，则 stationarity square identity 给 E||X||^m = O(alpha)，所以 X = O(alpha^{1/m})。这个能量平衡是整篇文章的机制核心。后续 Wasserstein metric 和 Poisson equation 都是在让这个平衡严格成立。
非 Gaussian 极限不是附带现象，而是 flatness 的必然结果。m=2 时 h0 线性，stationary diffusion 是 OU，因此 Gaussian；m>2 时 h0 非线性，stationary density 一般类似 exp(-potential)，不会 Gaussian。换句话说，Gaussian approximation 失效不是因为 noise 非 Gaussian，而是因为 drift linearization 丢失了主导项。
Markovian covariance 的处理也是实质贡献。若只看 marginal variance，会低估或错估 temporal dependence 的稳态扩散强度。Poisson equation 在这里不是技术装饰，而是把 correlated stream 转成正确 Brownian covariance 的唯一自然方法。
我认为最可能是核心贡献的部分是：固定 alpha ergodicity 的 weighted Wasserstein contraction + alpha^{-1/m} stationary generator limit 的组合。单独的 scaling heuristic 已经不新；真正新增是把它在多维、Markovian、subquadratic-tail 条件下做成完整 invariant-law theorem。
相对辅助的部分是数值实验和 separable unequal-exponent extension。实验主要验证机制，没有承担强 empirical claim。separable case 很有启发，但本质上是 coordinatewise theorem 的拼接；真正难的非可分 anisotropic flatness 仍未解决。
这不是 scaling / data 带来的增益，也不是工程 trick；它是更合适的 local asymptotic model。若迁移到 ML 语境，insight 是：constant learning-rate noise floor 的形状取决于 basin 的最低非零 Taylor order，而不是只取决于 Hessian sharpness。
Relation To Prior Work
最接近的谱系是 constant-step stochastic approximation / SGD invariant law theory，包括 Pflug、Mandt-Hoffman-Blei、Dieuleveut-Durmus-Bach、Chen-Mou-Maguluri，以及近期 Markovian SA bias / weak approximation。强凸路线的共同点是先得到 invariant law，再做 alpha -> 0；不同点是它们的局部模型是 linear drift，因此极限是 OU。
与 Chen-Mou-Maguluri 的关系：后者已经数值展示 quartic flat case 的 non-Gaussian scaling，但主要理论覆盖 Gaussian/contractive settings。本文把这个现象变成 flat-minimum theorem。
与 Wang et al. 的关系：Wang et al. 给出一维 flat convex Gibbs approximation 并依赖 conjecture / Stein regularity；本文绕开 finite-alpha quantitative approximation，改用 weak convergence + invariant generator equation，因此更稳健，但也不提供 rate。
与 Zhang et al. subquadratic SGD 的关系：那条线处理 beta<2 但 local m=2，所以 scaling 仍是 sqrt(alpha)。本文把 beta 和 m 解耦，说明 robust/logistic 这类 subquadratic tail 并不自动导致非 Gaussian scaling；只有 local flatness 才会。
与 Qu-Blanchet-Glynn 的关系：metric framework 借用已有思想，但关键 contraction estimate 是重新做的。看似只是采用 induced metric，实质创新在于 direction-kernel expectation contraction，因为 samplewise map 不必严格收缩。
总体上，这篇属于 stochastic approximation steady-state asymptotics 的理论推进，不属于 optimizer engineering，也不是 benchmark-driven ML paper。
Dataset / Evaluation
这篇没有传统 dataset / benchmark evaluation。实验是低维 numerical experiments，用来验证理论机制：median/quantile flatness 给出 alpha^{1/m} 尺度与 quartic non-Gaussian density；Markovian sign stream 验证 Green-Kubo covariance 而非 marginal variance；separable m1=2,m2=4 验证 common scaling 下只有最 flat coordinate 保留非退化极限；Charbonnier/logistic examples 用来说明 beta 控制 tail 而不改变 m=2 的 Gaussian local scaling。
这些 evaluation 支持的是机制 claim，而不是算法性能 claim。它们覆盖了 local flatness、temporal dependence、unequal exponents、subquadratic tails 四个理论点，设计上是合理的。但它们没有验证高维非可分目标、真实深度模型、复杂 Markovian data stream 或 finite-alpha rate。因此不能把实验理解成“该理论已经解释实际大模型训练噪声地板”。
文中未充分说明 finite alpha 下理论常数是否可用；实验也没有展示 contraction bound 的 tightness。数值部分更多是 sanity check，不是理论上限的压力测试。
Limitation
最重要的限制是结构假设强。目标函数要求凸、C2、唯一 minimizer 平移到 0，并且局部有有限阶 homogeneous expansion；这排除了指数平坦、非光滑 pinball 原始形式、非凸多 basin 和深度网络常见的退化 manifold minima。
Markovian noise 假设也偏强：driving chain 要 contractive recursion，甚至一步 contraction；论文最后也承认 multi-step contraction / Harris mixing 才更接近一般 Markovian data。现实数据流的 mixing 未必能写成这种 uniformly contractive map。
co-coercivity 和 mean perturbation 条件限制了 stochastic gradient map 的形状。虽然 logistic example 可验证，但在一般非凸模型、batchnorm、adaptive optimizer 或 data-dependent augmentation 下不一定成立。
非可分 anisotropic flatness 是核心未解点。separable unequal exponents 的结论很清楚，但真实 landscape 中不同方向通常耦合；此时多个 alpha^{1/m_i} scale 会相互作用，generator 可能出现 multi-time-scale averaging。当前结果没有解决这个真正困难的版本。
没有 finite-alpha convergence rate 是另一个上限。论文证明 weak convergence，但不给 W-distance rate 或 bias expansion；因此无法直接回答实际 alpha 下 Gaussian/non-Gaussian approximation 误差多大。增益来源不清的问题主要在 quantitative sharpness，而不是定性机制。
此外，固定 alpha ergodicity 的 contraction factor 1-c alpha^{m-1} 在 m 大时非常慢，说明 flat minima 虽然带来更大 stationary noise scale，也带来慢混合。这个理论上限对实际采样/训练时间很关键，但文中没有深入讨论其 computational implication。
Takeaway
- 1. 常步长 SGD 的稳态尺度应该由 local drift order 判断；Hessian-based sharpness 只覆盖 m=2，flat minima 必须看更高阶项。
- 2. beta<2 的 subquadratic tail 不等于非 Gaussian local limit。
- beta 管远端稳定性，m 管局部 invariant-law scaling；这一区分可以迁移到 robust statistics、risk estimation 和其他弱识别问题。
- 3. Markovian noise 下稳态扩散强度必须用 Green-Kubo covariance。
一句话总结
这篇论文把常步长 SGD 的稳态理论从强凸 OU/Gaussian 范式推进到 flat-minimum nonlinear-diffusion 范式，核心贡献是证明 local flatness exponent m 决定 alpha^{1/m} 尺度与非 Gaussian invariant-law limit，而 tail exponent beta 和 Markovian covariance 分别只控制 excursions 与扩散强度。