mlsanityData to 7 Oct 2026

Learning Provably Improves the Convergence of Gradient Descent

arXiv 2501.18092 ↗First posted 30 Jan 2025cs.LG math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

Learn to Optimize (L2O) trains deep neural network-based solvers for optimization, achieving success in accelerating convex problems and improving non-convex solutions. However, L2O lacks rigorous theoretical backing for its own training convergence, as existing analyses often use unrealistic assumptions -- a gap this work highlights empirically. We bridge this gap by proving the training convergence of L2O models that learn Gradient Descent (GD) hyperparameters for quadratic programming, leveraging the Neural Tangent Kernel (NTK) theory. We propose a deterministic initialization strategy to support our theoretical results and promote stable training over extended optimization horizons by mitigating gradient explosion. Our L2O framework demonstrates over 50% better optimality than GD and superior robustness over state-of-the-art L2O methods on synthetic datasets. The code of our method can be found from https://github.com/NetX-lab/MathL2OProof-Official.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 41 references that are in the map. First posted on arXiv 30 Jan 2025.

Builds on

21 of 41 references.

References · 21

  1. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  2. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
  3. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
  4. 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
  5. 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 231
  6. 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
  7. 2017Learning to Optimize: Training Deep Neural Networks for Wireless Resource Managementcited by 96
  8. 2018Theoretical Linear Convergence of Unfolded ISTA and its Practical Weights and Thresholdscited by 83
  9. 2018On the Convergence Rate of Training Recurrent Neural Networkscited by 82
  10. 2020Graph Neural Networks for Scalable Radio Resource Management: Architecture Design and Theoretical Analysiscited by 80
  11. 2020Global Convergence of Deep Networks with One Wide Layer Followed by Pyramidal Topologycited by 66
  12. 2017Learning Gradient Descent: Better Generalization and Longer Horizonscited by 50
9 more
  1. 2021On the Proof of Global Convergence of Gradient Descent for Deep ReLU Networks with Linear Widthscited by 44
  2. 2020Iterative Algorithm Induced Deep-Unfolding Neural Networks: Precoding Design for Multiuser MIMO Systemscited by 43
  3. 2020Safeguarded Learned Convex Optimizationcited by 24
  4. 2021Hyperparameter Tuning is All You Need for LISTAcited by 22
  5. 2023Towards Constituting Mathematical Structures for Learning to Optimizecited by 13
  6. 2024Convex Relaxations of ReLU Neural Networks Approximate Global Optima in Polynomial Timecited by 11
  7. 2024Exploring the loss landscape of regularized neural networks via convex dualitycited by 11
  8. 2023Learning to Generalize Provably in Learning to Optimizecited by 8
  9. 2021Provable Convergence of Nesterov's Accelerated Gradient Method for Over-Parameterized Neural Networkscited by 7

Built on by

Not cited yet.