Gaussian Universality in Neural Network Dynamics with Generalized Structured Input Distributions
Abstract
Analyzing neural network dynamics via stochastic gradient descent (SGD) is crucial to building theoretical foundations for deep learning. Previous work has analyzed structured inputs within the \textit{hidden manifold model}, often under the simplifying assumption of a Gaussian distribution. We extend this framework by modeling inputs as Gaussian mixtures to better represent complex, real-world data. Through empirical and theoretical investigation, we demonstrate that with proper standardization, the learning dynamics converges to the behavior seen in the simple Gaussian case. This finding exhibits a form of universality, where diverse structured distributions yield results consistent with Gaussian assumptions, thereby strengthening the theoretical understanding of deep learning models.
Cite
@article{arxiv.2405.00642,
title = {Gaussian Universality in Neural Network Dynamics with Generalized Structured Input Distributions},
author = {Jaeyong Bae and Hawoong Jeong},
journal= {arXiv preprint arXiv:2405.00642},
year = {2025}
}
Comments
Accepted for Bridging the Gap Between Practice and Theory in Deep Learning (BGPT) Workshop at ICLR 2024, [v1] 23 pages, 16 figures