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

Autor:	Sarmah, Bhaskarjit, Hall, Benika, Rao, Rohan, Patel, Sunil, Pasquali, Stefano, Mehta, Dhagash
Rok vydání:	2024
Předmět:	Computer Science - Computation and Language Computer Science - Machine Learning Quantitative Finance - Statistical Finance Statistics - Applications Statistics - Machine Learning
Druh dokumentu:	Working Paper
Popis:	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 Comment: 9 pages, 2 figures, 5 tables
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2408.04948 Zobrazit plný text záznamu View this record from Arxiv