arXiv · 2304.01762
Incorporating Unlabelled Data into Bayesian Neural Networks
Abstract
Conventional Bayesian Neural Networks (BNNs) are unable to leverage unlabelled data to improve their predictions. To overcome this limitation, we introduce Self-Supervised Bayesian Neural Networks, which use unlabelled data to learn models with suitable prior predictive distributions. This is achieved by leveraging contrastive pretraining techniques and optimising a variational lower bound. We then show that the prior predictive distributions of self-supervised BNNs capture problem semantics better than conventional BNN priors. In turn, our approach offers improved predictive performance over conventional BNNs, especially in low-budget regimes.
Explore related subjects
Keep this discovery
Mrinank Sharma, Tom Rainforth, Yee Whye Teh, Vincent Fortuin. 2023-04-04. Incorporating Unlabelled Data into Bayesian Neural Networks. https://arxiv.org/abs/2304.01762
Cite the original work for its findings. Save a collection to share your selection of sources.