AgentKGV: Agentic LLM-RAG Framework with Two-Stage Training for the Fact Verification of Knowledge Graphs

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics, Robotics & Autonomous Systems · Depth: Expert, quick

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

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

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.