Geometric Convergence of Gauss–Newton for Neural Networks: Riemannian Geometry and Adaptive Damping
Semih Cayci
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f9e851b9-24fc-4f9e-9ca3-c027bf7d8323Builds on7
- Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networksZiwei Ji, Matus TelgarskyICLR 2020 · 193 citations
- Challenges in Training PINNs: A Loss Landscape PerspectivePratik Rathore, Weimu Lei, Zachary Frangella, Lu Lu et al.ICML 2024 · 137 citations
- The Potential of Second-Order Optimization for LLMs: A Study with Full Gauss-NewtonNatalie Abreu, Nikhil Vyas, Sham M. Kakade, Depen MorwaniICLR 2026 · 21 citations
- Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimensionKedar Karhadkar, Michael Murray, Guido F. MontúfarNeurIPS 2024 · 15 citations
- Rethinking Gauss-Newton for learning over-parameterized modelsMichael Arbel, Romain Menegaux, Pierre WolinskiNeurIPS 2023 · 10 citations
Related papers
- 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 citations
- Convergence Rates for Gradient Descent on the Edge of Stability for Overparametrised Least SquaresLachlan E. MacDonald, Hancheng Min, Leandro Palma, Salma Tarmoun et al.NeurIPS 2025 · 5 citations
- Riemannian SAM: Sharpness-Aware Minimization on Riemannian ManifoldsJihun Yun, Eunho YangNeurIPS 2023 · 10 citations
- A Non-Parametric Regression Viewpoint : Generalization of Overparametrized Deep RELU Network Under Noisy ObservationsNamjoon Suh, Hyunouk Ko, Xiaoming HuoICLR 2022 · 15 citations
- Leveraging Non-uniformity in First-order Non-convex OptimizationJincheng Mei, Yue Gao, Bo Dai, Csaba Szepesvári et al.ICML 2021 · 55 citations
