Multi-Label classification with Missing Labels by Preserving Feature-Label Space Consistency
Zan Zhang, Depeng Zhang, Gongqing Wu · 2023
Multi-label classification deals with the problem where an instance is associated with multiple labels simultaneously. Most existing multi-label classification algorithms assume that the labels of the training data are complete. However, we can obtain only a partial label set of each instance in some real applications since labelling data is difficult or costly. Some existing works on multi-label classification with missing labels focus on exploiting label correlations to complete the original label space and simultaneously build a multi-label learning model using label specific features. However, these methods may be suboptimal since they do not preserve feature-label space consistency. In this paper, we propose a Space Consistency-based Multi-Label classification algorithm named SCML to address this issue. First, label correlation in label space is learned to augment the incomplete original label matrix to a new supplementary label matrix, and the multi-label classifier is constructed simultaneously based on the new supplementary label matrix. Then, correlation information in feature space is learned based on the probabilistic neighborhood similarities to preserve feature-label space consistency. Moreover, the proposed algorithm has an effective mechanism for learning label-specific features to improve the multi-label classification with missing labels. Extensive experiments on twelve benchmark data sets validate the effectiveness of the proposed approach for improving the generalization performance of state-of-the-art algorithms of multi-label learning with missing labels.