Hierarchical Classification Based on Label Distribution Learning

Authors

  • Changdong Xu Southeast University
  • Xin Geng Southeast University

DOI:

https://doi.org/10.1609/aaai.v33i01.33015533

Abstract

Hierarchical classification is a challenging problem where the class labels are organized in a predefined hierarchy. One primary challenge in hierarchical classification is the small training set issue of the local module. The local classifiers in the previous hierarchical classification approaches are prone to over-fitting, which becomes a major bottleneck of hierarchical classification. Fortunately, the labels in the local module are correlated, and the siblings of the true label can provide additional supervision information for the instance. This paper proposes a novel method to deal with the small training set issue. The key idea of the method is to represent the correlation among the labels by the label distribution. It generates a label distribution that contains the supervision information of each label for the given instance, and then learns a mapping from the instance to the label distribution. Experimental results on several hierarchical classification datasets show that our method significantly outperforms other state-of-theart hierarchical classification approaches.

Downloads

Published

2019-07-17

How to Cite

Xu, C., & Geng, X. (2019). Hierarchical Classification Based on Label Distribution Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 5533-5540. https://doi.org/10.1609/aaai.v33i01.33015533

Issue

Section

AAAI Technical Track: Machine Learning