Deep extreme multi-label learning

Wenjie Zhang, Junchi Yan, Xiangfeng Wang, Hongyuan Zha

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

97 Scopus citations

Abstract

Extreme multi-label learning (XML) or classification has been a practical and important problem since the boom of big data. The main challenge lies in the exponential label space which involves 2L possible label sets especially when the label dimension L is huge, e.g., in millions for Wikipedia labels. This paper is motivated to better explore the label space by originally establishing an explicit label graph. In the meanwhile, deep learning has been widely studied and used in various classification problems including multi-label classification, however it has not been properly introduced to XML, where the label space can be as large as in millions. In this paper, we propose a practical deep embedding method for extreme multi-label classification, which harvests the ideas of non-linear embedding and graph priors-based label space modeling simultaneously. Extensive experiments on public datasets for XML show that our method performs competitive against state-of-the-art result.

Original languageEnglish
Title of host publicationICMR 2018 - Proceedings of the 2018 ACM International Conference on Multimedia Retrieval
PublisherAssociation for Computing Machinery, Inc
Pages100-107
Number of pages8
ISBN (Print)9781450350464
DOIs
StatePublished - 5 Jun 2018
Event8th ACM International Conference on Multimedia Retrieval, ICMR 2018 - Yokohama, Japan
Duration: 11 Jun 201814 Jun 2018

Publication series

NameICMR 2018 - Proceedings of the 2018 ACM International Conference on Multimedia Retrieval

Conference

Conference8th ACM International Conference on Multimedia Retrieval, ICMR 2018
Country/TerritoryJapan
CityYokohama
Period11/06/1814/06/18

Keywords

  • Deep embedding
  • Extreme classification
  • Extreme multi-label learning

Fingerprint

Dive into the research topics of 'Deep extreme multi-label learning'. Together they form a unique fingerprint.

Cite this