学习率大小决定神经网络训练的效率阶段,可引发相变
From Information to Generative Exponent: Learning Rate Induces Phase Transitions in SGD
- 用不同学习率多步更新实现非相关梯度,突破传统限制
- 小学习率时样本复杂度由信息指数决定,大学习率时转为生成指数
- 新分层训练法无需重用样本即可提升效率,适合高效模型设计
为理解神经网络中的特征学习动态,近期理论研究聚焦于基于梯度的高斯单指标模型,其中标签是输入低维投影的非线性函数。在线SGD的样本复杂度由链接函数的信息指数决定,但通过在相同样本上进行多步梯度更新并采用不同学习率,可实现非相关更新规则,此时复杂度受更小的生成指数限制。然而,该结论仅在学习率足够大时成立。本文刻画了学习率与样本复杂度之间的关系,揭示在某些情况下存在从小学习率的‘信息指数阶段’到大学习率的‘生成指数阶段’的相变。该框架涵盖单遍SGD和批重复用的分析,并提出一种新的分层训练算法,通过各层使用不同学习率的双时间尺度方法,在不重用样本或修改损失函数(如平方误差)的前提下,突破相关查询的限制。理论研究表明,学习率的选择与算法设计同等重要,对统计与计算效率均有关键影响。
原文摘要 · Abstract (English)
To understand feature learning dynamics in neural networks, recent theoretical works have focused on gradient-based learning of Gaussian single-index models, where the label is a nonlinear function of a latent one-dimensional projection of the input. While the sample complexity of online SGD is determined by the information exponent of the link function, recent works improved this by performing multiple gradient steps on the same sample with different learning rates -- yielding a non-correlational update rule -- and instead are limited by the (potentially much smaller) generative exponent. However, this picture is only valid when these learning rates are sufficiently large. In this paper, we characterize the relationship between learning rate(s) and sample complexity for a broad class of gradient-based algorithms that encapsulates both correlational and non-correlational updates. We demonstrate that, in certain cases, there is a phase transition from an "information exponent regime" with small learning rate to a "generative exponent regime" with large learning rate. Our framework covers prior analyses of one-pass SGD and SGD with batch reuse, while also introducing a new layer-wise training algorithm that leverages a two-timescales approach (via different learning rates for each layer) to go beyond correlational queries without reusing samples or modifying the loss from squared error. Our theoretical study demonstrates that the choice of learning rate is as important as the design of the algorithm in achieving statistical and computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。