梯度下降训练宽神经网络,可达到最优非参数回归误差。
Gradient Descent Finds Over-Parameterized Neural Networks with Sharp Generalization for Nonparametric Regression
- 用早停梯度下降训练宽两层网络
- 误差率达最优的 $\mathcal{O}(ε_n^2)$,无需分布假设
- 适用于无特定数据分布限制的回归任务
本文研究通过梯度下降(GD)训练的过参数化两层神经网络在非参数回归中的表现。若采用早停策略,训练后的网络能达到非参数回归风险的尖锐率 $\mathcal{O}(ε_n^2)$,该速率与经典核回归在相同条件下由梯度下降训练所得一致,其中 $ε_n$ 是与网络相关的神经正切核(NTK)的临界总体速率,$n$ 为训练数据量。结果不依赖于协变量的具体分布,仅需其有界,突破了以往研究对球面均匀分布或特定条件分布的依赖。当 NTK 满足多项式特征值衰减时,该速率已被证明为极小极大最优。本工作正式弥合了经典核回归与有限宽度过参数化神经网络在无分布假设下通过梯度下降进行非参数回归的差距。同时,还回答了文献中关于停止时间、网络宽度下限及恒定学习率使用的若干开放问题。
原文摘要 · Abstract (English)
We study nonparametric regression by an over-parameterized two-layer neural network trained by gradient descent (GD) in this paper. We show that, if the neural network is trained by GD with early stopping, then the trained network renders a sharp rate of the nonparametric regression risk of $\mathcal{O}(ε_n^2)$, which is the same rate as that for the classical kernel regression trained by GD with early stopping, where $ε_n$ is the critical population rate of the Neural Tangent Kernel (NTK) associated with the network and $n$ is the size of the training data. It is remarked that our result does not require distributional assumptions about the covariate as long as the covariate is bounded, in a strong contrast with many existing results which rely on specific distributions of the covariates such as the spherical uniform data distribution or distributions satisfying certain restrictive conditions. The rate $\mathcal{O}(ε_n^2)$ is known to be minimax optimal for specific cases, such as the case that the NTK has a polynomial eigenvalue decay rate which happens under certain distributional assumptions on the covariates. Our result formally fills the gap between training a classical kernel regression model and training an over-parameterized but finite-width neural network by GD for nonparametric regression without distributional assumptions on the bounded covariate. We also provide confirmative answers to certain open questions or address particular concerns in the literature of training over-parameterized neural networks by GD with early stopping for nonparametric regression, including the characterization of the stopping time, the lower bound for the network width, and the constant learning rate used in GD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。