arXiv:2510.24466math.OCcs.LG2025-10NeurIPS被引 1

证明了神经网络梯度下降在大多数步长下不会陷入鞍点,可直接用于实际模型训练。

Non-Singularity of the Gradient Descent map for Neural Networks with Piecewise Analytic Activations

  • 基于分段解析激活函数,首次证明梯度下降映射非奇异
  • 适用于全连接、卷积和注意力层的主流网络结构
  • 为实际训练中的收敛性分析提供理论支持

深度网络训练理论已成为现代机器学习的核心问题,并推动了诸多实际进展。其中,梯度下降(GD)算法近年来受到广泛研究。一个关键假设是: GD 映射是非奇异的——即在前像下保持零测集。这一假设被用于证明 GD 能避开鞍点与极大值,并建立决定全局最小值收敛性的可计算量(对 GD 与随机梯度下降均适用)。然而,现有文献或假设该性质成立,或施加严格限制,如损失函数的 Lipschitz 光滑性(例如,深层 ReLU 网络在交叉熵损失下不满足),且分析局限于小步长情形。本文研究权重与偏置空间上的神经网络映射,首次证明在具有全连接、卷积或 softmax 注意力层的真实网络架构下,使用分段解析激活函数(包括 sigmoid、ReLU、leaky ReLU 等)时,梯度下降映射对几乎所有步长均为非奇异。该工作显著扩展了现有关于 GD 与 SGD 收敛性的结果,确保其适用于实际神经网络场景,并有望促进学习动态的进一步探索。

原文摘要 · Abstract (English)

The theory of training deep networks has become a central question of modern machine learning and has inspired many practical advancements. In particular, the gradient descent (GD) optimization algorithm has been extensively studied in recent years. A key assumption about GD has appeared in several recent works: the \emph{GD map is non-singular} -- it preserves sets of measure zero under preimages. Crucially, this assumption has been used to prove that GD avoids saddle points and maxima, and to establish the existence of a computable quantity that determines the convergence to global minima (both for GD and stochastic GD). However, the current literature either assumes the non-singularity of the GD map or imposes restrictive assumptions, such as Lipschitz smoothness of the loss (for example, Lipschitzness does not hold for deep ReLU networks with the cross-entropy loss) and restricts the analysis to GD with small step-sizes. In this paper, we investigate the neural network map as a function on the space of weights and biases. We also prove, for the first time, the non-singularity of the gradient descent (GD) map on the loss landscape of realistic neural network architectures (with fully connected, convolutional, or softmax attention layers) and piecewise analytic activations (which includes sigmoid, ReLU, leaky ReLU, etc.) for almost all step-sizes. Our work significantly extends the existing results on the convergence of GD and SGD by guaranteeing that they apply to practical neural network settings and has the potential to unlock further exploration of learning dynamics.

优化理论梯度下降神经网络

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