mlsanityData to 7 Oct 2026

Generalization Analysis of Online Stochastic Gradient Descent for Overparameterized Two-Layer Neural Networks

arXiv 2407.07670 ↗First posted 10 Jul 2024stat.ML cs.LGcited by 4

Not on the map. Why?

Abstract (arXiv)

We investigate the generalization performance of online stochastic gradient descent (SGD) for overparameterized two-layer neural networks under the Neural Tangent Kernel (NTK) regime. Leveraging the NTK approximation together with the convergence theory of stochastic approximation in reproducing kernel Hilbert spaces (RKHSs), we develop a unified framework that connects the optimization dynamics of neural networks with kernel-based statistical learning. This framework enables us to derive sharp convergence rates for the generalization error of the last iterate of online SGD. The obtained rates coincide with the optimal statistical rates for kernel methods under appropriate source assumptions on target function. In contrast to existing NTK analyses, which primarily establish optimization convergence or analyze averaged SGD, our results directly characterize the statistical behavior of the practically implemented last-iterate online SGD algorithm for streaming data. Moreover, our analysis substantially relaxes the required degree of overparameterization by reducing the network-width requirement from exponential to polynomial dependence on the sample size or the number of optimization iterations. These results provide a unified theoretical perspective on stochastic optimization, kernel methods, and statistical learning for overparameterized neural networks, while significantly narrowing the gap between existing theory and practical deep learning.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 20 of its 38 references that are in the map. First posted on arXiv 10 Jul 2024.

Builds on

20 of 38 references.

References · 20

  1. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
  2. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
  3. 2016Error bounds for approximations with deep ReLU networkscited by 567
  4. 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
  5. 2018Learning Overparameterized Neural Networks via Stochastic Gradient Descent on Structured Datacited by 260
  6. 2018Universality of Deep Convolutional Neural Networkscited by 176
  7. 2017Convergence Analysis of Two-layer Neural Networks with ReLU Activationcited by 137
  8. 2018On the Convergence Rate of Training Recurrent Neural Networkscited by 82
  9. 2017SGD Learns Over-parameterized Networks that Provably Generalize on Linearly Separable Datacited by 79
  10. 2017Learning One-hidden-layer Neural Networks with Landscape Designcited by 79
  11. 2023A Theory of Non-Linear Feature Learning with One Gradient Step in Two-Layer Neural Networkscited by 46
  12. 2020Optimal Rates for Averaged Stochastic Gradient Descent under Neural Tangent Kernel Regimecited by 35
8 more
  1. 2018Learning with SGD and Random Featurescited by 33
  2. 2019On Learning Over-parameterized Neural Networks: A Functional Approximation Perspectivecited by 28
  3. 2021Universal Consistency of Deep Convolutional Neural Networkscited by 16
  4. 2022Capacity dependent analysis for functional online learning algorithmscited by 10
  5. 2023How many Neurons do we need? A refined Analysis for Shallow Networks trained with Gradient Descentcited by 10
  6. 2017Fast and Strong Convergence of Online Learning Algorithmscited by 7
  7. 2023Optimality of Robust Online Learningcited by 5
  8. 2017Convergence of Unregularized Online Learning Algorithmscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 2; 2026: 2.

4 papers.