Constrained Nonnegative Matrix Factorization Based on Label Propagation for Data Representation

Junmin Liu,Yicheng Wang,Jing Ma,Di Han,Yifan Huang
DOI: https://doi.org/10.1109/tai.2023.3267029
2024-01-01
IEEE Transactions on Artificial Intelligence
Abstract:Nonnegative matrix factorization (NMF) algorithms are a series of dimensional reduction techniques widely used in data preprocessing. To improve the performance of clustering and the discrimination of the low-dimensional representation in NMF, we proposed a novel semi-supervised constrained nonnegative matrix factorization based on label propagation (LpCNMF). Specifically, the proposed LpCNMF adopts graph and label propagation as regularization terms, then makes use of a small amount of labeled data to predict the label information of the unlabeled data and finally obtains a predictive membership matrix with more label information. At the same time, we introduce an efficient alternating iterative algorithm to solve the optimization problem of the objective function in the LpCNMF. Unlike other NMF algorithms that only update the basis and coefficient matrices, the LpCNMF algorithm increases the update of the predictive membership matrix obtained by label propagation. Experimental results on various benchmark datasets demonstrate the superiority of our algorithm over existing state-of-the-art NMF algorithms. The code is publicly available at https://github.com/ycwang-libra/LpCNMF_github.git.
What problem does this paper attempt to address?