Abstract:Multi-label learning has garnered much research interest due to its wide range of real-world applications. Many multi-label learning methods have been proposed; however, few have addressed the class imbalance problem existing in multi-label data. Even though some studies have taken this issue into account, most of them have ignored the label correlations or only considered random correlations between them. In this study, we propose a novel partition-based imbalanced multi-label learning algorithm, named Multi-label Learning based on Hierarchical Clustering (MLHC), to tackle this problem. MLHC first carries out hierarchical clustering on the original label space to divide it into several disconnected subspaces, each of which contains several labels that are strongly correlated with each other. Then, for each label subspace, we use the problem transformation strategy to convert it into a multi-class problem by binary coding. Any multi-class imbalance learning algorithm can be applied to the transformed multi-class data. Finally, the classification results will be decoded to retrieve the corresponding label subspace, and all label subspace results are combined to show the predicted label vector in the original label space. We conducted experiments not only on thirteen benchmark multi-label datasets but also carried out them on XJTU-SY which is a multi-label engineering application dataset, and the results indicated that our proposed MLHC learning algorithm outperforms several state-of-the-art class imbalance multi-label learning algorithms, demonstrating the effectiveness and necessity of discovering label correlations and transforming the original imbalanced multi-label learning problem into multiple strongly correlated multi-class imbalanced learning problems.

Towards Label Imbalance in Multi-label Classification with Many Labels

Multi-label Classification: Dealing with Imbalance by Combining Labels

Towards Imbalanced Large Scale Multi-label Classification with Partially Annotated Labels

Multi-label sampling based on local label imbalance

Learning sample representativeness for class-imbalanced multi-label classification

Rebalancing Multi-Label Class-Incremental Learning

A partition-based problem transformation algorithm for classifying imbalanced multi-label data

Large-Scale Multi-Label Learning with Incomplete Label Assignments

A Divide-and-Conquer Approach for Large-Scale Multi-label Learning

Scalable Label Distribution Learning for Multi-Label Classification

Addressing the multi-label imbalance for neural networks: An approach based on stratified mini-batches

A General Framework for Multi-Label Learning Towards Class Correlations and Class Imbalance.

Balanced Masking Strategy for Multi-Label Image Classification.

Multi-Label Learning with Block Diagonal Labels

Graph-Based Class-Imbalance Learning With Label Enhancement

An Effective Approach for Multi-label Classification with Missing Labels

Multi-label Text Classification Model Based on Multi-level Constraint Augmentation and Label Association Attention

On the Value of Head Labels in Multi-Label Text Classification

When Noisy Labels Meet Long Tail Dilemmas: A Representation Calibration Method

Cost Sensitive Ranking Support Vector Machine for Multi-label Data Learning.

PLM: Partial Label Masking for Imbalanced Multi-label Classification