Hierarchical Multi-label Text Classification: Self-adaption Semantic Awareness Network Integrating Text Topic and Label Level Information

Rui Zhao,Xiao Wei,Cong Ding,Yongqi Chen
DOI: https://doi.org/10.1007/978-3-030-82147-0_33
2021-01-01
Abstract:Hierarchical multi-label text classification is used to assign documents to multiple categories stored in a hierarchical structure. However, the existing methods pay more attention to the local semantic information of the text, and make insufficient use of the label level information. In this paper, we propose a new self-adaption semantic awareness network integrating text topic and label level information. Compared with the existing methods, our method has two contributions: (1) On the basis of the traditional text encoder, we add the global topic information of the text and the hierarchical information of the label, describe the text from different levels and effectively use the correlation between the labels. (2) We design an adaptive fusion strategy that adaptively retains information helpful for classification from three different levels of features and exploits their mutual influence. Experimental results on two benchmark datasets show that our model is better than the state-of-the-art methods on six metrics, especially in the prediction of low-frequency labels with sparse samples.
What problem does this paper attempt to address?