arXiv · 2403.08121
Early Directional Convergence in Deep Homogeneous Neural Networks for Small Initializations
Abstract
This paper studies the gradient flow dynamics that arise when training deep homogeneous neural networks assumed to have locally Lipschitz gradients and an order of homogeneity strictly greater than two. It is shown here that for sufficiently small initializations, during the early stages of training, the weights of the neural network remain small in (Euclidean) norm and approximately converge in direction to the Karush-Kuhn-Tucker (KKT) points of the recently introduced neural correlation function. Additionally, this paper also studies the KKT points of the neural correlation function for feed-forward networks with (Leaky) ReLU and polynomial (Leaky) ReLU activations, deriving necessary and sufficient conditions for rank-one KKT points.
Explore related subjects
Keep this discovery
Akshay Kumar, Jarvis Haupt. 2024-03-12. Early Directional Convergence in Deep Homogeneous Neural Networks for Small Initializations. https://arxiv.org/abs/2403.08121
Cite the original work for its findings. Save a collection to share your selection of sources.