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 scales at least logarithmically in the number of samples . 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 is held constant versus . We prove our results through a novel application of the hemisphere transform.
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