Convex and Semi-Nonnegative Matrix Factorizations
C. DingTao LiMichael I. Jordan
Introduces Semi-NMF and Convex-NMF to enable nonnegative matrix factorization on mixed-sign data and kernel spaces while establishing theoretical and practical connections to K-means clustering.
Organizations frequently analyze complex, high-dimensional datasets such as text, system logs, and scientific measurements to group related items and identify underlying patterns. While standard Nonnegative Matrix Factorization (NMF) offers interpretable, parts-based representations, its practical utility has been restricted because it requires all input data to be strictly nonnegative. Many common analytical workflows, however, rely on centered, normalized, or mixed-sign data where standard NMF fails. The article resolves this limitation by developing and evaluating new matrix factorization frameworks—specifically Semi-NMF and Convex-NMF—that accept mixed-sign data while maintaining the nonnegativity and interpretability of factor outputs.
The research demonstrates that these methods can be computed efficiently through iterative multiplicative updating algorithms with guaranteed convergence to local minima. To evaluate performance and practical utility, the article compares the new methods against standard NMF, traditional Singular Value Decomposition (SVD), and standard K-means clustering across synthetic datasets and seven real-world datasets spanning document collections, system log messages, and physical measurements.
The findings establish three principal results: First, all matrix factorization variants consistently outperformed standard K-means clustering in clustering accuracy across every tested dataset, improving accuracy by roughly 5 to 17 percentage points in real-world scenarios. Second, Semi-NMF and Convex-NMF successfully operated directly on mixed-sign data without requiring artificial data shifting, which was shown to degrade clustering accuracy and sparsity. Third, Convex-NMF naturally produced highly sparse indicator factors (reducing non-zero elements to roughly 49% to 64%) and generated basis factors that closely align with true cluster centroids, unlike Semi-NMF or SVD.
These results provide senior stakeholders and data science teams with more reliable and interpretable alternatives to standard clustering algorithms, especially for non-spherical data clusters where standard K-means fails. Additionally, Convex-NMF enables non-linear "kernelized" factorizations because its updates depend solely on inner products of data points. For operational deployments, technical teams should consider Semi-NMF when raw clustering accuracy on mixed-sign data is paramount, and Convex-NMF when human interpretability, sparsity, and centroid discovery are critical priorities. Future work should explore optimal initialization strategies and broader non-linear kernel applications across large-scale industrial datasets.
Decision-makers should note that, like K-means and Expectation-Maximization algorithms, these multiplicative updates guarantee convergence only to local rather than global optima. Nevertheless, repeated experimental runs demonstrate high stability and consistent clustering results across varied initializations, providing strong confidence in their practical application.
- Paper: Algorithms for Non-negative Matrix Factorization, Daniel D. Lee et al. (2000). It introduces the classical non-negative matrix factorization formulation and multiplicative update rules that this paper adapts and generalizes to unconstrained and convex basis settings.
- Paper: K-means clustering via principal component analysis, C. Ding et al. (2004). It establishes the foundational theoretical connections between matrix factorizations, low-rank approximations, and continuous relaxations of K-means clustering.
- Paper: Kernel k-means: spectral clustering and normalized cuts, Inderjit S. Dhillon et al. (2004). It provides the mathematical equivalence between kernel clustering and spectral graph cuts necessary for understanding how Convex NMF operates in non-linear kernel spaces.
- Paper: Document clustering based on non-negative matrix factorization, Wei Xu et al. (2003). It demonstrates how non-negative factor matrices directly provide cluster membership indicators, motivating the semi-nonnegative relaxation for clustering mixed-sign data.
- Paper: Concept Decompositions for Large Sparse Text Data Using Clustering, I. Dhillon et al. (2004). It introduces concept decompositions where basis vectors are restricted to convex combinations of data points, directly prefiguring the Convex NMF constraint.
- Paper: Non-negative Matrix Factorization with Sparseness Constraints, Patrik O. Hoyer (2004). It formalizes explicit sparsity analysis and constraints in matrix factorization algorithms, providing the baseline context for analyzing solution sparseness.
- Paper: Graph Regularized Nonnegative Matrix Factorization for Data Representation, Deng Cai et al. (2011). It extends matrix factorization representations by incorporating graph affinity regularizers to capture manifold geometry alongside parts-based decompositions.
- Paper: Online Learning for Matrix Factorization and Sparse Coding, Julien Mairal et al. (2010). It scales matrix factorization and sparse coding formulations, including non-negative variants, to massive streaming datasets via online stochastic optimization.
- Paper: Revisiting Frank-Wolfe: Projection-Free Sparse Convex Optimization, Martin Jaggi (2013). It develops projection-free conditional gradient algorithms for convex and low-rank matrix optimization problems with structured sparsity constraints.
