Wide stable neural networks: Sample regularity, functional convergence and Bayesian inverse problems
Abstract
We study the large-width asymptotics of random fully connected neural networks with weights drawn from $\alpha$-stable distributions, a family of heavy-tailed distributions arising as the limiting distributions in the Gnedenko-Kolmogorov heavy-tailed central limit theorem. We show that in an arbitrary bounded Euclidean domain $\mathcal{U}$ with smooth boundary, the random field at the infinite-width limit, characterized in previous literature in terms of finite-dimensional distributions, has sample functions in the fractional Sobolev-Slobodeckij-type quasi-Banach function space $W^{s,p}(\mathcal{U})$ for integrability indices $p < \alpha$ and suitable smoothness indices $s$ depending on the activation function of the neural network, and establish the functional convergence of the processes in $\mathcal{P}(W^{s,p}(\mathcal{U}))$. This convergence result is leveraged in the study of functional posteriors for edge-preserving Bayesian inverse problems with stable neural network priors.
- Publication:
-
arXiv e-prints
- Pub Date:
- July 2024
- DOI:
- 10.48550/arXiv.2407.03909
- arXiv:
- arXiv:2407.03909
- Bibcode:
- 2024arXiv240703909S
- Keywords:
-
- Mathematics - Statistics Theory;
- Mathematics - Probability;
- Primary: 68T07;
- 62F15;
- 60G52;
- 60G17;
- Secondary: 46E35
- E-Print:
- 26 pages, 2 figures