Hierarchical Text Classification with LLM-Refined Taxonomies
By: Jonas Golde , Nicolaas Jedema , Ravi Krishnan and more
Potential Business Impact:
Fixes messy categories so computers understand them better.
Hierarchical text classification (HTC) depends on taxonomies that organize labels into structured hierarchies. However, many real-world taxonomies introduce ambiguities, such as identical leaf names under similar parent nodes, which prevent language models (LMs) from learning clear decision boundaries. In this paper, we present TaxMorph, a framework that uses large language models (LLMs) to transform entire taxonomies through operations such as renaming, merging, splitting, and reordering. Unlike prior work, our method revises the full hierarchy to better match the semantics encoded by LMs. Experiments across three HTC benchmarks show that LLM-refined taxonomies consistently outperform human-curated ones in various settings up to +2.9pp. in F1. To better understand these improvements, we compare how well LMs can assign leaf nodes to parent nodes and vice versa across human-curated and LLM-refined taxonomies. We find that human-curated taxonomies lead to more easily separable clusters in embedding space. However, the LLM-refined taxonomies align more closely with the model's actual confusion patterns during classification. In other words, even though they are harder to separate, they better reflect the model's inductive biases. These findings suggest that LLM-guided refinement creates taxonomies that are more compatible with how models learn, improving HTC performance.
Similar Papers
Leveraging Taxonomy and LLMs for Improved Multimodal Hierarchical Classification
Artificial Intelligence
Helps computers sort things in a better order.
Hierarchical Text Classification Using Black Box Large Language Models
Computation and Language
Lets computers sort texts into categories better.
Combining Language and Topic Models for Hierarchical Text Classification
Computation and Language
Helps computers sort texts into categories better.