Geometric Convergence of Gauss–Newton for Neural Networks: Riemannian Geometry and Adaptive Damping
Semih Cayci
摘要
Ill-conditioned kernel matrices can make first-order methods for training neural networks converge slowly. We establish non-asymptotic convergence bounds for the Gauss–Newton method in both under- and overparameterized regimes, showing it avoids these conditioning bottlenecks. In the underparameterized setting, Gauss–Newton gradient flow in parameter space induces a Riemannian gradient flow on a low-dimensional submanifold of function space. Using tools from Riemannian optimization, we show that, under an appropriate output scaling, the loss satisfies geodesic Polyak–Lojasiewicz and Lipschitz-smoothness conditions, implying geometric convergence to the optimal in-class predictor at an explicit rate independent of Gram-matrix conditioning. In the overparameterized setting, we identify adaptive, curvature-aware regularization schedules and prove fast geometric convergence to a global optimum for both Gauss–Newton gradient flow and discrete-time Gauss–Newton iterates, with rates independent of the minimum eigenvalue of the neural tangent kernel. Overall, Gauss–Newton can be provably faster in ill-conditioned regimes where first-order methods slow down.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper7
- Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networksZiwei Ji, Matus TelgarskyICLR 2020 · 被引用 193 次
- Challenges in Training PINNs: A Loss Landscape PerspectivePratik Rathore, Weimu Lei, Zachary Frangella, Lu Lu 等ICML 2024 · 被引用 137 次
- The Potential of Second-Order Optimization for LLMs: A Study with Full Gauss-NewtonNatalie Abreu, Nikhil Vyas, Sham M. Kakade, Depen MorwaniICLR 2026 · 被引用 21 次
- Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimensionKedar Karhadkar, Michael Murray, Guido F. MontúfarNeurIPS 2024 · 被引用 15 次
- Rethinking Gauss-Newton for learning over-parameterized modelsMichael Arbel, Romain Menegaux, Pierre WolinskiNeurIPS 2023 · 被引用 10 次
相关 Paper
- Convergence Rates of Non-Convex Stochastic Gradient Descent Under a Generic Lojasiewicz Condition and Local SmoothnessKevin Scaman, Cédric Malherbe, Ludovic Dos SantosICML 2022 · 被引用 24 次
- Convergence Rates for Gradient Descent on the Edge of Stability for Overparametrised Least SquaresLachlan E. MacDonald, Hancheng Min, Leandro Palma, Salma Tarmoun 等NeurIPS 2025 · 被引用 5 次
- Riemannian SAM: Sharpness-Aware Minimization on Riemannian ManifoldsJihun Yun, Eunho YangNeurIPS 2023 · 被引用 10 次
- A Non-Parametric Regression Viewpoint : Generalization of Overparametrized Deep RELU Network Under Noisy ObservationsNamjoon Suh, Hyunouk Ko, Xiaoming HuoICLR 2022 · 被引用 15 次
- Leveraging Non-uniformity in First-order Non-convex OptimizationJincheng Mei, Yue Gao, Bo Dai, Csaba Szepesvári 等ICML 2021 · 被引用 55 次
