An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Mao, Jialin, Griniasty, Itay, Sun, Yan, Transtrum, Mark K., Sethna, James P., Chaudhari, Pratik
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914162496503808
author Mao, Jialin
Griniasty, Itay
Sun, Yan
Transtrum, Mark K.
Sethna, James P.
Chaudhari, Pratik
author_facet Mao, Jialin
Griniasty, Itay
Sun, Yan
Transtrum, Mark K.
Sethna, James P.
Chaudhari, Pratik
contents Recent experiments have shown that training trajectories of multiple deep neural networks with different architectures, optimization algorithms, hyper-parameter settings, and regularization methods evolve on a remarkably low-dimensional "hyper-ribbon-like" manifold in the space of probability distributions. Inspired by the similarities in the training trajectories of deep networks and linear networks, we analytically characterize this phenomenon for the latter. We show, using tools in dynamical systems theory, that the geometry of this low-dimensional manifold is controlled by (i) the decay rate of the eigenvalues of the input correlation matrix of the training data, (ii) the relative scale of the ground-truth output to the weights at the beginning of training, and (iii) the number of steps of gradient descent. By analytically computing and bounding the contributions of these quantities, we characterize phase boundaries of the region where hyper-ribbons are to be expected. We also extend our analysis to kernel machines and linear models that are trained with stochastic gradient descent.
format Preprint
id arxiv_https___arxiv_org_abs_2505_08915
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Models
Mao, Jialin
Griniasty, Itay
Sun, Yan
Transtrum, Mark K.
Sethna, James P.
Chaudhari, Pratik
Machine Learning
Disordered Systems and Neural Networks
Statistical Mechanics
Recent experiments have shown that training trajectories of multiple deep neural networks with different architectures, optimization algorithms, hyper-parameter settings, and regularization methods evolve on a remarkably low-dimensional "hyper-ribbon-like" manifold in the space of probability distributions. Inspired by the similarities in the training trajectories of deep networks and linear networks, we analytically characterize this phenomenon for the latter. We show, using tools in dynamical systems theory, that the geometry of this low-dimensional manifold is controlled by (i) the decay rate of the eigenvalues of the input correlation matrix of the training data, (ii) the relative scale of the ground-truth output to the weights at the beginning of training, and (iii) the number of steps of gradient descent. By analytically computing and bounding the contributions of these quantities, we characterize phase boundaries of the region where hyper-ribbons are to be expected. We also extend our analysis to kernel machines and linear models that are trained with stochastic gradient descent.
title An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Models
topic Machine Learning
Disordered Systems and Neural Networks
Statistical Mechanics
url https://arxiv.org/abs/2505.08915