RFKG-CoT: Relation-Driven Adaptive Hop-count Selection and Few-Shot Path Guidance for Knowledge-Aware QA

TL;DR

RFKG-CoT enhances knowledge-aware question answering by dynamically selecting reasoning steps based on relations and guiding LLMs with few-shot path prompts, significantly improving accuracy over previous methods.

Contribution

It introduces a relation-driven adaptive hop-count selector and a few-shot path guidance mechanism, addressing rigidity and underutilization in existing KG-CoT approaches.

Findings

Achieves up to 14.7 percentage points accuracy improvement.

The hop-count selector and path prompt are complementary.

Transforms KG evidence into more faithful answers.

Abstract

Large language models (LLMs) often generate hallucinations in knowledge-intensive QA due to parametric knowledge limitations. While existing methods like KG-CoT improve reliability by integrating knowledge graph (KG) paths, they suffer from rigid hop-count selection (solely question-driven) and underutilization of reasoning paths (lack of guidance). To address this, we propose RFKG-CoT: First, it replaces the rigid hop-count selector with a relation-driven adaptive hop-count selector that dynamically adjusts reasoning steps by activating KG relations (e.g., 1-hop for direct "brother" relations, 2-hop for indirect "father-son" chains), formalized via a relation mask. Second, it introduces a few-shot in-context learning path guidance mechanism with CoT (think) that constructs examples in a "question-paths-answer" format to enhance LLMs' ability to understand reasoning paths. Experiments on four KGQA benchmarks show RFKG-CoT improves accuracy by up to 14.7 pp (Llama2-7B on WebQSP) over KG-CoT. Ablations confirm the hop-count selector and the path prompt are complementary, jointly transforming KG evidence into more faithful answers.