Efficient Hallucination Detection for LLMs Using Uncertainty-Aware Attention Heads
About
While large language models (LLMs) have become highly capable, they remain prone to factual inaccuracies, commonly referred to as "hallucinations." Uncertainty quantification (UQ) offers a promising way to mitigate this issue, but most existing methods are computationally intensive and/or require supervision. In this work, we propose Recurrent Attention-based Uncertainty Quantification (RAUQ), an unsupervised and efficient framework for identifying hallucinations. The method leverages an observation about transformer attention behavior: when incorrect information is generated, certain "uncertainty-aware" attention heads tend to reduce their focus on preceding tokens. RAUQ automatically detects these attention heads and combines their activation patterns with token-level confidence measures in a recurrent scheme, producing a sequence-level uncertainty estimate in just a single forward pass. Through experiments on twelve datasets spanning question answering, summarization, and translation across nine different LLMs, we show that RAUQ consistently outperforms state-of-the-art UQ baselines. Importantly, it incurs minimal overhead, requiring less than 1\% additional computation. Since it requires neither labeled data nor extensive parameter tuning, RAUQ serves as a lightweight, plug-and-play solution for real-time hallucination detection in white-box LLMs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Hallucination Detection | TriviaQA | -- | 625 | |
| Machine Translation | MT | Mean PRR49.5 | 109 | |
| Question Answering | QA | Mean PRR42.5 | 109 | |
| Summarization | Summ. | Mean PRR0.428 | 109 | |
| Hallucination Detection | CoQA | Mean AUROC0.4 | 107 | |
| Claim Verification | 9-dataset aggregate retrieval-free setting (test) | ROC-AUC63.9 | 70 | |
| Hallucination Detection | Summ. | ROC-AUC81.5 | 64 | |
| Hallucination Detection | MT | ROC-AUC72.7 | 64 | |
| Hallucination Detection | QA | ROC-AUC75.2 | 64 | |
| Hallucination Detection | MMLU | -- | 62 |