HG-RAG: Hierarchical Graph-Enhanced Retrieval-Augmented Generation for Power Systems
Abstract
Retrieval-augmented generation (RAG) has shown strong potential for knowledge-intensive tasks, yet its performance degrades sharply when applied to structured long-context documents in power systems, where dense entity–relation dependencies, cross-document references, and strict traceability requirements exist. To address this Structured Long-Context RAG (SLCRAG) challenge, this paper proposes a hierarchical graph-enhanced RAG (HG-RAG) framework tailored for power system question answering. HG-RAG constructs a globally consistent knowledge graph via sliding-window entity–relation extraction to mitigate semantic fragmentation, and employs multi-granularity structured indexing for precise entity/relation retrieval. A hierarchical structured retrieval mechanism with multi-hop expansion and semantic distillation maximizes recall while minimizing redundancy. Furthermore, a regex-enhanced retrieval module records authoritative file_path provenance and constrains downstream retrieval to the same source documents, effectively eliminating cross-document interference—especially in cases where different documents contain similar entities and relations. Combined with version control and deduplication-merging, HG-RAG supports incremental knowledge updates with minimal forgetting and negligible token overhead. Experimental results on a domain-authentic power system QA dataset demonstrate that HG-RAG outperforms LightRAG and GraphRAG, achieving up to 85.47% accuracy in short-answer tasks with significantly lower token consumption. Ablation studies confirm that semantic distillation primarily improves precision and efficiency, while regex-enhanced retrieval safeguards recall in edge cases.
Fields of Study & Subjects
Advanced Graph Neural Networks
Topic Modeling
Multimodal Machine Learning Applications
Publication details
- Journal / source
- Electronics
- Publisher
- MDPI AG
- Volume
- 15
- Issue
- 7
- Pages
- 1445–1445
- Publication date
- 30 March 2026
- Language
- EN