.png){kind=icon}
.png){kind=icon}
Return
Adaptive Hierarchical Text Classification Using ERNIE and Dynamic Threshold Pruning
.png){kind=icon}
OA
AI
DOI：10.1109/ACCESS.2024.3519954.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Hierarchical Text Classification (HTC) is a challenging task where labels are structured in a tree or Directed Acyclic Graph (DAG) format. Current approaches often struggle with data imbalance and fail to fully capture rich semantic information. This paper proposes an Adaptive Hierarchical Text Classification method, EDTPA (ERNIE and Dynamic Threshold Pruning-based Adaptive classification), which leverages Large Language Models (LLMs) for data augmentation to mitigate imbalanced datasets. The model first uses Graph Attention Networks (GAT) to capture hierarchical dependencies among labels, effectively modeling structured relationships. ERNIE enhances the semantic representation of both the text and hierarchical labels, optimizing the model's ability to process Chinese text. An attention mechanism strengthens the alignment between text and labels, improving accuracy. The model combines global and local information flows, while dynamic threshold pruning prunes low-probability branches, improving interpretability. Results on the Chinese Scientific Literature (CSL) dataset show EDTPA significantly outperforms baseline models in both Micro-F1 and Macro-F1 scores, effectively addressing data imbalance and improving classification performance.
Keywords:
Hierarchical text classification
data augmentation
large language model
graph attention network
ERNIE
Hierarchical text classification
data augmentation
large language model
graph attention network
ERNIE
Journal
IF:
3.6
Papers:
9.8W
Citations:
29.4W
Organization
No organization information available