Deep Neural Networks as Gaussian Processes
Jaehoon Lee , Yasaman Bahri , Roman Novak , Samuel S. Schoenholz, Jeffrey Pennington, Jascha Sohl-Dickstein Google Brain {jaehlee, yasamanb, romann, schsam, jpennin, Both authors contributed equally to this work.Work done as a member of the Google AI Residency program (g.co/airesidency).
Abstract
It has long been known that a single-layer fully-connected neural network with an i.i.d. prior over its parameters is equivalent to a Gaussian process (GP), in the limit of infinite network width. This correspondence enables exact Bayesian inference for infinite width neural networks on regression tasks by means of evaluating the corresponding GP. Recently, kernel functions which mimic multi-layer random neural networks have been developed, but only outside of a Bayesian framework. As such, previous work has not identified that these kernels can be used as covariance functions for GPs and allow fully Bayesian prediction with a deep neural network.
原文 arXiv:1711.00165;中英对照 + 大白话阅读 https://aha.fim.ai/paper/1711.00165v3