Hierarchical Text Classification Based on a Hierarchical Probability-Aware Balanced Model
This paper tackles the challenges of poor class separability, complex structural dependencies, and severe class imbalance in large-scale hierarchical text classification. To address these issues, we propose a Hierarchical Probability-Aware Balancing (HiPAB) model that jointly incorporates representation enhancement, structural dependency modeling, and probabilistic decision balancing. HiPAB improves fine-grained category discrimination by capturing hierarchical relationships and mitigating the impact of imbalanced data distributions. Extensive experiments show that HiPAB consistently outperforms strong baselines across multiple metrics, demonstrating superior robustness and generalization, particularly in deep-level nodes and long-tail categories. These results highlight HiPAB as an effective and scalable solution for real-world hierarchical text classification tasks.
Authors
- Yixuan Song
- Qingwu Fan (ORCID: https://orcid.org/0000-0001-6900-7823)
- Xinyue Guo
Institutions
- Beijing University of Technology (CN)
- Beijing Information Science & Technology University (CN)
Publication Details
- Published
- 2025-06-20
- DOI
- https://doi.org/10.1109/caibda65784.2025.11183054
- Primary Topic
- Text and Document Classification Technologies
- Type
- article
- Field-Weighted Citation Impact
- 0.00