HybridRAG: Integrating Knowledge Graphs and Vector Retrieval Augmented Generation for Efficient Information Extraction

TL;DR

HybridRAG combines knowledge graphs and vector retrieval to improve information extraction from complex financial texts, outperforming existing methods in accuracy and relevance.

Contribution

This paper introduces HybridRAG, a novel method integrating knowledge graphs with vector retrieval for enhanced question-answering in financial document analysis.

Findings

HybridRAG outperforms individual VectorRAG and GraphRAG methods in retrieval accuracy.

HybridRAG generates more accurate and contextually relevant answers.

The approach has potential applications beyond finance.

Abstract

Extraction and interpretation of intricate information from unstructured text data arising in financial applications, such as earnings call transcripts, present substantial challenges to large language models (LLMs) even using the current best practices to use Retrieval Augmented Generation (RAG) (referred to as VectorRAG techniques which utilize vector databases for information retrieval) due to challenges such as domain specific terminology and complex formats of the documents. We introduce a novel approach based on a combination, called HybridRAG, of the Knowledge Graphs (KGs) based RAG techniques (called GraphRAG) and VectorRAG techniques to enhance question-answer (Q&A) systems for information extraction from financial documents that is shown to be capable of generating accurate and contextually relevant answers. Using experiments on a set of financial earning call transcripts documents which come in the form of Q&A format, and hence provide a natural set of pairs of ground-truth Q&As, we show that HybridRAG which retrieves context from both vector database and KG outperforms both traditional VectorRAG and GraphRAG individually when evaluated at both the retrieval and generation stages in terms of retrieval accuracy and answer generation. The proposed technique has applications beyond the financial domain