572

Robust, randomized preconditioning for kernel ridge regression

Main:24 Pages
14 Figures
Bibliography:6 Pages
2 Tables
Abstract

This paper introduces two randomized preconditioning techniques for robustly solving kernel ridge regression (KRR) problems with a medium to large number of data points (104N10710^4 \leq N \leq 10^7). The first method, RPCholesky preconditioning, is capable of accurately solving the full-data KRR problem in O(N2)O(N^2) arithmetic operations, assuming sufficiently rapid polynomial decay of the kernel matrix eigenvalues. The second method, KRILL preconditioning, offers an accurate solution to a restricted version of the KRR problem involving kNk \ll N selected data centers at a cost of O((N+k2)klogk)O((N + k^2) k \log k) operations. The proposed methods solve a broad range of KRR problems and overcome the failure modes of previous KRR preconditioners, making them ideal for practical applications.

View on arXiv
Comments on this paper