提出新方法提升RAG中幻觉检测的可靠性
INTRYGUE: Induction-Aware Entropy Gating for Reliable RAG Uncertainty Estimation
- 用注意力机制捕捉归纳头活动,动态调整预测熵
- 在4个基准上验证,对6个大模型效果优于主流方法
- 适合需要可信不确定性估计的AI系统开发者
尽管检索增强生成(RAG)能提升大语言模型性能,但无法消除幻觉,准确检测仍至关重要。基于不确定性的方法因其低开销易于集成而受关注,其中预测熵是最常用信号。然而我们发现,在RAG场景下熵值不可靠,根源在于两种相互抵消的内部效应:归纳头会复制早期上下文模式,促进正确响应并降低熵,但同时也会激活熵神经元,导致熵值回升。因此,即使响应正确且基于上下文,仍可能获得高不确定性评分。为此,我们提出INTRYGUE(归纳感知熵门控),一种无需训练、机制可解释的方法,通过注意力机制估计归纳头活性来门控预测熵。在四个RAG类基准和六个开源大模型(4B至13B参数)上评估,该方法表现优于或匹配多数基线。结果表明,结合预测不确定性与可解释的上下文利用信号,有助于提升RAG中的幻觉检测能力。
原文摘要 · Abstract (English)
While retrieval-augmented generation (RAG) enhances LLM performance, it does not eliminate hallucinations, making accurate detection essential. Uncertainty-based methods are attractive for this purpose because they can be integrated into real-world pipelines with little overhead. One of the most widely used uncertainty signals is predictive entropy. We show, however, that entropy can be unreliable in RAG settings and trace this limitation to two opposing internal effects. Induction heads, which copy patterns from earlier context, causally support correct responses and lower predictive entropy, but they also appear to co-activate entropy neurons that push it back up. As a result, correct, context-grounded responses can still receive high uncertainty scores. To address this, we propose INTRYGUE (Induction-Aware Entropy Gating for Uncertainty Estimation), a training-free, mechanistically grounded method that gates predictive entropy by an attention-based estimate of induction-head activity. Evaluated across four RAG-style benchmarks and six open-source LLMs (4B to 13B parameters), INTRYGUE performs competitively against a wide range of baselines, matching or exceeding the strongest of them in most settings. Our findings suggest that hallucination detection in RAG benefits from combining predictive uncertainty with interpretable internal signals of context utilization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。