Normal approximation of Random Gaussian Neural Networks

Nicola Apollonio,Daniela De Canditiis,Giovanni Franzina,Paola Stolfi,Giovanni Luca Torrisi
2023-09-20
Abstract:In this paper we provide explicit upper bounds on some distances between the (law of the) output of a random Gaussian NN and (the law of) a random Gaussian vector. Our results concern both shallow random Gaussian neural networks with univariate output and fully connected and deep random Gaussian neural networks, with a rather general activation function. The upper bounds show how the widths of the layers, the activation functions and other architecture parameters affect the Gaussian approximation of the ouput. Our techniques, relying on Stein's method and integration by parts formulas for the Gaussian law, yield estimates on distances which are indeed integral probability metrics, and include the total variation and the convex distances. These latter metrics are defined by testing against indicator functions of suitable measurable sets, and so allow for accurate estimates of the probability that the output is localized in some region of the space. Such estimates have a significant interest both from a practitioner's and a theorist's perspective.
Probability,Analysis of PDEs
What problem does this paper attempt to address?
The paper attempts to address the problem of Gaussian approximation for the outputs of Random Gaussian Neural Networks. Specifically, it provides explicit upper bounds on the distance between the outputs of Random Gaussian Neural Networks and random Gaussian vectors, and explores how these bounds are influenced by network layer width, activation functions, and other architectural parameters. The main contributions of the paper include: 1. Providing explicit upper bounds for Gaussian approximation in terms of convex distance and 1-Wasserstein distance for deep Random Gaussian Neural Networks. 2. Addressing these problems even when the activation functions are not Lipschitz continuous. 3. Offering explicit upper bounds for Kolmogorov distance, total variation distance, and 1-Wasserstein distance for univariate outputs of shallow Random Gaussian Neural Networks. 4. Utilizing the Stein method and the integration by parts formula for Gaussian distributions to estimate the probability that the output lies within a certain region of space. These results are significant for both theoretical researchers and practitioners, especially in quantifying the Gaussian behavior of the outputs of random neural networks.