Skip to main navigation Skip to search Skip to main content

Hierarchical Taxonomy-Aware and Attentional Graph Capsule RCNNs for Large-Scale Multi-Label Text Classification

  • Hao Peng
  • , Jianxin Li*
  • , Senzhang Wang
  • , Lihong Wang
  • , Qiran Gong
  • , Renyu Yang
  • , Bo Li
  • , Philip S. Yu
  • , Lifang He
  • *Corresponding author for this work
  • Nanjing University of Aeronautics and Astronautics
  • National Computer Network Emergency Response Technical Team
  • Beihang University
  • University of Leeds
  • University of Illinois at Chicago
  • Lehigh University

Research output: Contribution to journalArticlepeer-review

Abstract

CNNs, RNNs, GCNs, and CapsNets have shown significant insights in representation learning and are widely used in various text mining tasks such as large-scale multi-label text classification. Most existing deep models for multi-label text classification consider either the non-consecutive and long-distance semantics or the sequential semantics. However, how to coherently take them into account is still far from studied. In addition, most existing methods treat output labels as independent medoids, ignoring the hierarchical relationships among them, which leads to a substantial loss of useful semantic information. In this paper, we propose a novel hierarchical taxonomy-aware and attentional graph capsule recurrent CNNs framework for large-scale multi-label text classification. Specifically, we first propose to model each document as a word order preserved graph-of-words and normalize it as a corresponding word matrix representation preserving both the non-consecutive, long-distance and local sequential semantics. Then the word matrix is input to the proposed attentional graph capsule recurrent CNNs for effectively learning the semantic features. To leverage the hierarchical relations among the class labels, we propose a hierarchical taxonomy embedding method to learn their representations, and define a novel weighted margin loss by incorporating the label representation similarity. Extensive evaluations on three datasets show that our model significantly improves the performance of large-scale multi-label text classification by comparing with state-of-the-art approaches.

Original languageEnglish
Article number8933476
Pages (from-to)2505-2519
Number of pages15
JournalIEEE Transactions on Knowledge and Data Engineering
Volume33
Issue number6
DOIs
StatePublished - 1 Jun 2021

Keywords

  • Multi-label classification
  • attention network
  • capsule network
  • document modeling
  • graph rcnn
  • taxonomy embedding

Fingerprint

Dive into the research topics of 'Hierarchical Taxonomy-Aware and Attentional Graph Capsule RCNNs for Large-Scale Multi-Label Text Classification'. Together they form a unique fingerprint.

Cite this