acceptodds
Published as a conference paper at NeurIPS 2026

It Cancels: O(r²) Cholesky Updates for Whitened Operators

Anupama Sridhar, Jack Shi, Ravi Deshpande, Jack Li, Alexander R Johansen, Michael P Snyder

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.