arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2410.08985cs.AIcs.CL
收藏

迈向可信的知识图谱推理:一种不确定性感知的视角

Towards Trustworthy Knowledge Graph Reasoning: An Uncertainty Aware Perspective

Vanderbilt University(范德堡大学);University of Oregon(俄勒冈大学);University of Illinois Chicago(伊利诺伊大学芝加哥分校);Air Force Research Lab(空军研究实验室)

Bo Ni, Yu Wang, Lu Cheng, Erik Blasch, Tyler Derr

AI总结:

本文提出不确定性感知的知识图谱推理框架UAG,通过保形预测和错误率控制模块为KG-LLM推理提供理论保证,在实现任意预定义覆盖率的同时平均将预测集大小缩减40%。

摘要:

近年来,知识图谱(KGs)已成功地与大型语言模型(LLMs)相结合,以减轻其幻觉并增强其推理能力,例如在基于知识图谱的检索增强框架中。然而,当前的KG-LLM框架缺乏严格的不确定性估计,限制了它们在高风险应用中的可靠部署。由于这些框架具有复杂的架构,且知识图谱组件与语言模型组件之间存在复杂的交互,直接将不确定性量化纳入KG-LLM框架面临诸多挑战。为弥补这一不足,我们提出了一种新的可信KG-LLM框架,即不确定性感知的知识图谱推理(UAG),它将不确定性量化引入KG-LLM框架。我们设计了一个不确定性感知的多步推理框架,利用保形预测为预测集提供理论保证。为了管理多步过程的错误率,我们额外引入了一个错误率控制模块,以调整各个组件内部的错误率。大量实验表明,我们提出的UAG能够实现任意预定义的覆盖率,同时与基线相比,平均将预测集/区间大小缩减40%。

英文摘要:

Recently, Knowledge Graphs (KGs) have been successfully coupled with Large Language Models (LLMs) to mitigate their hallucinations and enhance their reasoning capability, such as in KG-based retrieval-augmented frameworks. However, current KG-LLM frameworks lack rigorous uncertainty estimation, limiting their reliable deployment in high-stakes applications. Directly incorporating uncertainty quantification into KG-LLM frameworks presents challenges due to their complex architectures and the intricate interactions between the knowledge graph and language model components. To address this gap, we propose a new trustworthy KG-LLM framework, Uncertainty Aware Knowledge-Graph Reasoning (UAG), which incorporates uncertainty quantification into the KG-LLM framework. We design an uncertainty-aware multi-step reasoning framework that leverages conformal prediction to provide a theoretical guarantee on the prediction set. To manage the error rate of the multi-step process, we additionally introduce an error rate control module to adjust the error rate within the individual components. Extensive experiments show that our proposed UAG can achieve any pre-defined coverage rate while reducing the prediction set/interval size by 40% on average over the baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑