English

Robust, randomized preconditioning for kernel ridge regression

Numerical Analysis 2025-10-22 v5 Numerical Analysis Machine Learning

Abstract

This paper investigates preconditioned conjugate gradient techniques for solving kernel ridge regression (KRR) problems with a medium to large number of data points (104N10710^4 \leq N \leq 10^7), and it describes two methods with the strongest guarantees available. The first method, RPCholesky preconditioning, accurately solves 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 efficiently solve a range of KRR problems, making them well-suited for practical applications.

Keywords

Cite

@article{arxiv.2304.12465,
  title  = {Robust, randomized preconditioning for kernel ridge regression},
  author = {Mateo Díaz and Ethan N. Epperly and Zachary Frangella and Joel A. Tropp and Robert J. Webber},
  journal= {arXiv preprint arXiv:2304.12465},
  year   = {2025}
}

Comments

30 pages, 11 figures