Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Simple and Effective Framework for Strict Zero-Shot Hierarchical Classification

About

In recent years, large language models (LLMs) have achieved strong performance on benchmark tasks, especially in zero or few-shot settings. However, these benchmarks often do not adequately address the challenges posed in the real-world, such as that of hierarchical classification. In order to address this challenge, we propose refactoring conventional tasks on hierarchical datasets into a more indicative long-tail prediction task. We observe LLMs are more prone to failure in these cases. To address these limitations, we propose the use of entailment-contradiction prediction in conjunction with LLMs, which allows for strong performance in a strict zero-shot setting. Importantly, our method does not require any parameter updates, a resource-intensive process and achieves strong performance across multiple datasets.

Rohan Bhambhoria, Lei Chen, Xiaodan Zhu• 2023

Related benchmarks

TaskDatasetResultRank
Hierarchical classificationWeb of Sciences (WOS) Tree Depth = 2 (test)
Accuracy61.78
6
Hierarchical classificationAmazon Beauty Tree Depth = 2 (test)
Accuracy64.25
6
Hierarchical classificationAmazon Beauty Tree Depth = 3, 4, 5 (test)
Accuracy40.79
6
Showing 3 of 3 rows

Other info

Follow for update