Current browse context:
stat.ML
Change to browse by:
References & Citations
Statistics > Machine Learning
Title: Bayesian Deep Ensembles via the Neural Tangent Kernel
(Submitted on 11 Jul 2020 (v1), last revised 24 Oct 2020 (this version, v2))
Abstract: We explore the link between deep ensembles and Gaussian processes (GPs) through the lens of the Neural Tangent Kernel (NTK): a recent development in understanding the training dynamics of wide neural networks (NNs). Previous work has shown that even in the infinite width limit, when NNs become GPs, there is no GP posterior interpretation to a deep ensemble trained with squared error loss. We introduce a simple modification to standard deep ensembles training, through addition of a computationally-tractable, randomised and untrainable function to each ensemble member, that enables a posterior interpretation in the infinite width limit. When ensembled together, our trained NNs give an approximation to a posterior predictive distribution, and we prove that our Bayesian deep ensembles make more conservative predictions than standard deep ensembles in the infinite width limit. Finally, using finite width NNs we demonstrate that our Bayesian deep ensembles faithfully emulate the analytic posterior predictive when available, and can outperform standard deep ensembles in various out-of-distribution settings, for both regression and classification tasks.
Submission history
From: Bobby He [view email][v1] Sat, 11 Jul 2020 22:10:52 GMT (108kb,D)
[v2] Sat, 24 Oct 2020 16:51:14 GMT (366kb,D)
Link back to: arXiv, form interface, contact.