πVAE: a stochastic process prior for Bayesian deep learning with MCMC
File(s)s11222-022-10151-w.pdf (4.36 MB)
Published version
Author(s)
Type
Journal Article
Abstract
Stochastic processes provide a mathematically elegant way to model complex data. In theory, they provide flexible priors over function classes that can encode a wide range of interesting assumptions. However, in practice efficient inference by optimisation or marginalisation is difficult, a problem further exacerbated with big data and high dimensional input spaces. We propose a novel variational autoencoder (VAE) called the prior encoding variational autoencoder (πVAE). πVAE is a new continuous stochastic process. We use πVAE to learn low dimensional embeddings of function classes by combining a trainable feature mapping with generative model using a VAE. We show that our framework can accurately learn expressive function classes such as Gaussian processes, but also properties of functions such as their integrals. For popular tasks, such as spatial interpolation, πVAE achieves state-of-the-art performance both in terms of accuracy and computational efficiency. Perhaps most usefully, we demonstrate an elegant and scalable means of performing fully Bayesian inference for stochastic processes within probabilistic programming languages such as Stan.
Date Issued
2022-10-17
Date Acceptance
2022-09-09
Citation
Statistics and Computing, 2022, 32
ISSN
0960-3174
Publisher
Springer
Journal / Book Title
Statistics and Computing
Volume
32
Copyright Statement
This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/.
License URL
Sponsor
Medical Research Council (MRC)
Grant Number
MR/R015600/1
Subjects
cs.LG
cs.LG
stat.ML
Statistics & Probability
0104 Statistics
0802 Computation Theory and Mathematics
Publication Status
Published
Article Number
ARTN 96