Healing Products of Gaussian Processes

Abstract

Gaussian processes (GPs) are nonparametric Bayesian models that have been applied to regression and classification problems. One of the approaches to alleviate their cubic training cost is the use of local GP experts trained on subsets of the data. In particular, product-of-expert models combine the predictive distributions of local experts through a tractable product operation. While these expert models allow for massively distributed computation, their predictions typically suffer from erratic behaviour of the mean or uncalibrated uncertainty quantification. By calibrating predictions via a tempered softmax weighting, we provide a solution to these problems for multiple product-of-expert models, including the generalised product of experts and the robust Bayesian committee machine. Furthermore, we leverage the optimal transport literature and propose a new product-of-expert model that combines predictions of local experts by computing their Wasserstein barycenter, which can be applied to both regression and classification.

arXiv commentICML 2020

References in corpus (9)

• Gaussian Processes for Big Data
• Scalable Variational Gaussian Process Classification
• Kernel Interpolation for Scalable Structured Gaussian Processes (KISS-GP)
• Bayesian Hierarchical Mixtures of Experts
• Generalized Product of Experts for Automatic and Principled Fusion of Gaussian Process Predictions
• Distributed Variational Inference in Sparse Gaussian Process Regression and Latent Variable Models
• Nested Kriging predictions for datasets with large number of observations
• Generalized Robust Bayesian Committee Machine for Large-scale Gaussian Process Regression
• Transductive Log Opinion Pool of Gaussian Process Experts