arXiv:2602.05600cs.LG2026-02被引 3

揭示了深度学习中噪声与损失曲率的超线性关系,挑战了传统假设。

On the Superlinear Relationship between SGD Noise Covariance and Loss Landscape Curvature

  • 基于活动-权重对偶性,发现噪声协方差与每样本海塞矩阵平方期望成正比。
  • 实测显示各层噪声与曲率满足幂律关系,指数γ在1到2之间。
  • 适用于多种模型和数据集,为优化器设计提供理论依据。

随机梯度下降(SGD)引入各向异性的噪声,该噪声与损失曲面局部曲率相关,从而偏向平坦极小值。以往研究常假设负对数似然损失下费舍尔信息矩阵与海塞矩阵等价,导致认为SGD噪声协方差 𝐶 与海塞矩阵 𝐻 成比例。我们证明这一假设仅在严格条件下成立,通常不适用于深度神经网络。利用新发现的活动-权重对偶性,我们建立了一种更普适的关系:𝐶 ∝ 𝔼ₚ[𝐡ₚ²],其中 𝐡ₚ 是每样本海塞矩阵,𝐻 = 𝔼ₚ[𝐡ₚ]。因此,𝐶 与 𝐻 近似可交换而非完全一致。进一步发现在全连接层中,其对角元素满足逐层经验幂律 𝐶ᵢᵢ ∝ 𝐻ᵢᵢᵞ,层间拟合指数 γ 范围为 1 ≤ γ ≤ 2。跨数据集、架构与损失函数的实验验证了该层间边界,为深度学习中的噪声-曲率关系提供了统一刻画。

原文摘要 · Abstract (English)

Stochastic Gradient Descent (SGD) introduces anisotropic noise that is correlated with the local curvature of the loss landscape, thereby biasing optimization toward flat minima. Prior work often assumes an equivalence between the Fisher Information Matrix and the Hessian for negative log-likelihood losses, leading to the claim that the SGD noise covariance $\mathbf{C}$ is proportional to the Hessian $\mathbf{H}$. We show that this assumption holds only under restrictive conditions that are typically violated in deep neural networks. Using the recently discovered Activity--Weight Duality, we find a more general relationship agnostic to the specific loss formulation, showing that $\mathbf{C} \propto \mathbb{E}_p[\mathbf{h}_p^2]$, where $\mathbf{h}_p$ denotes the per-sample Hessian with $\mathbf{H} = \mathbb{E}_p[\mathbf{h}_p]$. As a consequence, $\mathbf{C}$ and $\mathbf{H}$ commute approximately rather than coincide exactly. We further find that, within the analyzed fully connected layers, their diagonal elements follow per-layer empirical power laws $C_{ii} \propto H_{ii}^γ$, with layer-dependent fitted exponents bounded by $1 \leq γ\leq 2$. Experiments across datasets, architectures, and loss functions support the resulting layerwise bounds, providing a unified characterization of the noise-curvature relationship in deep learning.

优化器损失曲率深度学习

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