English

Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networks

Machine Learning 2022-09-21 v1 Machine Learning

Abstract

While significant theoretical progress has been achieved, unveiling the generalization mystery of overparameterized neural networks still remains largely elusive. In this paper, we study the generalization behavior of shallow neural networks (SNNs) by leveraging the concept of algorithmic stability. We consider gradient descent (GD) and stochastic gradient descent (SGD) to train SNNs, for both of which we develop consistent excess risk bounds by balancing the optimization and generalization via early-stopping. As compared to existing analysis on GD, our new analysis requires a relaxed overparameterization assumption and also applies to SGD. The key for the improvement is a better estimation of the smallest eigenvalues of the Hessian matrices of the empirical risks and the loss function along the trajectories of GD and SGD by providing a refined estimation of their iterates.

Keywords

Cite

@article{arxiv.2209.09298,
  title  = {Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networks},
  author = {Yunwen Lei and Rong Jin and Yiming Ying},
  journal= {arXiv preprint arXiv:2209.09298},
  year   = {2022}
}

Comments

to appear in Neural Information Processing Systems (NeurIPS 2022)

R2 v1 2026-06-28T01:41:23.522Z