Multi-label learning with kernel extreme learning machine autoencoder

作者:

Highlights:

摘要

In multi-label learning, in order to improve the accuracy of classification, many scholars have considered the relationship between features and features, features and labels or labels and labels, but how to combine the correlation among them is rarely studied. Based on this, this paper proposes a multi-label learning algorithm with kernel extreme learning machine autoencoder. Firstly, the label space is reconstructed by using the non-equilibrium labels completion method in the label space. Then, the non-equilibrium labels space information is added to the input node of the kernel extreme learning machine autoencoder network, and the input features are output as the target. Finally, the kernel extreme learning machine is used for classification. Our method implements the information fusion between features and features, between labels and features, and between labels and labels. Compared with the traditional autoencoder network, the extreme learning machine autoencoder has no iterative process, which reduces the network training time and improves the classification accuracy. The experimental results of the proposed algorithm in the opening benchmark multi-label data sets show that the KELM-AE algorithm has some advantages over other comparative multi-label learning algorithms and the statistical hypothesis testing and stability analysis further illustrate the effectiveness of the proposed algorithm.

论文关键词:Multi-label learning,Extreme learning machine,Autoencoder,Non-equilibrium labels completion,Information entropy,Labels correlations

论文评审过程:Received 15 September 2018, Revised 1 April 2019, Accepted 4 April 2019, Available online 8 May 2019, Version of Record 4 June 2019.

论文官网地址:https://doi.org/10.1016/j.knosys.2019.04.002