Distributed Bayesian Inference in Massive Spatial Data

doi:10.1214/22-STS868

Distributed Bayesian Inference in Massive Spatial Data

2023-05-01

3

PRE

AI

R

Rajarshi Guhaniyogi *

DOI：10.1214/22-STS868

原文链接

原文求助

摘要

En 中文

Gaussian process (GP) regression is computationally expensive in spatial applications involving massive data. Various methods address this limitation, including a small number of Bayesian methods based on dis-tributed computations (or the divide-and-conquer strategy). Focusing on the latter literature, we achieve three main goals. First, we develop an extensible Bayesian framework for distributed spatial GP regression that embeds many popular methods. The proposed framework has three steps that partition the entire data into many subsets, apply a readily available Bayesian spatial pro-cess model in parallel on all the subsets, and combine the posterior distri-butions estimated on all the subsets into a pseudo posterior distribution that conditions on the entire data. The combined pseudo posterior distribution replaces the full data posterior distribution in prediction and inference prob-lems. Demonstrating our framework's generality, we extend posterior com-putations for (nondistributed) spatial process models with a stationary full -rank and a nonstationary low-rank GP priors to the distributed setting. Sec-ond, we contrast the empirical performance of popular distributed approaches with some widely-used, nondistributed alternatives and highlight their rela-tive advantages and shortcomings. Third, we provide theoretical support for our numerical observations and show that the Bayes L2-risks of the combined posterior distributions obtained from a subclass of the divide-and-conquer methods achieves the near-optimal convergence rate in estimating the true spatial surface with various types of covariance functions. Additionally, we provide upper bounds on the number of subsets to achieve these near-optimal rates.

Keyword:

Distributed Bayesian inference

Gaussian process

low-rank Gaussian process

massive spatial data

Wasserstein barycenter

Distributed Bayesian Inference in Massive Spatial Data

摘要

期刊

机构