arXiv:2606.02078cs.LG2026-06

提出动态p范数优化器,自动适应模型训练中的曲率变化。

Beyond $\ell_2$-norm and $\ell_\infty$-norm: A Curvature-Inspired $\ell_p$-Norm Scheme for Deep Neural Networks

  • 用可变p值的范数替代固定l2/l∞,随训练阶段动态调整
  • 在早期抑制高曲率方向,在后期平稳更新,收敛更快
  • 适合对收敛速度和泛化性能有要求的深度网络训练

现有深度神经网络优化器多依赖l2或l∞范数,难以适应参数维度间显著的曲率变化。训练初期曲率各向异性强,后期趋向平坦区域。l2范数优化器受高曲率方向主导,限制低曲率方向更新,导致收敛慢;l∞范数则因各坐标同幅更新,在平坦区易产生振荡。为此,本文提出一种动态p范数方案,融入SGD与SGDM,形成LPSGD与LPSGDM两个新优化器。通过大p(p>2)抑制早期高曲率方向,再按余弦退火策略逐步减小p至2,实现稳定精细更新。理论分析表明,两者在非凸设置下均达到O(T⁻¹/²)收敛率。在CIFAR-10、CIFAR-100和ImageNet-1K上,使用VGG-11、ResNet-18、ResNet-50等模型的大量实验验证了其更优的泛化性能。

原文摘要 · Abstract (English)

The existing optimizers for deep neural networks (DNNs) typically rely on either the $\ell_2$ norm or the $\ell_\infty$ norm, resulting in optimizers that do not adapt well to substantial changes in curvature across parameter dimensions. Generally, the training process of DNNs often exhibits strong curvature anisotropy in the early period, whereas in the later period, the training process of DNNs tends to move toward flatter regions with weaker anisotropy. Particularly, optimizers based on the \(\ell_2\)-norm are usually dominated by high-curvature directions, restricting updates of optimizers along with lower curvature direction and thus leading to a slower convergence rate. While optimizers based on the \(\ell_\infty\)-norm are prone to oscillations in flatter regions, due to the coordinate-wise updates of the same magnitude. To address these two extreme cases generated by $\ell_2$ and $\ell_\infty$ norms, we propose a novel $\ell_p$-norm scheme with a dynamical value of $p$ and incorporate it into stochastic gradient descent (SGD) and SGD with momentum (SGDM), leading to two novel optimizers with better generalization performance: ${\ell_p}$-SGD (LPSGD) and ${\ell_p}$-SGDM (LPSGDM). Particularly, the resulting optimizers suppress the dominance of high-curvature directions in the early period by utilizing a large $p$ ($p>2$), followed by a gradual decrease of $p$ toward 2 to enable more stable and refined updates, where the latter process is motivated by the cosine annealing strategy. We establish theoretical guarantees of the resulting algorithms and analyze that both LPSGD and LPSGDM achieve an \(O(T^{-1/2})\) convergence rate for the nonconvex setting. Extensive experiments are conducted on benchmark datasets, including CIFAR-10, CIFAR-100, and ImageNet-1K, with multiple DNNs such as VGG-11, ResNet-18, and ResNet-50.

优化器深度学习收敛性动态范数

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