An effective single-model learning for multi-label data

Abstract

Multi-label data classification (MLC) has become an increasingly active research area over the past decade. MLC refers to a classification problem where each instance can be associated with more than one class label. Capturing the correlation among labels and tackling the label imbalance are the main challenges in MLC. Problem transformation is one of the well-known approaches in this area that became the de-facto approach for MLC. Existing methods in this approach consider MLC as a collection of single-label tasks and solve each of them separately. To consider correlation among labels, some of them consider the combination of labels that appear in the training data as a separate label. The main drawback of these kinds of methods is the complexity of the model, which makes them not applicable in real-world applications. In this paper, we show how MLC can be efficiently and effectively tackled with a single classifier. Our proposed method maps the training data into a new sub-space for each label. Then, it pools all the mapped data together and efficiently trains a single classifier for all the labels together. Experimental results show that our method successfully tackles MLC tasks and outperforms the state-of-the-art methods.