Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Laplace Redux -- Effortless Bayesian Deep Learning

About

Bayesian formulations of deep learning have been shown to have compelling theoretical properties and offer practical functional benefits, such as improved predictive uncertainty quantification and model selection. The Laplace approximation (LA) is a classic, and arguably the simplest family of approximations for the intractable posteriors of deep neural networks. Yet, despite its simplicity, the LA is not as popular as alternatives like variational Bayes or deep ensembles. This may be due to assumptions that the LA is expensive due to the involved Hessian computation, that it is difficult to implement, or that it yields inferior results. In this work we show that these are misconceptions: we (i) review the range of variants of the LA including versions with minimal cost overhead; (ii) introduce "laplace", an easy-to-use software library for PyTorch offering user-friendly access to all major flavors of the LA; and (iii) demonstrate through extensive experiments that the LA is competitive with more popular alternatives in terms of performance, while excelling in terms of computational cost. We hope that this work will serve as a catalyst to a wider adoption of the LA in practical deep learning, including in domains where Bayesian approaches are not typically considered at the moment.

Erik Daxberger, Agustinus Kristiadi, Alexander Immer, Runa Eschenhagen, Matthias Bauer, Philipp Hennig• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 (test)
Accuracy95.6
3381
Image ClassificationCIFAR-100 (test)
Accuracy80.1
14
Out-of-Distribution DetectionPlaces365--
14
ClassificationBreast
Accuracy60
12
ClassificationGlass
Log Time (s)0.98
11
Classificationionosphere
Log Time (s)1.47
11
ClassificationUCI Australian
NLL0.69
11
ClassificationUCI Breast
NLL0.64
11
ClassificationUCI Glass
NLL2.06
11
ClassificationUCI Ionosphere
NLL0.72
11
Showing 10 of 30 rows

Other info

Follow for update