Utilizing Local Hierarchy with Adversarial Training for Hierarchical Text Classification

Zihan Wang, Peiyi Wang, Houfeng Wang


Abstract
Hierarchical text classification (HTC) is a challenging subtask of multi-label classification due to its complex taxonomic structure. Nearly all recent HTC works focus on how the labels are structured but ignore the sub-structure of ground-truth labels according to each input text which contains fruitful label co-occurrence information. In this work, we introduce this local hierarchy with an adversarial framework. We propose a HiAdv framework that can fit in nearly all HTC models and optimize them with the local hierarchy as auxiliary information. We test on two typical HTC models and find that HiAdv is effective in all scenarios and is adept at dealing with complex taxonomic hierarchies. Further experiments demonstrate that the promotion of our framework indeed comes from the local hierarchy and the local hierarchy is beneficial for rare classes which have insufficient training data.
Anthology ID:
2024.lrec-main.1504
Volume:
Proceedings of the 2024 Joint International Conference on Computational Linguistics, Language Resources and Evaluation (LREC-COLING 2024)
Month:
May
Year:
2024
Address:
Torino, Italia
Editors:
Nicoletta Calzolari, Min-Yen Kan, Veronique Hoste, Alessandro Lenci, Sakriani Sakti, Nianwen Xue
Venues:
LREC | COLING
SIG:
Publisher:
ELRA and ICCL
Note:
Pages:
17326–17336
Language:
URL:
https://aclanthology.org/2024.lrec-main.1504
DOI:
Bibkey:
Cite (ACL):
Zihan Wang, Peiyi Wang, and Houfeng Wang. 2024. Utilizing Local Hierarchy with Adversarial Training for Hierarchical Text Classification. In Proceedings of the 2024 Joint International Conference on Computational Linguistics, Language Resources and Evaluation (LREC-COLING 2024), pages 17326–17336, Torino, Italia. ELRA and ICCL.
Cite (Informal):
Utilizing Local Hierarchy with Adversarial Training for Hierarchical Text Classification (Wang et al., LREC-COLING 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.lrec-main.1504.pdf
Optional supplementary material:
 2024.lrec-main.1504.OptionalSupplementaryMaterial.zip