English

Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimension

Machine Learning 2024-05-24 v1 Machine Learning

Abstract

Bounds on the smallest eigenvalue of the neural tangent kernel (NTK) are a key ingredient in the analysis of neural network optimization and memorization. However, existing results require distributional assumptions on the data and are limited to a high-dimensional setting, where the input dimension d0d_0 scales at least logarithmically in the number of samples nn. In this work we remove both of these requirements and instead provide bounds in terms of a measure of the collinearity of the data: notably these bounds hold with high probability even when d0d_0 is held constant versus nn. We prove our results through a novel application of the hemisphere transform.

Keywords

Cite

@article{arxiv.2405.14630,
  title  = {Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimension},
  author = {Kedar Karhadkar and Michael Murray and Guido Montúfar},
  journal= {arXiv preprint arXiv:2405.14630},
  year   = {2024}
}

Comments

47 pages