Variational bootstrap for classification

Kozyrskiy, Bogdan; Milios, Dimitrios; Filippone, Maurizio
KES 2022, 26th International Conference on Knowledge-Based and Intelligent Information & Engineering Systems, 7-9 September 2022, Verona, Italy / Also published in Procedia Computer Science, Vol. 207

Carrying out Bayesian inference over parameters of statistical models is intractable when the likelihood and the prior are nonconjugate. Variational bootstrap provides a way to obtain samples from the posterior distribution over model parameters, where each sample is the solution of a task where the labels are perturbed. For Bayesian linear regression with a Gaussian likelihood, variational bootstrap yields samples from the exact posterior, whereas for nonlinear models with a Gaussian likelihood some guarantees of approaching the true posterior can be established. In this work, we extend variational bootstrap to the Bernoulli likelihood to tackle classification tasks. We use a transformation of the labels which allows us to turn the classification task into a regression one, and then we apply variational bootstrap to obtain samples from an approximate posterior distribution over the parameters of the model. Variational bootstrap allows us to employ advanced gradient optimization techniques which provide fast convergence. We provide experimental evidence that the proposed approach allows us to achieve classification accuracy and
uncertainty estimation comparable with MCMC methods at a fraction of the cost.

DOI
Type:
Conference
City:
Verona
Date:
2022-09-07
Department:
Data Science
Eurecom Ref:
7007
Copyright:
© Elsevier. Personal use of this material is permitted. The definitive version of this paper was published in KES 2022, 26th International Conference on Knowledge-Based and Intelligent Information & Engineering Systems, 7-9 September 2022, Verona, Italy / Also published in Procedia Computer Science, Vol. 207 and is available at : https://doi.org/10.1016/j.procs.2022.09.178

PERMALINK : https://www.eurecom.fr/publication/7007