Learning Discriminative Features using Multi-label Dual Space

Multi-label learning handles instances associated with multiple class labels. The original label space is a logical matrix with entries from the Boolean domain ∈ {0, 1}. Logical labels are not able to show the relative importance of each semantic label to the instances. The vast majority of existing methods map the input features to the label space using linear projections with taking into consideration the label dependencies using logical label matrix. However, the discriminative features are learned using one-way projection from the feature representation of an instance into a logical label space. Given that there is no manifold in the learning space of logical labels, which limits the potential of learned models. In this work, inspired from a real-world example in image annotation to reconstruct an image from the label importance and feature weights. We propose a novel method in multi-label learning to learn the projection matrix from the feature space to semantic label space and projects it back to the original feature space using encoder-decoder deep learning architecture. The key intuition which guides our method is that the discriminative features are identified due to map the features back and forth using two linear projections. To the best of our knowledge, this is one of the first attempts to study the ability to reconstruct the original features from the label manifold in multi-label learning. We show that the learned projection matrix identifies a subset of discriminative features across multiple semantic labels. Extensive experiments on realworld datasets show the superiority of the proposed method.

[1]  Zhi-Hua Zhou,et al.  ML-KNN: A lazy learning approach to multi-label learning , 2007, Pattern Recognit..

[2]  Weiwei Liu,et al.  Discriminative and Correlative Partial Multi-Label Learning , 2019, IJCAI.

[3]  Richard H. Bartels,et al.  Algorithm 432 [C2]: Solution of the matrix equation AX + XB = C [F4] , 1972, Commun. ACM.

[4]  Xin Geng,et al.  Multi-Label Manifold Learning , 2016, AAAI.

[5]  Dawei Zhao,et al.  Multi-label learning with kernel extreme learning machine autoencoder , 2019, Knowl. Based Syst..

[6]  Ashish Ghosh,et al.  Multi-label classification using a cascade of stacked autoencoder and extreme learning machines , 2019, Neurocomputing.

[7]  Zhen Wang,et al.  Learning Low-Rank Label Correlations for Multi-label Classification with Missing Labels , 2014, 2014 IEEE International Conference on Data Mining.

[8]  Wei Liu,et al.  Correlated Multi-label Classification with Incomplete Label Space and Class Imbalance , 2019, ACM Trans. Intell. Syst. Technol..

[9]  Marc'Aurelio Ranzato,et al.  A Unified Energy-Based Framework for Unsupervised Learning , 2007, AISTATS.

[10]  Ju-Sheng Mi,et al.  A novel approach for learning label correlation with application to feature selection of multi-label data , 2020, Inf. Sci..

[11]  Min-Ling Zhang,et al.  A Review on Multi-Label Learning Algorithms , 2014, IEEE Transactions on Knowledge and Data Engineering.

[12]  Zhi-Hua Zhou,et al.  Multi-Label Learning with Global and Local Label Correlation , 2017, IEEE Transactions on Knowledge and Data Engineering.

[13]  Rui Huang,et al.  Manifold-based constraint Laplacian score for multi-label feature selection , 2018, Pattern Recognit. Lett..

[14]  William Zhu,et al.  Multi-label feature selection via feature manifold learning and sparsity regularization , 2018, Int. J. Mach. Learn. Cybern..

[15]  Gan Sun,et al.  Dual Relation Semi-Supervised Multi-Label Learning , 2020, AAAI.