Precision aggregated local models

Adam M. Edwards,Robert B. Gramacy
DOI: https://doi.org/10.1002/sam.11547
2021-10-06
Abstract:Large-scale Gaussian process (GP) regression is infeasible for large training data due to cubic scaling of flops and quadratic storage involved in working with covariance matrices. Remedies in recent literature focus on divide-and-conquer, for example, partitioning into subproblems and inducing functional (and thus computational) independence. Such approximations can be speedy, accurate, and sometimes even more flexible than ordinary GPs. However, a big downside is loss of continuity at partition boundaries. Modern methods like local approximate GPs (LAGPs) imply effectively infinite partitioning and are thus both good and bad in this regard. Model averaging, an alternative to divide-and-conquer, can maintain absolute continuity but often over-smooths, diminishing accuracy. Here we propose putting LAGP-like methods into a local experts-like framework, blending partition-based speed with model-averaging continuity, as a flagship example of what we call precision aggregated local models (PALM). Using LAGPs, each selecting from total data pairs, our scheme is at most cubic in , quadratic in , and linear in . Extensive empirical illustration shows how PALM is at least as accurate as LAGP, can be much faster, and furnishes continuous predictions. Finally, we propose sequential updating scheme that greedily refines a PALM predictor up to a computational budget.
What problem does this paper attempt to address?