UniAI-GraphRAG: Synergizing Ontology-Guided Extraction, Multi-Dimensional Clustering, and Dual-Channel Fusion for Robust Multi-Hop Reasoning

TL;DR

UniAI-GraphRAG introduces an advanced framework for multi-hop reasoning that combines ontology-guided extraction, multi-dimensional clustering, and dual-channel retrieval to enhance accuracy and robustness in domain-specific QA tasks.

Contribution

The paper presents three novel techniques—ontology-guided extraction, multi-dimensional clustering, and dual-channel retrieval—that significantly improve GraphRAG performance across various complex queries.

Findings

Outperforms existing open-source GraphRAG models in F1 scores.

Enhances reasoning accuracy in multi-hop and temporal queries.

Demonstrates robustness across multiple domain-specific benchmarks.

Abstract

Retrieval-Augmented Generation (RAG) systems face significant challenges in complex reasoning, multi-hop queries, and domain-specific QA. While existing GraphRAG frameworks have made progress in structural knowledge organization, they still have limitations in cross-industry adaptability, community report integrity, and retrieval performance. This paper proposes UniAI-GraphRAG, an enhanced framework built upon open-source GraphRAG. The framework introduces three core innovations: (1) Ontology-Guided Knowledge Extraction that uses predefined Schema to guide LLMs in accurately identifying domain-specific entities and relations; (2) Multi-Dimensional Community Clustering Strategy that improves community completeness through alignment completion, attribute-based clustering, and multi-hop relationship clustering; (3) Dual-Channel Graph Retrieval Fusion that balances QA accuracy and performance through hybrid graph and community retrieval. Evaluation results on MultiHopRAG benchmark show that UniAI-GraphRAG outperforms mainstream open source solutions (e.g.LightRAG) in comprehensive F1 scores, particularly in inference and temporal queries. The code is available at https://github.com/UnicomAI/wanwu/tree/main/rag/rag_open_source/rag_core/graph.