Low-Rank Updates of Matrix Square Roots
Not on the map. Why?
Abstract (arXiv)
Models in which the covariance matrix has the structure of a sparse matrix plus a low rank perturbation are ubiquitous in data science applications. It is often desirable for algorithms to take advantage of such structures, avoiding costly matrix computations that often require cubic time and quadratic storage. This is often accomplished by performing operations that maintain such structures, e.g. matrix inversion via the Sherman-Morrison-Woodbury formula. In this paper we consider the matrix square root and inverse square root operations. Given a low rank perturbation to a matrix, we argue that a low-rank approximate correction to the (inverse) square root exists. We do so by establishing a geometric decay bound on the true correction's eigenvalues. We then proceed to frame the correction as the solution of an algebraic Riccati equation, and discuss how a low-rank solution to that equation can be computed. We analyze the approximation error incurred when approximately solving the algebraic Riccati equation, providing spectral and Frobenius norm forward and backward error bounds. Finally, we describe several applications of our algorithms, and demonstrate their utility in numerical experiments.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 6 of its 24 references that are in the map. First posted on arXiv 31 Jan 2022.
Builds on
6 of 24 references.
References · 6
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2014Sketching as a Tool for Numerical Linear Algebracited by 285
- 2019Lingvo: a Modular and Scalable Framework for Sequence-to-Sequence Modelingcited by 86
- 2020Fast Matrix Square Roots with Applications to Gaussian Processes and Bayesian Optimizationcited by 26
- 2017Low-rank updates of matrix functionscited by 3
- 2013A Riemannian approach to low-rank algebraic Riccati equationscited by 1
Built on by
Citing papers per year: 2023: 1; 2024: 0; 2025: 1; 2026: 1.
3 papers.
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 69
- 2025Perturbation Bounds for Low-Rank Inverse Approximations under Noisecited by 1
- 2026Approximating Measures on Function Spaces: Transport and Truncationcited by 0
- 2026Approximating Measures on Function Spaces: Transport and Truncation15 Sep 2026
- 2025Perturbation Bounds for Low-Rank Inverse Approximations under Noise29 Oct 2025
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scale12 Sep 2023