用规则指导检索与生成,大幅提升问答准确率
RuleRAG: Rule-Guided Retrieval-Augmented Generation with Language Models for Question Answering
- 引入规则引导检索与生成,统一推理逻辑方向
- 在RuleQA上召回率提升89.2%,准确率提高103.1%
- 适用于需要高精度推理的复杂问答场景
检索增强生成(RAG)在知识密集型问答中展现出巨大潜力。然而,现有方法仅依赖查询本身,未指定检索偏好,也未告知生成器如何引用文档,严重影响问答性能。为此,我们提出规则引导的RAG(RuleRAG-ICL),通过显式规则在上下文学习中引导检索器聚焦规则相关文档,并统一引导生成器依据相同规则进行推理。鉴于多数现有RAG数据集未考虑规则,而知识图谱(KGs)可提供高质量规则,我们基于KG构建了五个规则感知的RAG基准数据集RuleQA,以强调规则在检索与推理中的重要性。在RuleQA上的实验表明,RuleRAG-ICL使召回率@10提升89.2%,精确匹配准确率提高103.1%;微调版本RuleRAG-FT效果更优。此外,在四个现有RAG数据集上引入规则后,RuleRAG仍表现优异,验证了规则引导的普适性。
原文摘要 · Abstract (English)
Retrieval-augmented generation (RAG) has shown promising potential in knowledge intensive question answering (QA). However, existing approaches only consider the query itself, neither specifying the retrieval preferences for the retrievers nor informing the generators of how to refer to the retrieved documents for the answers, which poses a significant challenge to the QA performance. To address these issues, we propose Rule-guided Retrieval-Augmented Generation with LMs, which explicitly introduces rules for in-context learning (RuleRAG-ICL) to guide retrievers to recall related documents in the directions of rules and uniformly guide generators to reason attributed by the same rules. Moreover, most existing RAG datasets were constructed without considering rules and Knowledge Graphs (KGs) are recognized as providing high-quality rules. Therefore, we construct five rule-aware RAG benchmarks for QA, RuleQA, based on KGs to stress the significance of retrieval and reasoning with rules. Experiments on RuleQA demonstrate RuleRAG-ICL improves the retrieval quality of +89.2% in Recall@10 and answer accuracy of +103.1% in Exact Match, and RuleRAG-FT yields more enhancement. In addition, experiments on four existing RAG datasets show RuleRAG is also effective by offering rules in RuleQA to them, further proving the generalization of rule guidance in RuleRAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。