Semi-supervised non-negative matrix decomposition (SNMF) is a highly interpretable image clustering algorithm. We typically incorporate graph learning to enhance the model’s representation ability, while adding positive and negative labeling information and pairwise constraints to prevent overfitting and improve the discriminative performance. However, embedding positive and negative labels and pairwise constraints simultaneously in the SNMF model is challenging due to their dimension mismatch. Thus We propose a method named multi-constraint fusion based semi-supervised non-negative matrix decomposition (MCFSNMF), which effectively solves the above problem by constructing symmetric labeling matrices embedded with positive and negative labeling information and fusing them with pairwise constraint matrices. In addition, SNMF is essentially a feature extraction method, which cannot obtain clustering results directly. Therefore, we combine the label propagation algorithm to achieve feature learning and label assignment interaction, which not only avoids the impact of clustering post-processing operations but also enhances the representation of labeled data relative to unlabeled data. We compare multiple SNMF algorithms on six image datasets and show that the proposed algorithm can effectively enhance the clustering performance and validate the algorithm’s convergence.
Read full abstract