English

Spectral Analysis of the Neural Tangent Kernel for Deep Residual Networks

Machine Learning 2021-04-08 v1

Abstract

Deep residual network architectures have been shown to achieve superior accuracy over classical feed-forward networks, yet their success is still not fully understood. Focusing on massively over-parameterized, fully connected residual networks with ReLU activation through their respective neural tangent kernels (ResNTK), we provide here a spectral analysis of these kernels. Specifically, we show that, much like NTK for fully connected networks (FC-NTK), for input distributed uniformly on the hypersphere Sd1\mathbb{S}^{d-1}, the eigenfunctions of ResNTK are the spherical harmonics and the eigenvalues decay polynomially with frequency kk as kdk^{-d}. These in turn imply that the set of functions in their Reproducing Kernel Hilbert Space are identical to those of FC-NTK, and consequently also to those of the Laplace kernel. We further show, by drawing on the analogy to the Laplace kernel, that depending on the choice of a hyper-parameter that balances between the skip and residual connections ResNTK can either become spiky with depth, as with FC-NTK, or maintain a stable shape.

Keywords

Cite

@article{arxiv.2104.03093,
  title  = {Spectral Analysis of the Neural Tangent Kernel for Deep Residual Networks},
  author = {Yuval Belfer and Amnon Geifman and Meirav Galun and Ronen Basri},
  journal= {arXiv preprint arXiv:2104.03093},
  year   = {2021}
}