Piecewise Deterministic Markov Processes for Bayesian Neural Networks

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Goan, Ethan, Perrin, Dimitri, Mengersen, Kerrie, Fookes, Clinton
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910103560519680
author Goan, Ethan
Perrin, Dimitri
Mengersen, Kerrie
Fookes, Clinton
author_facet Goan, Ethan
Perrin, Dimitri
Mengersen, Kerrie
Fookes, Clinton
contents Inference on modern Bayesian Neural Networks (BNNs) often relies on a variational inference treatment, imposing violated assumptions of independence and the form of the posterior. Traditional MCMC approaches avoid these assumptions at the cost of increased computation due to its incompatibility to subsampling of the likelihood. New Piecewise Deterministic Markov Process (PDMP) samplers permit subsampling, though introduce a model specific inhomogenous Poisson Process (IPPs) which is difficult to sample from. This work introduces a new generic and adaptive thinning scheme for sampling from these IPPs, and demonstrates how this approach can accelerate the application of PDMPs for inference in BNNs. Experimentation illustrates how inference with these methods is computationally feasible, can improve predictive accuracy, MCMC mixing performance, and provide informative uncertainty measurements when compared against other approximate inference schemes.
format Preprint
id arxiv_https___arxiv_org_abs_2302_08724
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Piecewise Deterministic Markov Processes for Bayesian Neural Networks
Goan, Ethan
Perrin, Dimitri
Mengersen, Kerrie
Fookes, Clinton
Machine Learning
Other Statistics
Inference on modern Bayesian Neural Networks (BNNs) often relies on a variational inference treatment, imposing violated assumptions of independence and the form of the posterior. Traditional MCMC approaches avoid these assumptions at the cost of increased computation due to its incompatibility to subsampling of the likelihood. New Piecewise Deterministic Markov Process (PDMP) samplers permit subsampling, though introduce a model specific inhomogenous Poisson Process (IPPs) which is difficult to sample from. This work introduces a new generic and adaptive thinning scheme for sampling from these IPPs, and demonstrates how this approach can accelerate the application of PDMPs for inference in BNNs. Experimentation illustrates how inference with these methods is computationally feasible, can improve predictive accuracy, MCMC mixing performance, and provide informative uncertainty measurements when compared against other approximate inference schemes.
title Piecewise Deterministic Markov Processes for Bayesian Neural Networks
topic Machine Learning
Other Statistics
url https://arxiv.org/abs/2302.08724