Use knowledge-graph paths + chain-of-thought to guide LLMs for domain QA with only three LLM calls

April 16, 20247 min

Overview

Decision SnapshotNeeds Validation

The approach is practical and lightweight: use CoT to expand entities, enumerate multi-hop pairs in a KG, rank paths with PageRank, and prune neighbors with one LLM call. Evidence is moderate: experiments on three datasets show gains, but evaluation uses GPT-4 ranking and limited sample sizes.

Citations3

Evidence Strength0.70

Confidence0.78

Risk Signals9

Trust Signals

Findings with numeric evidence: 5/5

Findings with evidence refs: 5/5

Results with explicit delta: 5/5

Reproducibility

Status: Partial assets available

Open source: Partial

At A Glance

Cost impact: 60%

Production readiness: 60%

Novelty: 45%

Authors

Yuqi Wang, Boran Jiang, Yi Luo, Dawei He, Peng Cheng, Liangcai Gao

Links

Abstract / PDF / Data

Why It Matters For Business

RoK gives domain QA systems more accurate, interpretable answers while cutting LLM calls and API cost by structuring knowledge as ranked KG paths.

Who Should Care

Summary TLDR

RoK (Reasoning on Efficient Knowledge Paths) builds short reasoning paths from a knowledge graph and uses chain-of-thought (CoT) to expand question entities. It ranks candidate multi-hop paths with PageRank and supplies the top paths as structured background to an LLM. On three QA sets (medical and WebQuestions subsets), RoK improves entity-match and semantic scores versus vanilla LLMs and prior KG-based prompts while requiring only about three LLM calls per question.

Problem Statement

LLMs hallucinate and struggle on domain-specific multi-hop questions because they lack structured domain context and fine-tuning is costly or unavailable. Prior KG+LLM methods call LLM many times during path selection, which is slow and error-prone when an early path choice is wrong.

Main Contribution

A pipeline (RoK) that expands question entities via LLM chain-of-thought, enumerates multi-hop candidate paths in a knowledge graph, ranks paths with PageRank, and prunes neighbor triples with a single LLM call.

A practical subgraph retrieval method that uses CoT to increase entity linking coverage and PageRank to pick top-N reasoning paths as structured prompts.

Key Findings

RoK improves key-entity match accuracy on GenMedGPT-5k to 81.3% from GPT-3.5's 52.6%.

NumbersGenMedGPT-5k key-entity match: RoK 81.3% vs GPT-3.5 52.6%

Practical UseIf you care about extracting correct domain entities from user questions, add CoT-based entity expansion + KG path ranking to boost entity hit rates.

Evidence RefTable 3

RoK increases answer semantic similarity (BERTScore F1) on GenMedGPT-5k to 0.815 vs GPT-3.5 0.791 and prior MindMap 0.802.

NumbersBERTScore F1 GenMedGPT-5k: RoK 0.815, GPT-3.5 0.791, MindMap 0.802

Practical UseUsing ranked KG paths as structured context yields measurably closer answers to references than plain LLM or simple embedding retrieval.

Evidence RefTable 2

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
BERTScore F1 (GenMedGPT-5k)0.815 (RoK)0.791 (GPT-3.5)+0.024GenMedGPT-5kTable 2 shows BERTScore F1Table 2
BERTScore F1 (CMCQA)0.934 (RoK)0.915 (GPT-3.5)+0.019CMCQATable 2 shows BERTScore F1Table 2

What To Try In 7 Days

Add a CoT step to expand entities from user questions, then do simple entity linking to your domain KG.

Enumerate short multi-hop paths between linked entities, score nodes with PageRank, and surface top paths as structured context to your LLM.

Replace expensive multi-step LLM path selection with a single LLM pass to filter neighbor triples and one final answer call; measure entity hit rate and API calls.

Optimization Features

Token Efficiency
Provide compact reasoning paths instead of long document chunks
Inference Optimization
Reduce LLM calls via structured path selection

Reproducibility

Code AvailableNo
Data AvailableYes
Open Source StatusPartial
LicenseUnknown

Risks & Boundaries

Limitations

Requires a domain knowledge graph with sufficient coverage; missing entities reduce effectiveness.

PageRank is a heuristic and may prioritize high-degree hub nodes over semantically best paths.

When Not To Use

You lack a curated KG for the domain.

Questions are simple single-hop factual lookups where standard retrieval already suffices.

Failure Modes

Entity linking fails or links to wrong KG nodes, producing wrong paths.

PageRank selects hub-heavy but semantically irrelevant paths.

Core Entities

Models

GPT-3.5GPT-4LLAMA2 (mentioned generically)

Metrics

BERTScoreGPT-4 average rankingHits@1Accuracy

Datasets

GenMedGPT-5kCMCQAWebQuestionsFB15k-237