中文

超宽神经网络的非线性随机矩阵变形半圆律与集中性

统计理论 2023-04-17 v3 机器学习 概率论 机器学习 统计理论

摘要

本文研究形式为f(X)=1d1aσ(WX)f(X)=\frac{1}{\sqrt{d_1}}\boldsymbol{a}^\top \sigma\left(WX\right)的两层全连接神经网络,其中XRd0×nX\in\mathbb{R}^{d_0\times n}为确定性数据矩阵,WRd1×d0W\in\mathbb{R}^{d_1\times d_0}aRd1\boldsymbol{a}\in\mathbb{R}^{d_1}为随机高斯权重,σ\sigma为非线性激活函数。我们研究对应于f(X)f(X)的两个经验核矩阵:经验共轭核(CK)与神经正切核(NTK)的极限谱分布,超越线性宽度 regime(d1nd_1\asymp n)。我们聚焦于超宽 regime\textit{超宽 regime},即第一层宽度d1d_1远大于样本量nn。在对XXσ\sigma的适当假设下,当d1/nd_1/n\to\inftynn\to\infty时,出现变形半圆律。我们首先证明具有某种依赖性的广义样本协方差矩阵的该极限律。为在我们的神经网络模型中具体指明它,我们提供一个适用于具有随机权重和Lipschitz激活函数的神经网络的非线性Hanson-Wright不等式。我们还展示了经验CK与NTK在谱范数下围绕其极限核的非渐近集中性,以及它们最小本征值的下界。作为一个应用,我们表明由经验核诱导的随机特征回归在超宽 regime 下达到与其极限核回归相同的渐近性能。这使我们能够使用相应核回归计算随机特征回归的渐近训练与测试误差。

关键词

引用

@article{arxiv.2109.09304,
  title  = {Deformed semicircle law and concentration of nonlinear random matrices for ultra-wide neural networks},
  author = {Zhichao Wang and Yizhe Zhu},
  journal= {arXiv preprint arXiv:2109.09304},
  year   = {2023}
}

备注

49 pages, 5 figures