Abstract:This paper presents a new approach for Gaussian process (GP) regression for large datasets. The approach involves partitioning the regression input domain into multiple local regions with a different local GP model fitted in each region. Unlike existing local partitioned GP approaches, we introduce a technique for patching together the local GP models nearly seamlessly to ensure that the local GP models for two neighboring regions produce nearly the same response prediction and prediction error variance on the boundary between the two regions. This largely mitigates the well-known discontinuity problem that degrades the boundary accuracy of existing local partitioned GP methods. Our main innovation is to represent the continuity conditions as additional pseudo-observations that the differences between neighboring GP responses are identically zero at an appropriately chosen set of boundary input locations. To predict the response at any input location, we simply augment the actual response observations with the pseudo-observations and apply standard GP prediction methods to the augmented data. In contrast to heuristic continuity adjustments, this has an advantage of working within a formal GP framework, so that the GP-based predictive uncertainty quantification remains valid. Our approach also inherits a sparse block-like structure for the sample covariance matrix, which results in computationally efficient closed-form expressions for the predictive mean and variance. In addition, we provide a new spatial partitioning scheme based on a recursive space partitioning along local principal component directions, which makes the proposed approach applicable for regression domains having more than two dimensions. Using three spatial datasets and three higher dimensional datasets, we investigate the numerical performance of the approach and compare it to several state-of-the-art approaches.

Parallel cross-validation: A scalable fitting method for Gaussian process models

A Fast GP Regression Method Using Banded Sparsification of Inverse Covariance

Understanding and comparing scalable Gaussian process regression for big data

Towards Scalable Gaussian Process Modeling

Patchwork Kriging for Large-scale Gaussian Process Regression

Block Vecchia Approximation for Scalable and Efficient Gaussian Process Computations

Leveraging Locality and Robustness to Achieve Massively Scalable Gaussian Process Regression

ProSpar-GP: scalable Gaussian process modeling with massive non-stationary datasets

Iterative Methods for Full-Scale Gaussian Process Approximations for Large Spatial Data

GPU-Accelerated Vecchia Approximations of Gaussian Processes for Geospatial Data using Batched Matrix Computations

When Gaussian Process Meets Big Data: A Review of Scalable GPs

Scalable Gaussian Process Using Inexact Admm For Big Data

Scaling Gaussian Process Regression with Derivatives

A Robust Approach to Gaussian Processes Implementation

Compactly-supported nonstationary kernels for computing exact Gaussian processes on big data

Scalable mixed-domain Gaussian process modeling and model reduction for longitudinal data

Fast Calculation of Gaussian Process Multiple-Fold Cross-Validation Residuals and their Covariances

Gaussian Process Regression in the Flat Limit

Efficient Gaussian process based on BFGS updating and logdet approximation

Composite Gaussian Processes: Scalable Computation and Performance Analysis

Solving High Frequency and Multi-Scale PDEs with Gaussian Processes