March 5, 2025
Journal Article

Randomized Algorithms for Symmetric Nonnegative Matrix Factorization

Abstract

Symmetric Nonnegative Matrix Factorization (SymNMF) is a technique in data analysis and machine learning that approximates a matrix with a product of a nonnegative, low-rank matrix and it transpose. To design faster and more scalable algorithms for SymNMF we develop two randomized algorithms for its computation. The first method uses randomized matrix sketching to compute an initial low-rank approximation to the input matrix and proceeds to uses this as a low-rank input to rapidly compute a SymNMF. The second methods uses randomized leverage score sampling to approximately solve constrained least squares problems. Many successful methods for SymNMF rely on (approximately) solving sequences of constrained least squares problems. We prove theoretically that leverage score sampling can approximately solve constrained least squares problems to e-accuracy. Finally we demonstrate both methods work in practice by applying them to graph clustering tasks on large real world data sets. These experiments show that our methods approximately maintain solution quality and achieve significant speed ups for both large dense and large sparse problems.

Published: March 5, 2025

Citation

Hayashi K., S.G. Aksoy, G. Ballard, H. Park, and H. Park. 2025. Randomized Algorithms for Symmetric Nonnegative Matrix Factorization. SIAM Journal on Matrix Analysis and Applications 46, no. 1:584 - 625. PNNL-SA-193926. doi:10.1137/24M1638355