Lune

NeurIPS2025Top-tier venue

Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classification

Yuanfan Li, Yunwen Lei, Zheng-Chu Guo, Yiming Ying

2025Year
4Citations
1Top-tier citations

Abstract

Recent advances have significantly improved our understanding of the generalization performance of gradient descent (GD) methods in deep neural networks. A natural and fundamental question is whether GD can achieve generalization rates comparable to the minimax optimal rates established in the kernel setting. Existing results either yield suboptimal rates of O(1/n)O(1/\sqrt{n}), or focus on networks with smooth activation functions, incurring exponential dependence on network depth LL. In this work, we establish optimal generalization rates for GD with deep ReLU networks by carefully trading off optimization and generalization errors, achieving only polynomial dependence on depth. Specifically, under the assumption that the data are NTK separable from the margin γ\gamma, we prove an excess risk rate of O~(L6/(nγ2))\widetilde{O}(L^6 / (n \gamma^2)), which aligns with the optimal SVM-type rate O~(1/(nγ2))\widetilde{O}(1 / (n \gamma^2)) up to depth-dependent factors. A key technical contribution is our novel control of activation patterns near a reference model, enabling a sharper Rademacher complexity bound for deep ReLU networks trained with gradient descent.

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.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 5a0f649e-591a-4196-bb1a-44c4a5f5a072

Cited by top-tier papers1

Ask how each one uses it

Builds on9

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines