arXiv:2502.03792stat.MLcs.LG2025-02

用衰减学习率训练两层神经网络,可让模型更稳定且泛化性更强。

Step by Step: Adaptive Gradient Descent for Training L-Lipschitz Neural Networks

  • 采用渐进衰减的学习率,提升模型的Lipschitz正则性。
  • 收敛速度不受影响,仍能快速逼近最优解。
  • 适合关注模型稳定性与泛化能力的研究者。

我们证明,在经验风险最小化中,对两层具有Lipschitz激活函数的神经网络使用标准梯度下降(GD)训练时,若引入学习率的最终衰减,可使所得网络表现出高度Lipschitz正则性,即较小的Lipschitz常数。此外,该衰减不影响以Huber损失衡量的经验风险收敛至非凸目标函数临界点的速率。基于此,我们推导出使用衰减学习率的两层神经网络的泛化界,其依赖于可训练参数数量的次线性关系,表明这类网络的统计行为与过参数化无关。通过一系列简单数值实验验证了理论结果,令人惊讶的是,使用固定步长的GD训练的网络也展现出与衰减学习率相似的学习与正则性特性,暗示标准GD训练的神经网络可能已具备高度正则性。

原文摘要 · Abstract (English)

We demonstrate that applying an eventual decay to the learning rate (LR) in empirical risk minimization (ERM), where the mean-squared-error loss is minimized using standard gradient descent (GD) for training a two-layer neural network with Lipschitz activation functions, ensures that the resulting network exhibits a high degree of Lipschitz regularity, that is, a small Lipschitz constant. Moreover, we show that this decay does not hinder the convergence rate of the empirical risk, now measured with the Huber loss, toward a critical point of the non-convex empirical risk. From these findings, we derive generalization bounds for two-layer neural networks trained with GD and a decaying LR with a sub-linear dependence on its number of trainable parameters, suggesting that the statistical behaviour of these networks is independent of overparameterization. We validate our theoretical results with a series of toy numerical experiments, where surprisingly, we observe that networks trained with constant step size GD exhibit similar learning and regularity properties to those trained with a decaying LR. This suggests that neural networks trained with standard GD may already be highly regular learners.

梯度下降Lipschitz泛化性正则化

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