Scalable Gaussian Processes with Latent Kronecker Structure

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Lin, Jihao Andreas, Ament, Sebastian, Balandat, Maximilian, Eriksson, David, Hernández-Lobato, José Miguel, Bakshy, Eytan
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909738298507264
author Lin, Jihao Andreas
Ament, Sebastian
Balandat, Maximilian
Eriksson, David
Hernández-Lobato, José Miguel
Bakshy, Eytan
author_facet Lin, Jihao Andreas
Ament, Sebastian
Balandat, Maximilian
Eriksson, David
Hernández-Lobato, José Miguel
Bakshy, Eytan
contents Applying Gaussian processes (GPs) to very large datasets remains a challenge due to limited computational scalability. Matrix structures, such as the Kronecker product, can accelerate operations significantly, but their application commonly entails approximations or unrealistic assumptions. In particular, the most common path to creating a Kronecker-structured kernel matrix is by evaluating a product kernel on gridded inputs that can be expressed as a Cartesian product. However, this structure is lost if any observation is missing, breaking the Cartesian product structure, which frequently occurs in real-world data such as time series. To address this limitation, we propose leveraging latent Kronecker structure, by expressing the kernel matrix of observed values as the projection of a latent Kronecker product. In combination with iterative linear system solvers and pathwise conditioning, our method facilitates inference of exact GPs while requiring substantially fewer computational resources than standard iterative methods. We demonstrate that our method outperforms state-of-the-art sparse and variational GPs on real-world datasets with up to five million examples, including robotics, automated machine learning, and climate applications.
format Preprint
id arxiv_https___arxiv_org_abs_2506_06895
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Scalable Gaussian Processes with Latent Kronecker Structure
Lin, Jihao Andreas
Ament, Sebastian
Balandat, Maximilian
Eriksson, David
Hernández-Lobato, José Miguel
Bakshy, Eytan
Machine Learning
Applying Gaussian processes (GPs) to very large datasets remains a challenge due to limited computational scalability. Matrix structures, such as the Kronecker product, can accelerate operations significantly, but their application commonly entails approximations or unrealistic assumptions. In particular, the most common path to creating a Kronecker-structured kernel matrix is by evaluating a product kernel on gridded inputs that can be expressed as a Cartesian product. However, this structure is lost if any observation is missing, breaking the Cartesian product structure, which frequently occurs in real-world data such as time series. To address this limitation, we propose leveraging latent Kronecker structure, by expressing the kernel matrix of observed values as the projection of a latent Kronecker product. In combination with iterative linear system solvers and pathwise conditioning, our method facilitates inference of exact GPs while requiring substantially fewer computational resources than standard iterative methods. We demonstrate that our method outperforms state-of-the-art sparse and variational GPs on real-world datasets with up to five million examples, including robotics, automated machine learning, and climate applications.
title Scalable Gaussian Processes with Latent Kronecker Structure
topic Machine Learning
url https://arxiv.org/abs/2506.06895