When to Trust: A Causality-Aware Calibration Framework for Accurate Knowledge Graph Retrieval-Augmented Generation

Read full text
Authors: Jing Fei Ren, Bowen Li, Ziqi Xu, Xikun Zhang, Haytham M. Fayek, Xiaodong Li
Cited by: 1
References: 32

Abstract

Knowledge Graph Retrieval-Augmented Generation (KG-RAG) extends the RAG paradigm by incorporating structured knowledge from knowledge graphs, enabling Large Language Models (LLMs) to perform more precise and explainable reasoning. While KG-RAG improves factual accuracy in complex tasks, existing KG-RAG models are often severely overconfident, producing high-confidence predictions even when retrieved sub-graphs are incomplete or unreliable, which raises concerns for deployment in high-stakes domains. To address this issue, we propose Ca2KG, a Causality-aware Calibration framework for KG-RAG. Ca2KG integrates counterfactual prompting, which exposes retrieval-dependent uncertainties in knowledge quality and reasoning reliability, with a panel-based re-scoring mechanism that stabilises predictions across interventions. Extensive experiments on two complex QA datasets demonstrate that Ca2KG consistently improves calibration while maintaining or even enhancing predictive accuracy. The source code can be found at~ https://aisuko.github.io/ca2kg/.

Fields of Study & Subjects

Advanced Graph Neural Networks Graph Theory and Algorithms Bayesian Modeling and Causal Inference

Publication details

Journal / source
Proceedings of the ACM Web Conference 2026
Publisher
ACM
Pages
3859–3869
Publication date
9 April 2026