AgentKGV: Agentic LLM-RAG Framework with Two-Stage Training for the Fact Verification of Knowledge Graphs
Summary
AgentKGV is an Agentic LLM-RAG framework designed for the fact verification of knowledge graphs (KGs), addressing factual errors arising from noisy sources and extraction failures at industrial scale. The framework integrates dynamic routing and iterative query rewriting to effectively handle surface-form mismatches during document-level retrieval. To enhance accuracy and cost-efficiency for industrial deployment, AgentKGV employs a two-stage training strategy. This includes turn-level distillation-based Supervised Fine-Tuning (SFT) to transfer reasoning abilities from a large teacher model to a smaller one for stable query rewriting, and trajectory-level Generalized Policy Optimization (GRPO) to optimize the search policy and reduce unnecessary retrieval. On the long-tail-predicate split of the T-REx benchmark, AgentKGV improves macro-F1 over single-turn RAG by 5.5 %p, with the two-stage training adding a further 9.4 %p improvement. GRPO specifically reduces average search calls from 3.24 to 1.63 without accuracy loss.
Key takeaway
For Machine Learning Engineers building knowledge graph verification systems, AgentKGV offers a robust approach to improve factual accuracy and operational efficiency. You should consider integrating agentic LLM-RAG with dynamic routing and iterative query rewriting to handle complex retrieval challenges. Furthermore, adopting a two-stage training strategy, particularly GRPO, can significantly reduce inference costs by cutting search calls by nearly half without compromising verification accuracy.
Key insights
AgentKGV uses agentic LLM-RAG with two-stage training to verify knowledge graph facts, improving accuracy and reducing retrieval costs.
Principles
- Iterative query rewriting improves retrieval.
- Distillation transfers reasoning to smaller models.
- Policy optimization reduces search overhead.
Method
AgentKGV combines dynamic routing and iterative query rewriting for KG fact verification. Its two-stage training uses distillation-based SFT and trajectory-level GRPO to optimize reasoning and retrieval.
In practice
- Implement agentic RAG for KG verification.
- Apply two-stage training for efficiency.
- Use GRPO to cut retrieval calls.
Topics
- Knowledge Graphs
- Fact Verification
- LLM-RAG Frameworks
- Agentic AI
- Two-Stage Training
- Policy Optimization
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.