Multi-label learning based on iterative label propagation over graph

Bin Fu, Zhihai Wang*, Guandong Xu, Longbing Cao

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

9 Citations (Scopus)

Abstract

One key challenge in multi-label learning is how to exploit label dependency effectively, and existing methods mainly address this issue via training a prediction model for each label based on the combination of original features and the labels on which it depends on. However, the influence of label dependency might be depressed due to the significant imbalance in dimensionality of feature set and dependent label set in this way, also the dynamic interaction between labels cannot be utilized effectively. In this paper, we propose a new framework to exploit the dependencies between labels iteratively and interactively. Every label's prediction will be updated through iterative process of propagation, other than being determined directly by a prediction model. Specifically, we utilize a graph model to encode the dependencies between labels, and employ the random-walk with restart (RWR) strategy to propagate the dependency among all labels iteratively until the predictions for all the labels converge. We validate our approach by experiments, and the results demonstrate that it yields significant improvements compared with several state-of-the-art algorithms.

Original languageEnglish
Pages (from-to)85-90
Number of pages6
JournalPattern Recognition Letters
Volume42
DOIs
Publication statusPublished - 1 Jun 2014
Externally publishedYes

Keywords

  • Multi-label learning
  • Label dependency
  • Random walk with restart
  • Label propagation

Fingerprint

Dive into the research topics of 'Multi-label learning based on iterative label propagation over graph'. Together they form a unique fingerprint.

Cite this