Log In Sign Up

Scalable Gaussian-process regression and variable selection using Vecchia approximations

by   Jian Cao, et al.

Gaussian process (GP) regression is a flexible, nonparametric approach to regression that naturally quantifies uncertainty. In many applications, the number of responses and covariates are both large, and a goal is to select covariates that are related to the response. For this setting, we propose a novel, scalable algorithm, coined VGPR, which optimizes a penalized GP log-likelihood based on the Vecchia GP approximation, an ordered conditional approximation from spatial statistics that implies a sparse Cholesky factor of the precision matrix. We traverse the regularization path from strong to weak penalization, sequentially adding candidate covariates based on the gradient of the log-likelihood and deselecting irrelevant covariates via a new quadratic constrained coordinate descent algorithm. We propose Vecchia-based mini-batch subsampling, which provides unbiased gradient estimators. The resulting procedure is scalable to millions of responses and thousands of covariates. Theoretical analysis and numerical studies demonstrate the improved scalability and accuracy relative to existing methods.


page 13

page 16


High-dimensional Censored Regression via the Penalized Tobit Likelihood

The Tobit model has long been the standard method for regression with a ...

Approximate Bayesian inference from noisy likelihoods with Gaussian process emulated MCMC

We present an efficient approach for doing approximate Bayesian inferenc...

Gaussian Process Priors for Dynamic Paired Comparison Modelling

Dynamic paired comparison models, such as Elo and Glicko, are frequently...

Scalable Training of Inference Networks for Gaussian-Process Models

Inference in Gaussian process (GP) models is computationally challenging...

Hierarchical Gaussian Process Models for Regression Discontinuity/Kink under Sharp and Fuzzy Designs

We propose nonparametric Bayesian estimators for causal inference exploi...

A Scalable Gaussian Process for Large-Scale Periodic Data

The periodic Gaussian process (PGP) has been increasingly used to model ...