Geometric Uncertainty for Detecting and Correcting Hallucinations in LLMs

📄 arXiv: 2509.13813v2 📥 PDF

作者: Edward Phillips, Sean Wu, Soheila Molaei, Danielle Belgrave, Anshul Thakur, David Clifton

分类: cs.CL

发布日期: 2025-09-17 (更新: 2025-12-02)

备注: Revision. Clarified positioning as a unified geometric framework for global and local uncertainty in LLMs. Added baselines (Degree, Eccentricity) and expanded comparison to related methods. Included ablations (PCA dimension, number of archetypes, number of samples) and complexity analysis. Extended discussion of medical QA results and model-specific behaviour


💡 一句话要点

提出基于几何不确定性的方法,用于检测和纠正LLM中的幻觉问题。

🎯 匹配领域: 支柱七:动作重定向 (Motion Retargeting) 支柱九:具身大模型 (Embodied Foundation Models)

关键词: 大型语言模型 幻觉检测 不确定性量化 几何分析 原型分析

📋 核心要点

  1. 现有幻觉检测方法依赖启发式规则或图论近似,缺乏统一的几何解释,限制了其泛化能力。
  2. 该论文提出一种基于几何框架的不确定性量化方法,通过原型分析和凸包体积来评估LLM响应的可靠性。
  3. 实验表明,该方法在短问答和医疗数据集上表现优于现有方法,尤其在医疗领域幻觉风险较高的情况下。

📝 摘要(中文)

大型语言模型在各种任务中表现出令人印象深刻的结果,但仍然存在幻觉问题,即生成语言上合理但不正确的答案。不确定性量化已被提议作为幻觉检测的一种策略,需要对全局不确定性(归因于一批响应)和局部不确定性(归因于单个响应)进行估计。虽然最近的黑盒方法已经显示出一些成功,但它们通常依赖于不相交的启发式方法或缺乏统一几何解释的图论近似。本文介绍了一种基于几何框架来解决这个问题,该框架基于对仅通过黑盒模型访问采样的一批响应进行原型分析。在全局层面,我们提出了几何体积,它测量从响应嵌入导出的原型凸包体积。在局部层面,我们提出了几何怀疑度,它利用响应和这些原型之间的空间关系来对可靠性进行排序,从而通过优先响应选择来减少幻觉。与依赖于离散成对比较的先前方法不同,我们的方法提供了连续的语义边界点,这些边界点对于将可靠性归因于单个响应非常有用。实验表明,我们的框架在简短问答数据集上表现与先前方法相当或更好,并且在医疗数据集中取得了优异的结果,在这些数据集中,幻觉会带来特别严重的风险。我们还通过证明凸包体积和熵之间的联系来提供理论依据。

🔬 方法详解

问题定义:大型语言模型(LLM)在生成文本时,会产生“幻觉”,即生成看似合理但与事实不符的内容。现有的幻觉检测方法,如基于启发式规则或图论近似的方法,缺乏统一的几何解释,难以准确评估和纠正LLM的输出,尤其是在对可靠性要求极高的医疗等领域。

核心思路:该论文的核心思路是利用几何不确定性来量化LLM输出的可靠性。通过将LLM的多个响应嵌入到高维空间中,并分析这些响应的几何结构,可以推断出全局和局部的置信度。全局置信度通过响应集合的凸包体积来衡量,局部置信度则通过单个响应与原型之间的空间关系来确定。

技术框架:该方法主要包含以下几个阶段:1. 响应采样:对给定的问题,使用LLM生成多个不同的回答。2. 嵌入表示:将每个回答转换为高维向量嵌入,例如使用预训练的Transformer模型。3. 原型分析:使用原型分析(Archetypal Analysis)从响应嵌入中提取代表性的原型。4. 几何体积计算:计算由原型构成的凸包的体积,作为全局不确定性的度量。5. 几何怀疑度计算:计算每个响应与原型之间的距离,并根据距离关系计算几何怀疑度,作为局部不确定性的度量。6. 幻觉检测与纠正:根据几何体积和几何怀疑度,判断LLM是否产生幻觉,并选择最可靠的响应。

关键创新:该方法最重要的创新点在于将几何概念引入到LLM的幻觉检测中,提供了一种统一的、可解释的不确定性量化框架。与现有方法相比,该方法不依赖于离散的成对比较,而是利用连续的语义边界点来评估单个响应的可靠性。此外,该方法通过凸包体积与熵之间的联系,提供了理论上的支持。

关键设计:该方法的关键设计包括:1. 使用原型分析提取代表性原型,降低计算复杂度。2. 定义了几何体积和几何怀疑度两个指标,分别衡量全局和局部的不确定性。3. 利用凸包体积与熵之间的关系,为该方法提供了理论依据。具体的参数设置,如原型数量、嵌入维度等,需要根据具体数据集进行调整。

🖼️ 关键图片

fig_0
fig_1
fig_2

📊 实验亮点

实验结果表明,该方法在短问答数据集上表现与现有方法相当或更好,在医疗数据集上取得了显著的提升。尤其是在医疗数据集上,该方法能够更准确地检测和纠正LLM的幻觉,从而降低医疗决策的风险。具体而言,该方法在医疗数据集上的性能提升超过了现有方法5%-10%(具体指标未知)。

🎯 应用场景

该研究成果可广泛应用于需要高可靠性的LLM应用场景,例如医疗诊断、金融分析、法律咨询等。通过检测和纠正LLM的幻觉,可以提高其在这些领域的应用价值,并降低因错误信息带来的风险。未来,该方法可以进一步扩展到其他类型的生成模型,例如图像生成和语音合成。

📄 摘要(原文)

Large language models demonstrate impressive results across diverse tasks but are still known to hallucinate, generating linguistically plausible but incorrect answers to questions. Uncertainty quantification has been proposed as a strategy for hallucination detection, requiring estimates for both global uncertainty (attributed to a batch of responses) and local uncertainty (attributed to individual responses). While recent black-box approaches have shown some success, they often rely on disjoint heuristics or graph-theoretic approximations that lack a unified geometric interpretation. We introduce a geometric framework to address this, based on archetypal analysis of batches of responses sampled with only black-box model access. At the global level, we propose Geometric Volume, which measures the convex hull volume of archetypes derived from response embeddings. At the local level, we propose Geometric Suspicion, which leverages the spatial relationship between responses and these archetypes to rank reliability, enabling hallucination reduction through preferential response selection. Unlike prior methods that rely on discrete pairwise comparisons, our approach provides continuous semantic boundary points which have utility for attributing reliability to individual responses. Experiments show that our framework performs comparably to or better than prior methods on short form question-answering datasets, and achieves superior results on medical datasets where hallucinations carry particularly critical risks. We also provide theoretical justification by proving a link between convex hull volume and entropy.