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

ArXiv ID: 2408.04948 “View on arXiv”

Authors: Unknown

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

Keywords: Retrieval Augmented Generation (RAG), Knowledge Graphs (KGs), Large Language Models (LLMs), Vector Database, HybridRAG, General Finance

Complexity vs Empirical Score

  • Math Complexity: 2.0/10
  • Empirical Rigor: 7.5/10
  • Quadrant: Street Traders
  • Why: The paper focuses on an implementation-heavy RAG methodology with empirical validation on financial datasets, but relies on established ML techniques without dense mathematical derivations.
  flowchart TD
    A["Research Goal: Enhance Info Extraction from Financial Text"] --> B["Input: Financial Earnings Call Transcripts"]
    B --> C["HybridRAG Methodology"]
    C --> D{"Parallel Retrieval"}
    D --> E["VectorRAG: Vector DB Search"]
    D --> F["GraphRAG: Knowledge Graph Search"]
    E & F --> G["Context Synthesis & LLM Generation"]
    G --> H["Outcome: HybridRAG Outperforms VectorRAG & GraphRAG"]
    H --> I["Result: Accurate & Contextual Q&A"]