Please use this identifier to cite or link to this item:
Title: LA-HCN: label-based attention for hierarchical multi-label text classification neural network
Authors: Zhang, Xinyi
Xu, Jiahao
Soh, Charlie
Chen, Lihui
Keywords: Engineering::Electrical and electronic engineering
Issue Date: 2022
Source: Zhang, X., Xu, J., Soh, C. & Chen, L. (2022). LA-HCN: label-based attention for hierarchical multi-label text classification neural network. Expert Systems With Applications, 187, 115922-.
Project: AISG-100E-2019-031
Journal: Expert Systems with Applications
Abstract: Hierarchical multi-label text classification (HMTC) has been gaining popularity in recent years thanks to its applicability to a plethora of real-world applications. The existing HMTC algorithms largely focus on the design of classifiers, such as the local, global, or a combination of them. However, very few studies have focused on hierarchical feature extraction and explore the association between the hierarchical labels and the text. In this paper, we propose a Label-based Attention for Hierarchical Multi-label Text Classification Neural Network (LA-HCN), where the novel label-based attention module is designed to hierarchically extract important information from the text based on the labels from different hierarchy levels. Besides, hierarchical information is shared across levels while preserving the hierarchical label-based information. Separate local and global document embeddings are obtained and used to facilitate the respective local and global classifications. In our experiments, LA-HCN outperforms other state-of-the-art neural network-based HMTC algorithms on four public HMTC datasets. The ablation study also demonstrates the effectiveness of the proposed label-based attention module as well as the novel local and global embeddings and classifications. By visualizing the learned attention (words), we find that LA-HCN is able to extract meaningful information corresponding to the different labels which provides explainability that may be helpful for the human analyst.
ISSN: 0957-4174
DOI: 10.1016/j.eswa.2021.115922
Schools: School of Electrical and Electronic Engineering 
Rights: © 2021 Elsevier Ltd. All rights reserved.
Fulltext Permission: none
Fulltext Availability: No Fulltext
Appears in Collections:EEE Journal Articles

Citations 20

Updated on Dec 4, 2023

Web of ScienceTM
Citations 20

Updated on Oct 25, 2023

Page view(s)

Updated on Dec 7, 2023

Google ScholarTM




Items in DR-NTU are protected by copyright, with all rights reserved, unless otherwise indicated.