Share your thoughts, 1 month free Claude Pro on usSee more
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
Out-of-Distribution DetectionCIFAR-10
AUROC90.23
121
Out-of-Distribution DetectionImageNet--
108
Out-of-Distribution DetectionCIFAR-10 vs CIFAR-100
AUROC82.38
70
ClassificationBreast
Accuracy60
36
ClassificationGlass
Accuracy17
32
Out-of-Distribution DetectionCIFAR10 vs. SVHN
AUROC84.22
31
Out-of-Distribution DetectionImageNet-R
ROC AUC0.7997
28
Out-of-Distribution DetectionPlaces365--
21
Out-of-Distribution DetectionImageNet A
AUROC83.55
19
Showing 10 of 59 rows

Other info

Follow for update