面向类别不平衡的多标签学习

Towards Class-Imbalance Aware Multi-Label Learning

IEEE Transactions on Cybernetics · 2020
被引 68
ABS 3

中文导读

提出一种名为COCOA的简单有效策略,通过同时利用标签相关性和探索类别不平衡,为每个标签训练多个多类不平衡学习器并聚合预测,以提升多标签学习在不平衡数据上的性能。

Abstract

Multi-label learning deals with training examples each represented by a single instance while associated with multiple class labels. Due to the exponential number of possible label sets to be considered by the predictive model, it is commonly assumed that label correlations should be well exploited to design an effective multi-label learning approach. On the other hand, class-imbalance stands as an intrinsic property of multi-label data which significantly affects the generalization performance of the multi-label predictive model. For each class label, the number of training examples with positive labeling assignment is generally much less than those with negative labeling assignment. To deal with the class-imbalance issue for multi-label learning, a simple yet effective class-imbalance aware learning strategy called cross-coupling aggregation (COCOA) is proposed in this article. Specifically, COCOA works by leveraging the exploitation of label correlations as well as the exploration of class-imbalance simultaneously. For each class label, a number of multiclass imbalance learners are induced by randomly coupling with other labels, whose predictions on the unseen instance are aggregated to determine the corresponding labeling relevancy. Extensive experiments on 18 benchmark datasets clearly validate the effectiveness of COCOA against state-of-the-art multi-label learning approaches especially in terms of imbalance-specific evaluation metrics.

机器学习多标签分类类别不平衡计算机科学