ReDeEP: Detecting Hallucination in Retrieval-Augmented Generation via Mechanistic Interpretability
Zhongxiang Sun, Xiaoxue Zang, Kai Zheng, Yang Song, Jun Xu, Xiao Zhang, Weijie Yu, Yang Song, Han Li
TL;DR
This work investigates RAG hallucinations by disentangling the LLM's use of external context from its parametric knowledge using mechanistic interpretability. It proposes ReDeEP to detect hallucinations via decoupled External Context Score $\mathcal{E}$ and Parametric Knowledge Score $\mathcal{P}$, and demonstrates that hallucinations arise from underutilization of external context by Copying Heads and overreliance on parametric knowledge by Knowledge FFNs. AARF provides a mitigation by reweighting Copying Heads and Knowledge FFNs without updating model parameters. The approach yields significant gains on RAGTruth and Dolly (AC) datasets across multiple LLaMA backbones and improves truthfulness as validated by GPT-4o evaluations, offering a practical, efficient path to reliable retrieval-augmented systems.
Abstract
Retrieval-Augmented Generation (RAG) models are designed to incorporate external knowledge, reducing hallucinations caused by insufficient parametric (internal) knowledge. However, even with accurate and relevant retrieved content, RAG models can still produce hallucinations by generating outputs that conflict with the retrieved information. Detecting such hallucinations requires disentangling how Large Language Models (LLMs) utilize external and parametric knowledge. Current detection methods often focus on one of these mechanisms or without decoupling their intertwined effects, making accurate detection difficult. In this paper, we investigate the internal mechanisms behind hallucinations in RAG scenarios. We discover hallucinations occur when the Knowledge FFNs in LLMs overemphasize parametric knowledge in the residual stream, while Copying Heads fail to effectively retain or integrate external knowledge from retrieved content. Based on these findings, we propose ReDeEP, a novel method that detects hallucinations by decoupling LLM's utilization of external context and parametric knowledge. Our experiments show that ReDeEP significantly improves RAG hallucination detection accuracy. Additionally, we introduce AARF, which mitigates hallucinations by modulating the contributions of Knowledge FFNs and Copying Heads.
