Paper Reading AI Learner

Neural Probe-Based Hallucination Detection for Large Language Models

2025-12-24 05:10:19
Shize Liang, Hongzhi Wang

Abstract

Large language models(LLMs) excel at text generation and knowledge question-answering tasks, but they are prone to generating hallucinated content, severely limiting their application in high-risk domains. Current hallucination detection methods based on uncertainty estimation and external knowledge retrieval suffer from the limitation that they still produce erroneous content at high confidence levels and rely heavily on retrieval efficiency and knowledge coverage. In contrast, probe methods that leverage the model's hidden-layer states offer real-time and lightweight advantages. However, traditional linear probes struggle to capture nonlinear structures in deep semantic this http URL overcome these limitations, we propose a neural network-based framework for token-level hallucination detection. By freezing language model parameters, we employ lightweight MLP probes to perform nonlinear modeling of high-level hidden states. A multi-objective joint loss function is designed to enhance detection stability and semantic disambiguity. Additionally, we establish a layer position-probe performance response model, using Bayesian optimization to automatically search for optimal probe insertion layers and achieve superior training this http URL results on LongFact, HealthBench, and TriviaQA demonstrate that MLP probes significantly outperform state-of-the-art methods in accuracy, recall, and detection capability under low false-positive conditions.

Abstract (translated)

大型语言模型(LLMs)在文本生成和知识问答任务中表现出色,但它们容易产生虚构内容,这严重限制了其在高风险领域的应用。当前基于不确定性估计和外部知识检索的幻觉检测方法存在局限性:即便自信度很高时仍会产生错误内容,并且高度依赖于检索效率和知识覆盖范围。相比之下,利用模型隐藏层状态的探针方法提供了实时性和轻量级的优势。然而,传统的线性探针难以捕捉深层语义中的非线性结构。为克服这些限制,我们提出了一种基于神经网络的框架来进行标记级别的幻觉检测。通过冻结语言模型参数,我们使用轻量级MLP探针对高级隐藏状态进行非线性建模。设计了一个多目标联合损失函数以增强检测稳定性和语义歧义消除能力。此外,建立了层位置-探针性能响应模型,并采用贝叶斯优化自动搜索最佳的探针插入层,从而实现优越的训练效果。在LongFact、HealthBench和TriviaQA上的实验结果表明,在低假阳性条件下,MLP探针对比当前最先进的方法在准确性、召回率和检测能力方面有显著提升。

URL

https://arxiv.org/abs/2512.20949

PDF

https://arxiv.org/pdf/2512.20949.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot