Your Extreme Multi-label Classifier is Secretly a Hierarchical Text Classifier for Free
Not on the map. Why?
Abstract (arXiv)
Assigning a set of labels to a given text is a classification problem with many real-world applications, such as recommender systems. Two separate research streams address this issue. Hierarchical Text Classification (HTC) focuses on datasets with label pools of hundreds of entries, accompanied by a semantic label hierarchy. In contrast, eXtreme Multi-Label Text Classification (XML) considers very large sets of labels with up to millions of entries but without an explicit hierarchy. In XML methods, it is common to construct an artificial hierarchy in order to deal with the large label space before or during the training process. Here, we investigate how state-of-the-art HTC models perform when trained and tested on XML datasets and vice versa using three benchmark datasets from each of the two streams. Our results demonstrate that XML models, with their internally constructed hierarchy, are very effective HTC models. HTC models, on the other hand, are not equipped to handle the sheer label set size of XML datasets and achieve poor transfer results. We further argue that for a fair comparison in HTC and XML, more than one metric like F1 should be used but complemented with P@k and R-Precision.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 23 of its 44 references that are in the map. First posted on arXiv 20 Nov 2024.
Builds on
23 of 44 references.
References · 23
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2023GPT-4 Technical Reportcited by 18,396
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2017Graph Attention Networkscited by 8,923
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,811
- 2021Do Transformers Really Perform Bad for Graph Representation?cited by 320
- 2017HDLTex: Hierarchical Deep Learning for Text Classificationcited by 106
- 2023Text Classification via Large Language Modelscited by 106
- 2016DiSMEC - Distributed Sparse Machines for Extreme Multi-label Classificationcited by 71
- 2021Fast Multi-Resolution Transformer Fine-tuning for Extreme Multi-label Text Classificationcited by 67
- 2021LightXML: Transformer with Dynamic Negative Sampling for High-Performance Extreme Multi-label Text Classificationcited by 61
11 more
- 2018A no-regret generalization of hierarchical softmax to extreme multi-label classificationcited by 52
- 2022Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text Classificationcited by 52
- 2023Open, Closed, or Small Language Models for Text Classification?cited by 34
- 2022HPT: Hierarchy-aware Prompt Tuning for Hierarchical Text Classificationcited by 31
- 2021Extreme Multi-label Learning for Semantic Matching in Product Searchcited by 28
- 2021HTCInfoMax: A Global Model for Hierarchical Text Classification via Information Maximizationcited by 25
- 2022Exploiting Global and Local Hierarchies for Hierarchical Text Classificationcited by 15
- 2024Revisiting Adversarial Training under Long-Tailed Distributionscited by 13
- 2024HILL: Hierarchy-aware Information Lossless Contrastive Learning for Hierarchical Text Classificationcited by 13
- 2021InceptionXML: A Lightweight Framework with Synchronized Negative Sampling for Short Text Extreme Classificationcited by 8
- 2021s2s-ft: Fine-Tuning Pretrained Transformer Encoders for Sequence-to-Sequence Learningcited by 7
Built on by
Citing papers per year: 2024: 1; 2025: 0; 2026: 1.
2 papers.