Joint Embedding of Words and Category Labels for Hierarchical Multi-label Text Classification

by   Jingpeng Zhao, et al.

Text classification has become increasingly challenging due to the continuous refinement of classification label granularity and the expansion of classification label scale. To address that, some research has been applied onto strategies that exploit the hierarchical structure in problems with a large number of categories. At present, hierarchical text classification (HTC) has received extensive attention and has broad application prospects. Making full use of the relationship between parent category and child category in text classification task can greatly improve the performance of classification. In this paper, We propose a joint embedding of text and parent category based on hierarchical fine-tuning ordered neurons LSTM (HFT-ONLSTM) for HTC. Our method makes full use of the connection between the upper-level and lower-level labels. Experiments show that our model outperforms the state-of-the-art hierarchical model at a lower computation cost.



There are no comments yet.



Joint Input-Label Embedding for Neural Text Classification

Neural text classification methods typically treat output classes as cat...

Hierarchy-Aware T5 with Path-Adaptive Mask Mechanism for Hierarchical Text Classification

Hierarchical Text Classification (HTC), which aims to predict text label...

Description Based Text Classification with Reinforcement Learning

The task of text classification is usually divided into two stages: tex...

HTCInfoMax: A Global Model for Hierarchical Text Classification via Information Maximization

The current state-of-the-art model HiAGM for hierarchical text classific...

Multi-label Dataless Text Classification with Topic Modeling

Manually labeling documents is tedious and expensive, but it is essentia...

Unsupervised Label Refinement Improves Dataless Text Classification

Dataless text classification is capable of classifying documents into pr...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.