arXiv:2605.05244cs.IRcs.AI2026-05

提出可信检索增强生成方法,用置信度预测提升生成事实准确性。

Towards Dependable Retrieval-Augmented Generation Using Factual Confidence Prediction

论文配图:Towards Dependable Retrieval-Augmented Generation Using Factual Confidence Prediction
图 1 · 摘自论文原文
  • 分两阶段:先用校准预测筛选可靠检索片段,再用注意力机制评估答案与上下文一致性。
  • 在部分数据集上提升答案质量达6%,可检测77%的不一致生成结果。
  • 适合需要高可靠性生成的工业应用,如客服、医疗问答等场景。

通过检索增强生成(RAG)将特定知识注入大语言模型是当前主流技术,广泛应用于各类产业AI系统。核心挑战在于判断检索到的上下文是否真正支持生成内容,而非引入无关信息。为此,本文提出一种两阶段方法来预测RAG输出的事实可信度。首先,采用校准预测筛选高概率来自正确来源的检索片段,在部分数据集上可使答案质量提升6%;但该方法的统计保证依赖于样本可交换性假设,因此我们设计诊断指标以评估不同检索器设置的适用性。其次,利用基于注意力的事实性分类器,量化生成答案与给定上下文的一致性置信度,最高可实现77%的不一致答案检测率。本工作推动了具备可验证性的新型认证型RAG系统的建立,适用于广泛的自然语言产业应用。

原文摘要 · Abstract (English)

Incorporating specific knowledge into large language models via retrieval-augmented generation (RAG) is a widespread technique that fuels many of today's industry AI applications. A fundamental problem is to assess if the context retrieved by some similarity search provides indeed supporting facts, or instead misguides the generator with irrelevant information. It is critical to associate meaningful confidence measures about the factuality of the retrieval process with the generated answers. We present a new, two-staged approach to predict fact faithfulness of the output of retrieval-augmented generations. First, we employ conformal prediction to select only those retrieved chunks who have a high chance to come from the correct source. This approach in itself can improve answer quality by up to 6% in some of the studied datasets, however, the associated statistical guarantees do not hold generally, since the assumption of sample exchangeability depends on the retriever setup. We present diagnostic metrics to assess whether a setup is suitable. Second, we quantify confidence in the consistency of a generated final answer with a given retrieved context, using an attention-based factuality classifier. This approach can detect inconsistent answers with a chance of up to 77%. Our work helps to establish a novel type of certified RAG systems for a broad range of natural language industry applications.

检索增强可信生成置信度预测RAG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。