Stochastic optimization on matrices and a graphon McKean-Vlasov limit
Not on the map. Why?
Abstract (arXiv)
We consider stochastic gradient descents on the space of large symmetric matrices of suitable functions that are invariant under permuting the rows and columns using the same permutation. We establish deterministic limits of these random curves as the dimensions of the matrices go to infinity while the entries remain bounded. Under a ``small noise'' assumption the limit is shown to be the gradient flow of functions on graphons whose existence was established in Oh, Somani, Pal, and Tripathi, texitJ Theor Probab 37, 1469--1522 (2024). We also consider limits of stochastic gradient descents with added properly scaled reflected Brownian noise. The limiting curve of graphons is characterized by a family of stochastic differential equations with reflections and can be thought of as an extension of the classical McKean-Vlasov limit for interacting diffusions to the graphon setting. The proofs introduce a family of infinite-dimensional exchangeable arrays of reflected diffusions and a novel notion of propagation of chaos for large matrices of diffusions converging to such arrays in a suitable sense.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 9 of its 66 references that are in the map. First posted on arXiv 2 Oct 2022.
Builds on
9 of 66 references.
References · 9
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2014Convex Optimization: Algorithms and Complexitycited by 569
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
- 2018Stochastic Modified Equations and Dynamics of Stochastic Gradient Algorithms I: Mathematical Foundationscited by 109
- 2022On the SDEs and Scaling Rules for Adaptive Gradient Algorithmscited by 100
- 2022Random initialisations performing above chance and how to find themcited by 23
- 2021Gradient flows on graphons: existence, convergence, continuity equationscited by 3
Built on by
Citing papers per year: 2023: 1.
1 paper.
- 2023Path convergence of Markov chains on large graphscited by 0
- 2023Path convergence of Markov chains on large graphs18 Aug 2023