0

Robust, randomized preconditioning for kernel ridge regression

We investigate preconditioned conjugate gradient methods for kernel ridge regression (KRR) problems with a moderate to large number of data points ($10^4 \leq N \leq 10^7$). We develop and analyze two randomized preconditioners with complementary guarantees.

Year
2023
Hosting
Abstract onlyARXIV-DEFAULT

Cite

Notes

Only stored in your browser.

Attribution

Abstract & full text
arxiv.org/abs/2304.12465ARXIV-DEFAULT
TL;DR
Semantic Scholar
Attribution policy →

Abstract

We investigate preconditioned conjugate gradient methods for kernel ridge regression (KRR) problems with a moderate to large number of data points (10^4 \leq N \leq 10^7). We develop and analyze two randomized preconditioners with complementary guarantees. For full-data KRR, RPCholesky preconditioning requires O(N^2) arithmetic operations to achieve fixed accuracy under sufficiently rapid eigenvalue decay of the kernel matrix. For restricted KRR with k\ll N centers, KRILL preconditioning requires O((N+k^2)k\log k) operations with no eigenvalue-decay assumption. Experiments on benchmark and scientific data sets demonstrate the robustness of both methods relative to existing preconditioners.