Bayesian posterior approximation with stochastic ensembles

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Balabanov, Oleksandr, Mehlig, Bernhard, Linander, Hampus
Format: Preprint
Published: 2022
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913183494569984
author Balabanov, Oleksandr
Mehlig, Bernhard
Linander, Hampus
author_facet Balabanov, Oleksandr
Mehlig, Bernhard
Linander, Hampus
contents We introduce ensembles of stochastic neural networks to approximate the Bayesian posterior, combining stochastic methods such as dropout with deep ensembles. The stochastic ensembles are formulated as families of distributions and trained to approximate the Bayesian posterior with variational inference. We implement stochastic ensembles based on Monte Carlo dropout, DropConnect and a novel non-parametric version of dropout and evaluate them on a toy problem and CIFAR image classification. For both tasks, we test the quality of the posteriors directly against Hamiltonian Monte Carlo simulations. Our results show that stochastic ensembles provide more accurate posterior estimates than other popular baselines for Bayesian inference.
format Preprint
id arxiv_https___arxiv_org_abs_2212_08123
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Bayesian posterior approximation with stochastic ensembles
Balabanov, Oleksandr
Mehlig, Bernhard
Linander, Hampus
Machine Learning
Computer Vision and Pattern Recognition
We introduce ensembles of stochastic neural networks to approximate the Bayesian posterior, combining stochastic methods such as dropout with deep ensembles. The stochastic ensembles are formulated as families of distributions and trained to approximate the Bayesian posterior with variational inference. We implement stochastic ensembles based on Monte Carlo dropout, DropConnect and a novel non-parametric version of dropout and evaluate them on a toy problem and CIFAR image classification. For both tasks, we test the quality of the posteriors directly against Hamiltonian Monte Carlo simulations. Our results show that stochastic ensembles provide more accurate posterior estimates than other popular baselines for Bayesian inference.
title Bayesian posterior approximation with stochastic ensembles
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2212.08123