Mehler’s Formula, Branching Process, and Compositional Kernels of Deep Neural Networks

Download:

Abstract:

We utilize a connection between compositional kernels and branching processes via Mehler’s formula to study deep neural networks. This new probabilistic insight provides us a novel perspective on the mathematical role of activation functions in compositional neural networks. We study the unscaled and rescaled limits of the compositional kernels and explore the different phases of the limiting behavior, as the compositional depth increases. We investigate the memorization capacity of the compositional kernels and neural networks by characterizing the interplay among compositional depth, sample size, dimensionality, and non-linearity of the activation. Explicit formulas on the eigenvalues of the compositional kernel are provided, which quantify the complexity of the corresponding reproducing kernel Hilbert space. On the methodological front, we propose a new random features algorithm, which compresses the compositional layers by devising a new activation function.

Citation

Tengyuan Liang, and Hai Tran-Bach. 2022. “Mehler’s Formula, Branching Process, and Compositional Kernels of Deep Neural Networks.” Journal of the American Statistical Association (Theory and Methods) 117 (539): 1324–1337.

@article{Liang_2021,
   title={Mehler’s Formula, Branching Process, and Compositional Kernels of Deep Neural Networks},
   volume={117},
   ISSN={1537-274X},
   url={http://dx.doi.org/10.1080/01621459.2020.1853547},
   DOI={10.1080/01621459.2020.1853547},
   number={539},
   journal={Journal of the American Statistical Association},
   author={Liang, Tengyuan and Tran-Bach, Hai},
   year={2021},
   month=jan, pages={1324–1337} }