UltRAG: a Universal Simple Scalable Recipe for Knowledge Graph RAG
Dobrik Georgiev, Kheeran Naidu, Alberto Cattaneo, Federico Monti, Carlo Luschi, Daniel Justus

TL;DR
ULTRAG introduces a novel framework for Knowledge Graph question answering that leverages off-the-shelf neural modules, enabling state-of-the-art performance without retraining large language models.
Contribution
It presents a new retrieval framework for Knowledge Graphs that achieves high accuracy and scalability without retraining LLMs or neural query executors.
Findings
ULTRAG outperforms existing KG-RAG methods in accuracy.
It enables LLMs to handle Wikidata-scale graphs efficiently.
The approach achieves comparable or lower computational costs.
Abstract
Large language models (LLMs) frequently generate confident yet factually incorrect content when used for language generation (a phenomenon often known as hallucination). Retrieval augmented generation (RAG) tries to reduce factual errors by identifying information in a knowledge corpus and putting it in the context window of the model. While this approach is well-established for document-structured data, it is non-trivial to adapt it for Knowledge Graphs (KGs), especially for queries that require multi-node/multi-hop reasoning on graphs. We introduce ULTRAG, a general framework for retrieving information from Knowledge Graphs that shifts away from classical RAG. By endowing LLMs with off-the-shelf neural query executing modules, we highlight how readily available language models can achieve state-of-the-art results on Knowledge Graph Question Answering (KGQA) tasks without any…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
