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

Paraphrase and Aggregate with Large Language Models for Minimizing Intent Classification Errors

About

Large language models (LLM) have achieved remarkable success in natural language generation but lesser focus has been given to their applicability in decision making tasks such as classification. We show that LLMs like LLaMa can achieve high performance on large multi-class classification tasks but still make classification errors and worse, generate out-of-vocabulary class labels. To address these critical issues, we introduce Paraphrase and AGgregate (PAG)-LLM approach wherein an LLM generates multiple paraphrases of the input query (parallel queries), performs multi-class classification for the original query and each paraphrase, and at the end aggregate all the classification labels based on their confidence scores. We evaluate PAG-LLM on two large multi-class classication datasets: CLINC, and Banking and show 22.7% and 15.1% error reduction. We show that PAG-LLM is especially effective for hard examples where LLM is uncertain, and reduces the critical misclassification and hallucinated label generation errors

Vikas Yadav, Zheng Tang, Vijay Srinivasan• 2024

Related benchmarks

TaskDatasetResultRank
Text ClassificationAG News (test)
Accuracy77.71
210
Text ClassificationTREC (test)
Accuracy71.52
113
Text ClassificationIMDB (test)
Accuracy93.22
77
Text ClassificationPubmed standard (test)
Accuracy74.44
22
Showing 4 of 4 rows

Other info

Follow for update