Generalization Guarantees of Gradient Descent for Multi-Layer Neural Networks

Puyu Wang, Yunwen Lei, Di Wang +2

Recently, significant progress has been made in understanding the generalization of neural networks (NNs) trained by gradient descent (GD) using the algorithmic stability approach.…