Timezone: »
Poster
Giga-scale Kernel Matrix-Vector Multiplication on GPU
Robert Hu · Siu Lun Chau · Dino Sejdinovic · Joan Glaunès
Kernel matrix-vector multiplication (KMVM) is a foundational operation in machine learning and scientific computing. However, as KMVM tends to scale quadratically in both memory and time, applications are often limited by these computational constraints. In this paper, we propose a novel approximation procedure coined \textit{Faster-Fast and Free Memory Method} ($\text{F}^3$M) to address these scaling issues of KMVM for tall~($10^8\sim 10^9$) and skinny~($D\leq7$) data. Extensive experiments demonstrate that $\text{F}^3$M has empirical \emph{linear time and memory} complexity with a relative error of order $10^{-3}$ and can compute a full KMVM for a billion points \emph{in under a minute} on a high-end GPU, leading to a significant speed-up in comparison to existing CPU methods. We demonstrate the utility of our procedure by applying it as a drop-in for the state-of-the-art GPU-based linear solver FALKON, \emph{improving speed 1.5-5.5 times} at the cost of $<1\%$ drop in accuracy. We further demonstrate competitive results on \emph{Gaussian Process regression} coupled with significant speedups on a variety of real-world datasets.
Author Information
Robert Hu (Amazon)
Siu Lun Chau (University of Oxford)
Dino Sejdinovic (University of Adelaide)
Joan Glaunès (Université Paris Cité)
More from the Same Authors
-
2022 : Bayesian inference for aerosol vertical profiles »
Shahine Bouabid · Duncan Watson-Parris · Dino Sejdinovic -
2022 Poster: Explaining Preferences with Shapley Values »
Robert Hu · Siu Lun Chau · Jaime Ferrando Huertas · Dino Sejdinovic -
2022 Poster: RKHS-SHAP: Shapley Values for Kernel Methods »
Siu Lun Chau · Robert Hu · Javier González · Dino Sejdinovic -
2022 Poster: Generalized Variational Inference in Function Spaces: Gaussian Measures meet Bayesian Deep Learning »
Veit David Wild · Robert Hu · Dino Sejdinovic -
2021 Poster: BayesIMP: Uncertainty Quantification for Causal Data Fusion »
Siu Lun Chau · Jean-Francois Ton · Javier González · Yee Teh · Dino Sejdinovic -
2021 Poster: Deconditional Downscaling with Gaussian Processes »
Siu Lun Chau · Shahine Bouabid · Dino Sejdinovic