GROUNDEDKG-RAG: Grounded Knowledge Graph Index for Long-document Question Answering

TL;DR

GroundedKG-RAG is a knowledge graph-based retrieval-augmented system for long-document question answering that improves factual accuracy, efficiency, and interpretability by grounding in source text.

Contribution

It introduces a novel grounding method using explicit knowledge graphs constructed from SRL and AMR parses for long-document QA.

Findings

Performs on par with state-of-the-art models at lower cost

Outperforms a competitive baseline in accuracy

Provides interpretable and human-readable knowledge graphs

Abstract

Retrieval-augmented generation (RAG) systems have been widely adopted in contemporary large language models (LLMs) due to their ability to improve generation quality while reducing the required input context length. In this work, we focus on RAG systems for long-document question answering. Current approaches suffer from a heavy reliance on LLM descriptions resulting in high resource consumption and latency, repetitive content across hierarchical levels, and hallucinations due to no or limited grounding in the source text. To improve both efficiency and factual accuracy through grounding, we propose GroundedKG-RAG, a RAG system in which the knowledge graph is explicitly extracted from and grounded in the source document. Specifically, we define nodes in GroundedKG as entities and actions, and edges as temporal or semantic relations, with each node and edge grounded in the original sentences. We construct GroundedKG from semantic role labeling (SRL) and abstract meaning representation (AMR) parses and then embed it for retrieval. During querying, we apply the same transformation to the query and retrieve the most relevant sentences from the grounded source text for question answering. We evaluate GroundedKG-RAG on examples from the NarrativeQA dataset and find that it performs on par with a state-of-the art proprietary long-context model at smaller cost and outperforms a competitive baseline. Additionally, our GroundedKG is interpretable and readable by humans, facilitating auditing of results and error analysis.