研究噪声标签下两层线性网络的训练动态,揭示其分阶段学习机制。
On the Learning Dynamics of Two-layer Linear Networks with Label Noise SGD
- 分析噪声标签下SGD的双阶段学习过程:权重衰减后进入丰富模式。
- 发现噪声促进模型从懒惰区向丰富区过渡,提升泛化能力。
- 适用于理解深度学习优化机制,尤其关注噪声与泛化的关联。
深度学习成功的关键因素之一在于梯度训练算法中固有噪声带来的隐式偏差。受训练时使用噪声标签可提升模型泛化能力的实证观察启发,我们深入探究了带标签噪声的随机梯度下降(SGD)的内在机制。针对两层过参数化线性网络,我们分析了标签噪声SGD的学习动态,揭示了两阶段行为:在第一阶段,模型权重幅度逐渐减小,模型脱离懒惰区,进入丰富区;在第二阶段,模型权重与真实插值器对齐程度提高,最终收敛。分析表明,标签噪声在推动模型从懒惰区到丰富区的转变中起关键作用,并最小化解释了其经验有效性。此外,我们将这些见解扩展至尖锐感知最小化(SAM),证明标签噪声SGD的原则也适用于更广泛的优化算法。在合成和真实世界设置下的大量实验有力支持了该理论。代码已开源:https://github.com/a-usually/Label-Noise-SGD。
原文摘要 · Abstract (English)
One crucial factor behind the success of deep learning lies in the implicit bias induced by noise inherent in gradient-based training algorithms. Motivated by empirical observations that training with noisy labels improves model generalization, we delve into the underlying mechanisms behind stochastic gradient descent (SGD) with label noise. Focusing on a two-layer over-parameterized linear network, we analyze the learning dynamics of label noise SGD, unveiling a two-phase learning behavior. In \emph{Phase I}, the magnitudes of model weights progressively diminish, and the model escapes the lazy regime; enters the rich regime. In \emph{Phase II}, the alignment between model weights and the ground-truth interpolator increases, and the model eventually converges. Our analysis highlights the critical role of label noise in driving the transition from the lazy to the rich regime and minimally explains its empirical success. Furthermore, we extend these insights to Sharpness-Aware Minimization (SAM), showing that the principles governing label noise SGD also apply to broader optimization algorithms. Extensive experiments, conducted under both synthetic and real-world setups, strongly support our theory. Our code is released at https://github.com/a-usually/Label-Noise-SGD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。