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 language | English |
---|---|
Pages (from-to) | 85-90 |
Number of pages | 6 |
Journal | Pattern Recognition Letters |
Volume | 42 |
DOIs | |
Publication status | Published - 1 Jun 2014 |
Externally published | Yes |
Keywords
- Multi-label learning
- Label dependency
- Random walk with restart
- Label propagation