ML-FOREST: A multi-label tree ensemble method for multi-label classification

Qingyao Wu*, Mingkui Tan, Hengjie Song, Jian Chen, Kwok Po NG

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

50 Citations (Scopus)

Abstract

Multi-label classification deals with the problem where each example is associated with multiple class labels. Since the labels are often dependent to other labels, exploiting label dependencies can significantly improve the multi-label classification performance. The label dependency in existing studies is often given as prior knowledge or learned from the labels only. However, in many real applications, such prior knowledge may not be available, or labeled information might be very limited. In this paper, we propose a new algorithm, called Ml-Forest , to learn an ensemble of hierarchical multi-label classifier trees to reveal the intrinsic label dependencies. In Ml-Forest, we construct a set of hierarchical trees, and develop a label transfer mechanism to identify the multiple relevant labels in a hierarchical way. In general, the relevant labels at higher levels of the trees capture more discriminable label concepts, and they will be transferred into lower level children nodes that are harder to discriminate. The relevant labels in the hierarchy are then aggregated to compute label dependency and make the final prediction. Our empirical study shows encouraging results of the proposed algorithm in comparison with the state-of-the-art multi-label classification algorithms under Friedman test and post-hoc Nemenyi test.

Original languageEnglish
Article number7492171
Pages (from-to)2665-2680
Number of pages16
JournalIEEE Transactions on Knowledge and Data Engineering
Volume28
Issue number10
DOIs
Publication statusPublished - Oct 2016

Scopus Subject Areas

  • Information Systems
  • Computer Science Applications
  • Computational Theory and Mathematics

User-Defined Keywords

  • ensemble methods
  • label dependency
  • label transfer
  • Multi-label classification
  • tree classifier

Fingerprint

Dive into the research topics of 'ML-FOREST: A multi-label tree ensemble method for multi-label classification'. Together they form a unique fingerprint.

Cite this