Multi-label classification is the task of predicting a set of labels for a given input instance. Classifier chains are a state-of-the-art method for tackling such problems, which essentially converts this problem into a sequential prediction problem, where the labels are first ordered in an arbitrary fashion, and the task is to predict a sequence of binary values for these labels. In this paper, we replace classifier chains with recurrent neural networks, a sequence-to-sequence prediction algorithm which has recently been successfully applied to sequential prediction tasks in many domains. The key advantage of such an approach is that it allows to share parameters across all classifiers in the prediction chain, a key property of multi-target prediction problems. As both, classifier chains and recurrent neural networks depend on a fixed ordering of the labels, which is typically not part of a multi-label problem specification, we also compare different ways of ordering the label set, and give some recommendations on suitable ordering strategies.
Jinseok Nam (TU Darmstadt)
Eneldo Loza Mencía (Technische Universität Darmstadt)
Hyunwoo J Kim (University of Wisconsin-Madison)
Johannes Fürnkranz (TU Darmstadt)
Related Events (a corresponding poster, oral, or spotlight)
2017 Poster: Maximizing Subset Accuracy with Recurrent Neural Networks in Multi-label Classification »
Thu. Dec 7th 02:30 -- 06:30 AM Room Pacific Ballroom #42