Subspace embedding with random Khatri-Rao products and its application to eigensolvers (2405.11962v1)
Abstract: Various iterative eigenvalue solvers have been developed to compute parts of the spectrum for a large sparse matrix, including the power method, Krylov subspace methods, contour integral methods, and preconditioned solvers such as the so called LOBPCG method. All of these solvers rely on random matrices to determine, e.g., starting vectors that have, with high probability, a non-negligible overlap with the eigenvectors of interest. For this purpose, a safe and common choice are unstructured Gaussian random matrices. In this work, we investigate the use of random Khatri-Rao products in eigenvalue solvers. On the one hand, we establish a novel subspace embedding property that provides theoretical justification for the use of such structured random matrices. On the other hand, we highlight the potential algorithmic benefits when solving eigenvalue problems with Kronecker product structure, as they arise frequently from the discretization of eigenvalue problems for differential operators on tensor product domains. In particular, we consider the use of random Khatri-Rao products within a contour integral method and LOBPCG. Numerical experiments indicate that the gains for the contour integral method strongly depend on the ability to efficiently and accurately solve (shifted) matrix equations with low-rank right-hand side. The flexibility of LOBPCG to directly employ preconditioners makes it easier to benefit from Khatri-Rao product structure, at the expense of having less theoretical justification.
- Oblivious sketching of high-degree polynomial kernels. In Proceedings of the 2020 ACM-SIAM Symposium on Discrete Algorithms, pages 141–160. SIAM, Philadelphia, PA, 2020.
- A numerical method for nonlinear eigenvalue problems using contour integrals. JSIAM Lett., 1:52–55, 2009.
- Johnson-Lindenstrauss embeddings with Kronecker structure. SIAM J. Matrix Anal. Appl., 43(4):1806–1850, 2022.
- A practical randomized CP tensor decomposition. SIAM J. Matrix Anal. Appl., 39(2):876–901, 2018.
- Low rank methods for a class of generalized Lyapunov equations and related issues. Numer. Math., 124(3):441–470, 2013.
- On the ADI method for Sylvester equations. J. Comput. Appl. Math., 233(4):1035–1045, 2009.
- Wolf-Jürgen Beyn. An integral method for solving nonlinear eigenvalue problems. Linear Algebra Appl., 436(10):3839–3863, 2012.
- Randomized interpolative decomposition of separated representations. J. Comput. Phys., 281:116–134, 2015.
- On approximation of the eigenvalues of perturbed periodic Schrödinger operators. J. Phys. A, 40(31):9319–9329, 2007.
- Norm and trace estimation with random rank-one vectors. SIAM J. Matrix Anal. Appl., 42(1):202–223, 2021.
- Optimal approximate matrix product in terms of stable rank. arXiv preprint arXiv:1507.02268, 2015.
- Optimal approximate matrix product in terms of stable rank. In 43rd International Colloquium on Automata, Languages, and Programming, volume 55 of LIPIcs. Leibniz Int. Proc. Inform., pages Art. No. 11, 14. Schloss Dagstuhl. Leibniz-Zent. Inform., Wadern, 2016.
- T. A. Davis. Direct methods for sparse linear systems, volume 2 of Fundamentals of Algorithms. Society for Industrial and Applied Mathematics (SIAM), Philadelphia, PA, 2006.
- A multilinear singular value decomposition. SIAM J. Matrix Anal. Appl., 21(4):1253–1278, 2000.
- A robust and efficient implementation of LOBPCG. SIAM J. Sci. Comput., 40(5):C655–C676, 2018.
- Matrix computations. Johns Hopkins Studies in the Mathematical Sciences. Johns Hopkins University Press, Baltimore, MD, fourth edition, 2013.
- Zolotarev quadrature rules and load balancing for the FEAST eigensolver. SIAM J. Sci. Comput., 37(4):A2100–A2122, 2015.
- Finding structure with randomness: probabilistic algorithms for constructing approximate matrix decompositions. SIAM Rev., 53(2):217–288, 2011.
- U. Hetmaniuk and R. Lehoucq. Basis selection in LOBPCG. J. Comput. Phys., 218(1):324–332, 2006.
- Block locally optimal preconditioned eigenvalue xolvers (BLOPEX) in hypre and PETSc. SIAM J. Sci. Comput., 29(5):2224–2239, 2007.
- Andrew V. Knyazev. Toward the optimal preconditioned eigensolver: locally optimal block preconditioned conjugate gradient method. SIAM J. Sci. Comput., 23(2):517–541, 2001. Copper Mountain Conference (2000).
- A geometric theory for preconditioned inverse iteration. III. A short and sharp convergence estimate for generalized eigenvalue problems. Linear Algebra Appl., 358:95–114, 2003.
- Tensor decompositions and applications. SIAM Review, 51(3):455–500, 2009.
- Recompression of Hadamard products of tensors in Tucker format. SIAM J. Sci. Comput., 39(5):A1879–A1902, 2017.
- Low-rank tensor methods with subspace correction for symmetric eigenvalue problems. SIAM J. Sci. Comput., 36(5):A2346–A2368, 2014.
- Preconditioned low-rank methods for high-dimensional elliptic PDE eigenvalue problems. Comput. Methods Appl. Math., 11(3):363–381, 2011.
- Patrick Kürschner. Efficient Low-Rank Solution of Large-Scale Matrix Equations, volume 45 of Forschungsberichte aus dem Max-Planck-Institut für Dynamik komplexer technischer Systeme. Shaker Verlag, Aachen, 2016.
- Rafał Latała. Estimation of moments of sums of independent real random variables. Ann. Probab., 25(3):1502–1513, 1997.
- Probability in Banach spaces. Classics in Mathematics. Springer-Verlag, Berlin, 2011.
- Randomized numerical linear algebra: foundations and algorithms. Acta Numer., 29:403–572, 2020.
- Jiří Matoušek. On variants of the Johnson-Lindenstrauss lemma. Random Structures Algorithms, 33(2):142–156, 2008.
- Hutchinson’s estimator is bad at Kronecker-trace-estimation. arXiv preprint arXiv:2309.04952, 2023.
- Randomized contour integral methods for eigenvalue problems. Foundations of Computational Mathematics 2023 conference, 2023.
- Randomized numerical linear algebra: A perspective on the field with an eye to software. arXiv preprint arXiv:2302.11474v2, 2023.
- Rasmus Pagh. Compressed matrix multiplication. ACM Trans. Comput. Theory, 5(3):Art. 9, 17, 2013.
- Davide Palitta. Matrix equation techniques for certain evolutionary partial differential equations. J. Sci. Comput., 87(3):Paper No. 99, 36, 2021.
- Eric Polizzi. Density-matrix-based algorithm for solving eigenvalue problems. Phys. Rev. B, 79:115112, Mar 2009.
- Tensorized random projections. In Proceedings of the International Conference on Artificial Intelligence and Statistics, pages 3306–3316. PMLR, 2020.
- A projection method for generalized eigenvalue problems using numerical integration. J. Comput. Appl. Math., 159(1):119–128, 2003.
- Tamas Sarlos. Improved approximation algorithms for large matrices via random projections. In Proceedings of the 47th Annual IEEE Symposium on Foundations of Computer Science (FOCS), FOCS ’06, page 143–152, USA, 2006. IEEE Computer Society.
- Tensor random projection for low memory dimension reduction. In Proceedings of the 32nd Conference on Neural Information Processing Systems. PMLR, 2018.
- Ping Tak Peter Tang and Eric Polizzi. FEAST as a subspace iteration eigensolver accelerated by approximate spectral projection. SIAM J. Matrix Anal. Appl., 35(2):354–390, 2014.
- The exponentially convergent trapezoidal rule. SIAM Rev., 56(3):385–458, 2014.
- Ledyard R. Tucker. Some mathematical notes on three-mode factor analysis. Psychometrika, 31:279–311, 1966.
- Roman Vershynin. Introduction to the non-asymptotic analysis of random matrices. In Yonina C. Eldar and GittaEditors Kutyniok, editors, Compressed Sensing: Theory and Applications, pages 210–268. Cambridge University Press, 2012.
- Roman Vershynin. High-dimensional probability, volume 47 of Cambridge Series in Statistical and Probabilistic Mathematics. Cambridge University Press, Cambridge, 2018.
- Roman Vershynin. Concentration inequalities for random tensors. Bernoulli, 26(4):3139–3162, 2020.
- David P. Woodruff. Sketching as a tool for numerical linear algebra. Found. Trends Theor. Comput. Sci., 10(1-2):iv+157, 2014.
- Cluster robust estimates for block gradient-type eigensolvers. Math. Comp., 88(320):2737–2765, 2019.