Fuzzt Set-Based Kernel Extreme Learning Machine Autoencoder for Multi-Label Classification
Qingshuo Zhang, Eric C.C. Tsang, Meng Hu, Qiang He, Degang Chen · 2021
The multi-label learning algorithm based on an extreme learning machine has the advantage of high efficiency and generalization ability, but its classification ability is weak due to ignoring the correlation between features and labels. Accordingly, in this paper, the fuzzy set-based kernel extreme learning machine autoencoder for multi-label classification (KELM-AE-fuzzy) is proposed. Firstly, the correlation between features and labels is analyzed based on fuzzy set theory, and the correlation label membership matrix and label completion matrix are constructed. Then, the kernel extreme learning machine autoencoder is used to fuse the correlation label membership matrix with the original feature space and generate the reconstructed feature space. Eventually, kernel extreme learning machine (KELM) is used as a classifier, where the label matrix is used with the label completion matrix. Comparative experiments on several multi-label datasets demonstrate that KELM-AE-fuzzy outperforms other multi-label algorithms, and the effectiveness of the proposed algorithm is verified.