Bayesian Deep Convolutional Networks with Many Channels are Gaussian Processes
Roman Novak Lechao Xiao Thanks: Google AI Residents (g.co/airesidency). $ˆ†,ˆ‡$ Equal contribution. Jaehoon Lee Yasaman Bahri Greg Yang [0.15cm]Jiri Hron, Daniel A. Abolafia, Jeffrey Pennington, Jascha Sohl-Dickstein Affiliation: danabo, jpennin, [0.2cm] Google Brain Microsoft Research AI Department of Engineering Affiliation: danabo, jpennin, University of Cambridge [0.2cm] {romann, xlc, jaehlee, yasamanb,
Abstract
There is a previously identified equivalence between wide fully connected neural networks (FCNs) and Gaussian processes (GPs). This equivalence enables, for instance, test set predictions that would have resulted from a fully Bayesian, infinitely wide trained FCN to be computed without ever instantiating the FCN, but by instead evaluating the corresponding GP. In this work, we derive an analogous equivalence for multi-layer convolutional neural networks (CNNs) both with and without pooling layers, and achieve state of the art results on CIFAR10 for GPs without trainable kernels. We also introduce a Monte Carlo method to estimate the GP corresponding to a given neural network architecture, even in cases where the analytic form has too many terms to be computationally feasible.
原文 arXiv:1810.05148;中英对照 + 大白话阅读 https://aha.fim.ai/paper/1810.05148v4