arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.06366cs.LGstat.ML

深度线性网络中鞍点到鞍点区域的随机梯度下降

Stochastic Gradient Descent in the Saddle-to-Saddle Regime of Deep Linear Networks

Moirai;University of Oxford(牛津大学);University of California, Berkeley(加州大学伯克利分校);Iliad

Guillaume Corlouer, Avi Semler, Alexander Strang, Alexander Gietelink Oldenziel

AI总结:

研究深度线性网络中随机梯度下降在鞍点到鞍点区域的动力学,推导出各模式下的随机微分方程分解,并分析SGD噪声对特征学习的影响。

摘要:

深度线性网络(DLNs)被用作深度神经网络训练动态的可分析模型。尽管在DLNs中已知梯度下降表现出鞍点到鞍点动态,但随机梯度下降(SGD)噪声对这一区域的影响仍不明确。我们研究了在鞍点到鞍点区域训练DLNs时SGD的动力学。我们将训练动态建模为具有各向异性、状态依赖性噪声的随机 Langevin 动力学。在假设权重对齐且平衡的情况下,我们推导出动态的精确分解,将其分解为一组一维每模式随机微分方程。这表明最大扩散沿模式先于相应特征完全学习。我们还推导出每个模式的SGD平稳分布:在无标签噪声时,其特定特征的边缘分布与梯度流的平稳分布一致,而在存在标签噪声时,它近似于玻尔兹曼分布。最后,我们通过实验验证理论结果在无对齐或平衡权重的情况下仍成立。这些结果表明SGD噪声编码了特征学习进展的信息,但不根本改变鞍点到鞍点动态。

英文摘要:

Deep linear networks (DLNs) are used as an analytically tractable model of the training dynamics of deep neural networks. While gradient descent in DLNs is known to exhibit saddle-to-saddle dynamics, the impact of stochastic gradient descent (SGD) noise on this regime remains poorly understood. We investigate the dynamics of SGD during training of DLNs in the saddle-to-saddle regime. We model the training dynamics as stochastic Langevin dynamics with anisotropic, state-dependent noise. Under the assumption of aligned and balanced weights, we derive an exact decomposition of the dynamics into a system of one-dimensional per-mode stochastic differential equations. This establishes that the maximal diffusion along a mode precedes the corresponding feature being completely learned. We also derive the stationary distribution of SGD for each mode: in the absence of label noise, its marginal distribution along specific features coincides with the stationary distribution of gradient flow, while in the presence of label noise it approximates a Boltzmann distribution. Finally, we confirm experimentally that the theoretical results hold qualitatively even without aligned or balanced weights. These results establish that SGD noise encodes information about the progression of feature learning but does not fundamentally alter the saddle-to-saddle dynamics.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑