[BibTeX] [RIS]
Predicting Unseen Labels using Label Hierarchies in Large-Scale Multi-label Learning
Type of publication: Inproceedings
Citation: nam15wsabieH
Booktitle: Proceedings of the European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases
Year: 2015
Pages: 102-118
Publisher: Springer International Publishing
Location: Porto, Portugal
ISSN: 0302-9743
ISBN: 978-3-319-23527-1
URL: http://www.ke.tu-darmstadt.de/publications/papers/ECML2015Nam.pdf
DOI: 10.1007/978-3-319-23528-8_7
Abstract: An important problem in multi-label classification is to capture label patterns or underlying structures that have an impact on such patterns. One way of learning underlying structures over labels is to project both instances and labels into the same space where an instance and its relevant labels tend to have similar representations. In this paper, we present a novel method to learn a joint space of instances and labels by leveraging a hierarchy of labels. We also present an efficient method for pretraining vector representations of labels, namely label embeddings, from large amounts of label co-occurrence patterns and hierarchical structures of labels. This approach also allows us to make predictions on labels that have not been seen during training. We empirically show that the use of pretrained label embeddings allows us to obtain higher accuracies on unseen labels even when the number of labels are quite large. Our experimental results also demonstrate qualitatively that the proposed method is able to learn regularities among labels by exploiting a label hierarchy as well as label co-occurrences.
Authors Nam, Jinseok
Loza Mencía, Eneldo
Kim, Hyunwoo J.
Fürnkranz, Johannes