神经网络训练中的普适锐度动态:不动点分析、稳定性边缘与通向混沌的路径
机器学习
2025-02-17 v2 无序系统与神经网络
混沌动力学
机器学习
摘要
在神经网络的梯度下降动态中,损失 Hessian 矩阵的最大特征值(锐度)在整个训练过程中表现出多种稳健现象。这包括训练早期锐度可能下降的早期阶段(锐度降低),以及后期行为如渐进锐化与稳定性边缘。我们证明,在单个训练样本上训练的简单层线性网络(UV 模型)展现了真实世界场景中所观察到的全部本质锐度现象学。通过分析函数空间中动态不动点的结构以及函数更新的向量场,我们揭示了这些锐度趋势背后的潜在机制。我们的分析揭示了(i)早期锐度降低与渐进锐化背后的机制,(ii)稳定性边缘所需的条件,(iii)初始化与参数化的关键作用,以及(iv)随着学习率增大,在稳定性边缘流形上出现的倍周期通向混沌路径。最后,我们证明这一简化模型的多种预测可推广至真实世界场景,并讨论了其局限性。
引用
@article{arxiv.2311.02076,
title = {Universal Sharpness Dynamics in Neural Network Training: Fixed Point Analysis, Edge of Stability, and Route to Chaos},
author = {Dayal Singh Kalra and Tianyu He and Maissam Barkeshli},
journal= {arXiv preprint arXiv:2311.02076},
year = {2025}
}
备注
Accepted at ICLR 2025 (camera-ready version). Update: added language modeling experiments