SOURCE-LINKED INTELLIGENCE

From Confusion to Clarity: Confusion-Aware Retrieval and Knowledge Injection for Text Classification

arXiv · AI, language, vision and robotics · article · Sep 1, 2026 · UTC

Large language models (LLMs) struggle to classify text into taxonomies with many semantically similar labels, as the distinctions are domain-specific and not captured by pre-training. To handle large label spaces, a common approach retrieves top-$K$ candidate labels by embedding similarity and prompt the LLM to choose among them. However, top-$K$ retrieval reduces the number of candidates but does not help the model tell similar ones apart. When two similar labels both appear as candidates, the model lacks the signal to choose correctly between them. We propose a framework that (1) identifies

Read original source ↗ Open in workspace

recordType

paper

region

Global