提出近邻高斯过程回归的完整理论,解释其高效与稳定的原因。
The Theory and Practice of Highly Scalable Gaussian Process Regression with Nearest Neighbours
- 基于最近邻构建高斯过程模型,实现大规模数据下的快速预测。
- 证明预测误差、校准度和负对数似然均在大样本下收敛至理论极限。
- 揭示模型对超参数不敏感,适合追求可扩展性的实际应用者。
高斯过程(GP)回归是广泛应用的非参数建模工具,但其训练复杂度为立方级,难以处理海量数据。一种实用方法是仅利用每个测试点的最近邻进行预测,如地理空间问题中的最近邻高斯过程(NNGP)及更通用场景的可扩展GPnn方法。尽管这些方法表现出良好性能,其大样本理论仍不完整。本文建立了NNGP和GPnn回归的理论框架,在温和正则性假设下,推导出均方误差(MSE)、校准系数(CAL)和负对数似然(NLL)的几乎必然逐点极限。进一步研究了L₂风险,证明了普遍一致性,并表明风险达到石氏最小极大率n⁻²α/(2p+d),其中α和p刻画回归问题的光滑性。还证明了在紧致超参数集上MSE的均匀收敛性,并显示其关于长度尺度、核尺度和噪声方差的导数在渐近下趋于零,给出明确速率。这解释了GPnn对超参数调优的鲁棒性。结果为NNGP/GPnn提供了严格的统计基础,使其成为全量GP模型的高可扩展且有原则的替代方案。
原文摘要 · Abstract (English)
Gaussian process ($GP$) regression is a widely used non-parametric modeling tool, but its cubic complexity in the training size limits its use on massive data sets. A practical remedy is to predict using only the nearest neighbours of each test point, as in Nearest Neighbour Gaussian Process ($NNGP$) regression for geospatial problems and the related scalable $GPnn$ method for more general machine-learning applications. Despite their strong empirical performance, the large-$n$ theory of $NNGP/GPnn$ remains incomplete. We develop a theoretical framework for $NNGP$ and $GPnn$ regression. Under mild regularity assumptions, we derive almost sure pointwise limits for three key predictive criteria: mean squared error ($MSE$), calibration coefficient ($CAL$), and negative log-likelihood ($NLL$). We then study the $L_2$-risk, prove universal consistency, and show that the risk attains Stone's minimax rate $n^{-2α/(2p+d)}$, where $α$ and $p$ capture regularity of the regression problem. We also prove uniform convergence of $MSE$ over compact hyper-parameter sets and show that its derivatives with respect to lengthscale, kernel scale, and noise variance vanish asymptotically, with explicit rates. This explains the observed robustness of $GPnn$ to hyper-parameter tuning. These results provide a rigorous statistical foundation for $NNGP/GPnn$ as a highly scalable and principled alternative to full $GP$ models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。