Ensemble methods for multi-label classification

Lior Rokach, Alon Schclar, Ehud Itach

Research output: Contribution to journalArticlepeer-review

60 Scopus citations

Abstract

Ensemble methods have been shown to be an effective tool for solving multi-label classification tasks. In the RAndom k-labELsets (RAKEL) algorithm, each member of the ensemble is associated with a small randomly-selected subset of k labels. Then, a single label classifier is trained according to each combination of elements in the subset. In this paper we adopt a similar approach, however, instead of randomly choosing subsets, we select the minimum required subsets of k labels that cover all labels and meet additional constraints such as coverage of inter-label correlations. Construction of the cover is achieved by formulating the subset selection as a minimum set covering problem (SCP) and solving it by using approximation algorithms. Every cover needs only to be prepared once by offline algorithms. Once prepared, a cover may be applied to the classification of any given multi-label dataset whose properties conform with those of the cover. The contribution of this paper is two-fold. First, we introduce SCP as a general framework for constructing label covers while allowing the user to incorporate cover construction constraints. We demonstrate the effectiveness of this framework by proposing two construction constraints whose enforcement produces covers that improve the prediction performance of random selection by achieving better coverage of labels and inter-label correlations. Second, we provide theoretical bounds that quantify the probabilities of random selection to produce covers that meet the proposed construction criteria. The experimental results indicate that the proposed methods improve multi-label classification accuracy and stability compared to the RAKEL algorithm and to other state-of-the-art algorithms.

Original languageEnglish
Pages (from-to)7507-7523
Number of pages17
JournalExpert Systems with Applications
Volume41
Issue number16
DOIs
StatePublished - 15 Nov 2014

Keywords

  • Ensemble learning
  • Multi-label classification

Fingerprint

Dive into the research topics of 'Ensemble methods for multi-label classification'. Together they form a unique fingerprint.

Cite this