Persistent Nonnegative Matrix Factorization via Multi-Scale Graph Regularization
Organizations: School of Mathematics and Statistics, Xi’an Jiaotong University, Xi’an, Shaanxi Province, 710049, China
Abstract
Matrix factorization techniques, especially Nonnegative Matrix Factorization (NMF), have been widely used for dimensionality reduction and interpretable data representation. However, existing NMF-based methods are inherently single-scale and fail to capture the evolution of connectivity structures across resolutions. In this work, we propose persistent nonnegative matrix factorization (pNMF), a scale-parameterized family of NMF problems, that produces a sequence of persistence-aligned embeddings rather than a single one. By leveraging persistent homology, we identify a canonical minimal sufficient scale set at which the underlying connectivity undergoes qualitative changes. These canonical scales induce a sequence of graph Laplacians, leading to a coupled NMF formulation with scale-wise geometric regularization and explicit cross-scale consistency constraint. We analyze the structural properties of the embeddings along the scale parameter and establish bounds on their increments between consecutive scales. The resulting model defines a nontrivial solution path across scales, rather than a single factorization, which poses new computational challenges. We develop a sequential alternating optimization algorithm with guaranteed convergence. Numerical experiments on synthetic and single-cell RNA sequencing datasets demonstrate the effectiveness of the proposed approach in multi-scale low-rank embeddings.
Figures & tables
| Dataset | Metric | NMF | GNMF | TNMF | -TNMF | pNMF | pNMF-best |
|---|---|---|---|---|---|---|---|
| GSE75748time | ARI | 0.611 | 0.602 | 0.644 | 0.592 | 0.722 | 0.727 |
| NMI | 0.735 | 0.731 | 0.768 | 0.711 | 0.854 | 0.860 | |
| Purity | 0.745 | 0.720 | 0.749 | 0.751 | 0.850 | 0.852 | |
| Accuracy | 0.693 | 0.679 | 0.744 | 0.736 | 0.850 | 0.852 | |
| GSE94820 | ARI | 0.547 | 0.615 | 0.617 | 0.607 | 0.629 | 0.629 |
| NMI | 0.665 | 0.710 | 0.705 | 0.699 | 0.724 | 0.724 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
Non-Negative Matrix Factorization
Non-negative matrix factorization (NMF) decomposes nonnegative data into additive, interpretable latent components for analysis and discovery.
See all 33 papers