A Generalized Deep Learning Clustering Algorithm Based on Non-Negative Matrix Factorization
Dexian Wang,Tianrui Li,Ping Deng,Fan Zhang,Wei Huang,Pengfei Zhang,Jia Liu
DOI: https://doi.org/10.1145/3584862
IF: 4.157
2023-02-20
ACM Transactions on Knowledge Discovery from Data
Abstract:Clustering is a popular research topic in the field of data mining, in which the clustering method based on non-negative matrix factorization (NMF) has been widely employed. However, in the update process of NMF, there is no learning rate to guide the update as well as the update depends on the data itself, which leads to slow convergence and low clustering accuracy. To solve these problems, a generalized deep learning clustering (GDLC) algorithm based on NMF is proposed in this paper. Firstly, a nonlinear constrained NMF (NNMF) algorithm is constructed to achieve sequential update of the elements in the matrix guided by the learning rate. Then the gradient values corresponding to the element update are transformed into generalized weights and generalized biases, by inputting the elements as well as their corresponding generalized weights and generalized biases into the nonlinear activation function to construct the GDLC algorithm. In addition, for improving the understanding of the GDLC algorithm, its detailed inference procedure and algorithm design are provided. Finally, the experimental results on eight datasets show that the GDLC algorithm has efficient performance.
computer science, information systems, software engineering