English

Revisiting Hierarchical Text Classification: Inference and Metrics

Computation and Language 2024-10-14 v2 Machine Learning

Abstract

Hierarchical text classification (HTC) is the task of assigning labels to a text within a structured space organized as a hierarchy. Recent works treat HTC as a conventional multilabel classification problem, therefore evaluating it as such. We instead propose to evaluate models based on specifically designed hierarchical metrics and we demonstrate the intricacy of metric choice and prediction inference method. We introduce a new challenging dataset and we evaluate fairly, recent sophisticated models, comparing them with a range of simple but strong baselines, including a new theoretically motivated loss. Finally, we show that those baselines are very often competitive with the latest models. This highlights the importance of carefully considering the evaluation methodology when proposing new methods for HTC. Code implementation and dataset are available at \url{https://github.com/RomanPlaud/revisitingHTC}.

Keywords

Cite

@article{arxiv.2410.01305,
  title  = {Revisiting Hierarchical Text Classification: Inference and Metrics},
  author = {Roman Plaud and Matthieu Labeau and Antoine Saillenfest and Thomas Bonald},
  journal= {arXiv preprint arXiv:2410.01305},
  year   = {2024}
}

Comments

Accepted at CoNLL 2024

R2 v1 2026-06-28T19:04:48.836Z