Structured Partial Stochasticity in Bayesian Neural Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autore principale: Rochussen, Tommy
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929405354311680
author Rochussen, Tommy
author_facet Rochussen, Tommy
contents Bayesian neural network posterior distributions have a great number of modes that correspond to the same network function. The abundance of such modes can make it difficult for approximate inference methods to do their job. Recent work has demonstrated the benefits of partial stochasticity for approximate inference in Bayesian neural networks; inference can be less costly and performance can sometimes be improved. I propose a structured way to select the deterministic subset of weights that removes neuron permutation symmetries, and therefore the corresponding redundant posterior modes. With a drastically simplified posterior distribution, the performance of existing approximate inference schemes is found to be greatly improved.
format Preprint
id arxiv_https___arxiv_org_abs_2405_17666
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Structured Partial Stochasticity in Bayesian Neural Networks
Rochussen, Tommy
Machine Learning
Bayesian neural network posterior distributions have a great number of modes that correspond to the same network function. The abundance of such modes can make it difficult for approximate inference methods to do their job. Recent work has demonstrated the benefits of partial stochasticity for approximate inference in Bayesian neural networks; inference can be less costly and performance can sometimes be improved. I propose a structured way to select the deterministic subset of weights that removes neuron permutation symmetries, and therefore the corresponding redundant posterior modes. With a drastically simplified posterior distribution, the performance of existing approximate inference schemes is found to be greatly improved.
title Structured Partial Stochasticity in Bayesian Neural Networks
topic Machine Learning
url https://arxiv.org/abs/2405.17666