Enhancing Label Correlations in multi-label classification through global-local label specific feature learning to Fill Missing labels

Abstract

In multi-label classification, challenges arise from missing labels due to subjective analysis or label ambiguity. This makes it difficult to accurately capture label correlations and enhance classifier performance. Previous research has primarily focused on obtaining label correlations with sparsity assumptions from incomplete label matrices to fill in these missing labels. However, this approach leads to inaccurate label correlations since it relies on incomplete label matrix information. To overcome this issue and improve accuracy, this paper proposes a new method called LCFM, which effectively learns label correlations using a combination of global and local label-specific features to recover missing labels. Firstly, we rely on the sparsity assumption of Lasso regression coefficients to select label-specific features from the training samples. Secondly, we introduce l2-norm regularization to learn global label correlations through global label-specific features. simultaneously, we utilize similar label-specific features to describe local label correlations. Finally, we minimize the difference between predicted results and true labels using the binary cross-entropy loss function and employ the Accelerated Proximal Gradient (APG) algorithm for model optimization. Experimental results comparing LCFM with five benchmark indicators on nine datasets demonstrate that our proposed method outperforms other advanced methods, showing strong competitiveness.