Graph Regularized Nonnegative Matrix Factorization for Data Representation
Deng CaiXiaofei HeJiawei HanThomas S. Huang
Proposes a graph-regularized nonnegative matrix factorization algorithm that preserves the intrinsic geometric structure of high-dimensional data by incorporating a nearest-neighbor graph into parts-based representation learning.
Modern data analysis tasks in pattern recognition, computer vision, and information retrieval routinely handle high-dimensional datasets such as images and text corpora. Standard Non-negative Matrix Factorization has gained widespread use because its additive constraints naturally learn parts-based, interpretable representations of data. However, traditional factorization methods assume that data resides in a flat Euclidean space, failing to capture the underlying geometric and manifold structures where nearby data points should share similar representations.
To address this limitation, the article introduces and evaluates Graph Regularized Non-negative Matrix Factorization. This framework incorporates an affinity graph to model the local geometric manifold of the data space into the matrix factorization process, enforcing the principle that points close to each other in the original space remain close in the reduced representation space.
The researchers developed iterative multiplicative optimization rules with proven convergence and evaluated the method against standard matrix factorization, spectral clustering, and subspace techniques across three benchmark datasets: the COIL20 object image library, the CMU PIE facial database, and the TDT2 text corpus comprising 9,394 documents. Performance was measured primarily using clustering accuracy and normalized mutual information across various cluster numbers and neighborhood configurations.
The experimental findings show that the proposed regularized approach consistently outperforms all baseline methods across every tested dataset. On the COIL20 image dataset, average accuracy reached 82.5 percent compared to 68.9 percent for standard factorization and 76.2 percent for spectral clustering. On the CMU PIE facial database, average accuracy reached 77.4 percent compared to 63.3 percent for baseline factorization and 25.9 percent for standard k-means. On the TDT2 document corpus, the method achieved an average accuracy of 92.0 percent versus 80.4 percent for standard non-negative factorization and 67.7 percent for singular value decomposition. Furthermore, the learned basis representations demonstrated higher sparsity and visual interpretability, and the optimization converged rapidly, typically within 100 iterations.
These results demonstrate that capturing intrinsic data geometry while enforcing non-negativity significantly improves representation quality and downstream clustering performance. Practitioners can achieve higher clustering precision and topic separation with minimal added computational overhead due to the sparsity of the nearest neighbor graph. The approach also offers a flexible framework that can incorporate domain knowledge such as class labels or network structures.
Organizations implementing this method should select weighting schemes tailored to data types, such as dot-product cosine weighting for text and heat-kernel weighting for images, while keeping the neighbor parameter low to avoid connecting disparate data categories. Future work should focus on automating parameter selection, particularly for regularization strength and heat kernel width, and adapting the method to capture angular similarity rather than purely Euclidean distances.
- Paper: Algorithms for Non-negative Matrix Factorization, Daniel D. Lee et al. (2000). Read this foundational account of NMF’s multiplicative updates and convergence first; the source adapts that factorization framework by adding graph-based regularization.
- Paper: Laplacian Eigenmaps and Spectral Techniques for Embedding and Clustering, Mikhail Belkin et al. (2001). Its construction of a neighborhood graph and Laplacian to preserve local manifold structure provides the geometric machinery the source incorporates into NMF.
- Paper: Graph Embedding and Extensions: A General Framework for Dimensionality Reduction, Shuicheng Yan et al. (2007). This graph-embedding framework clarifies how an affinity graph encodes desired relationships between data points, the principle the source uses to regularize learned representations.
- Paper: Document clustering based on non-negative matrix factorization, Wei Xu et al. (2003). Its application of NMF to document clustering, including graph-based weighting, gives useful prior context for the source’s graph-regularized factorization experiments.
No sufficiently relevant recommendations were found.
