Leveraging Locality and Robustness to Achieve Massively Scalable Gaussian Process Regression
Robert Allison, Anthony Stephenson, Samuel F, Edward O. Pyzer-Knapp
Abstract
The accurate predictions and principled uncertainty measures provided by GP regression incur O(n^3) cost which is prohibitive for modern-day large-scale applications. This has motivated extensive work on computationally efficient approximations. We introduce a new perspective by exploring robustness properties and limiting behaviour of GP nearest-neighbour (GPnn) prediction. We demonstrate through theory and simulation that as the data-size n increases, accuracy of estimated parameters and GP model assumptions become increasingly irrelevant to GPnn predictive accuracy. Consequently, it is sufficient to spend small amounts of work on parameter estimation in order to achieve high MSE accuracy, even in the presence of gross misspecification. In contrast, as n tends to infinity, uncertainty calibration and NLL are shown to remain sensitive to just one parameter, the additive noise-variance; but we show that this source of inaccuracy can be corrected for, thereby achieving both well-calibrated uncertainty measures and accurate predictions at remarkably low computational cost. We exhibit a very simple GPnn regression algorithm with stand-out performance compared to other state-of-the-art GP approximations as measured on large UCI datasets. It operates at a small fraction of those other methods' training costs, for example on a basic laptop taking about 30 seconds to train on a dataset of size n = 1.6 x 10^6.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 758ed908-42e8-4804-9978-1c40d592f383Cited by top-tier papers1
Ask how each one uses itBuilds on3
- Parametric Gaussian Process RegressorsMartin Jankowiak, Geoff Pleiss, Jacob R. GardnerICML 2020 · 82 citations
- Healing Products of Gaussian Process ExpertsSamuel Cohen, Rendani Mbuvha, Tshilidzi Marwala, Marc Peter DeisenrothICML 2020 · 34 citations
- Sparse within Sparse Gaussian Processes using Neighbor InformationGia-Lac Tran, Dimitrios Milios, Pietro Michiardi, Maurizio FilipponeICML 2021 · 19 citations
Related papers
- Computation-Aware Gaussian Processes: Model Selection And Linear-Time InferenceJonathan Wenger, Kaiwen Wu, Philipp Hennig, Jacob R. Gardner et al.NeurIPS 2024 · 15 citations
- Variational nearest neighbor Gaussian processLuhuan Wu, Geoff Pleiss, John P. CunninghamICML 2022 · 18 citations
- Variational Linearized Laplace Approximation for Bayesian Deep LearningLuis A. Ortega Andrés, Simón Rodríguez Santana, Daniel Hernández-LobatoICML 2024 · 12 citations
- Input Dependent Sparse Gaussian ProcessesBahram Jafrasteh, Carlos Villacampa-Calvo, Daniel Hernández-LobatoICML 2022 · 7 citations
- DNNR: Differential Nearest Neighbors RegressionYoussef Nader, Leon Sixt, Tim LandgrafICML 2022 · 20 citations
