It Cancels: O(r²) Cholesky Updates for Whitened Operators
Abstract
Many sequential Bayesian methods maintain a whitened operator , where is the Cholesky factor of a kernel, precision, or curvature matrix. When the underlying matrix changes by a rank- perturbation, existing techniques propagate only one-sided quantities ; refreshing the sandwiched operator requires a full re-factorization that dominates iteration cost, consuming up to 91% of per-step time in natural gradient methods. We show that this cost is unnecessary. For pre-whitened inputs, intermediate factors of cancel exactly from both sides of the augmented Givens rotations, and the updated can be recovered in time by applying the rotations double-sidedly to a zero-padded matrix, followed by a rank-1 correction. We provide a self-contained derivation, stability bounds, and implementations as a C++ extension, MLX kernel, and Triton GPU kernel. On streaming sparse GP regression, online Bayesian optimization on protein fitness landscapes, and natural gradient preconditioning, the method matches full re-factorization in accuracy with up to speedups. At , training time drops from 1075s to 198s, making larger curvature charts affordable and producing strictly better test MSE.
Related papers
See more related submissions when the market opens.