Utilizing Local Hierarchy with Adversarial Training for Hierarchical Text Classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Zihan, Wang, Peiyi, Wang, Houfeng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910389614149632
author Wang, Zihan
Wang, Peiyi
Wang, Houfeng
author_facet Wang, Zihan
Wang, Peiyi
Wang, Houfeng
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_18825
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Utilizing Local Hierarchy with Adversarial Training for Hierarchical Text Classification
Wang, Zihan
Wang, Peiyi
Wang, Houfeng
Computation and Language
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.
title Utilizing Local Hierarchy with Adversarial Training for Hierarchical Text Classification
topic Computation and Language
url https://arxiv.org/abs/2402.18825