Large deviations of one-hidden-layer neural networks
We study large deviations in the context of stochastic gradient descent for one-hidden-layer neural networks with quadratic loss. We derive a quenched large deviation principle, where we condition on an initial weight measure, and an annealed large deviation principle for the empirical weight evolution during training when letting the number of neurons and the number of training iterations simultaneously tend to infinity. The weight evolution is treated as an interacting dynamic particle system. The distinctive aspect compared to prior work on interacting particle systems lies in the discrete particle updates, simultaneously with a growing number of particles.
Authors
- Daniel Willhalm (ORCID: https://orcid.org/0000-0001-8869-7460)
- Christian Hirsch (ORCID: https://orcid.org/0000-0003-4136-3740)
Institutions
- University of Groningen (NL)
- Aarhus University (DK)
- Toronto Zoo (CA)
Publication Details
- Journal
- Stochastics and Dynamics
- Published
- 2024-12-01
- DOI
- https://doi.org/10.1142/s0219493725500029
- Citations
- 4
- Primary Topic
- Neural Networks and Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.39
Funders
- Villum Fonden