HYDRA: A Multi-Head Encoder-only Architecture for Hierarchical Text Classification
Fabian Karl, Ansgar Scherp
Abstract
We introduce HYDRA, a simple yet effective multi-head encoder-only architecture for hierarchical text classification that treats each level in the hierarchy as a separate classification task with its own label space. State-of-the-art approaches rely on complex components like graph encoders, label semantics, and autoregressive decoders. We demonstrate that such complexity is often unnecessary. Through parameter sharing and level-specific parameterization, HYDRA enables flat models to incorporate hierarchical awareness without architectural complexity. Experiments on four benchmarks (NYT, RCV1-V2, BGC, and WOS) demonstrate that HYDRA always increases the performance over flat models and matches or exceeds the performance of complex state-ofthe-art methods. The source code is available at https://github.com/FKarl/HYDRA
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on6
- Deberta: decoding-Enhanced Bert with Disentangled AttentionPengcheng He, Xiaodong Liu, Jianfeng Gao, Weizhu ChenICLR 2021 · 3,729 citations
- DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding SharingPengcheng He, Jianfeng Gao, Weizhu ChenICLR 2023 · 394 citations
- Hierarchy-Aware Global Model for Hierarchical Text ClassificationJie Zhou, Chunping Ma, Dingkun Long, Guangwei Xu et al.ACL 2020 · 171 citations
- Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text ClassificationZihan Wang, Peiyi Wang, Lianzhe Huang, Xin Sun et al.ACL 2022 · 157 citations
- Exploiting Global and Local Hierarchies for Hierarchical Text ClassificationTing Jiang, Deqing Wang, Leilei Sun, Zhongzhi Chen et al.EMNLP 2022 · 29 citations
Related papers
- Hierarchical Text Classification as Sub-hierarchy Sequence GenerationSanghun Im, Gibaeg Kim, Heung-Seon Oh, Seongung Jo et al.AAAI 2023 · 11 citations
- HiTIN: Hierarchy-aware Tree Isomorphism Network for Hierarchical Text ClassificationHe Zhu, Chong Zhang, Junjie Huang, Junran Wu et al.ACL 2023 · 17 citations
- Hierarchical Multi-label Text Classification with Horizontal and Vertical Category CorrelationsLinli Xu, Sijie Teng, Ruoyu Zhao, Junliang Guo et al.EMNLP 2021 · 22 citations
- Hierarchy-aware Label Semantics Matching Network for Hierarchical Text ClassificationHaibin Chen, Qianli Ma, Zhenxi Lin, Jiangyue YanACL 2021
- MATCH: Metadata-Aware Text Classification in A Large HierarchyYu Zhang, Zhihong Shen, Yuxiao Dong, Kuansan Wang et al.WWW 2021 · 39 citations
