DGRAG: Distributed Graph-based Retrieval-Augmented Generation in Edge-Cloud Systems

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhou, Wenqing, Yan, Yuxuan, Yang, Qianqian
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908792757682176
author Zhou, Wenqing
Yan, Yuxuan
Yang, Qianqian
author_facet Zhou, Wenqing
Yan, Yuxuan
Yang, Qianqian
contents Retrieval-Augmented Generation (RAG) improves factuality by grounding LLMs in external knowledge, yet conventional centralized RAG requires aggregating distributed data, raising privacy risks and incurring high retrieval latency and cost. We present DGRAG, a distributed graph-driven RAG framework for edge-cloud collaborative systems. Each edge device organizes local documents into a knowledge graph and periodically uploads subgraph-level summaries to the cloud for lightweight global indexing without exposing raw data. At inference time, queries are first answered on the edge; a gate mechanism assesses the confidence and consistency of multiple local generations to decide whether to return a local answer or escalate the query. For escalated queries, the cloud performs summary-based matching to identify relevant edges, retrieves supporting evidence from them, and generates the final response with a cloud LLM. Experiments on distributed question answering show that DGRAG consistently outperforms decentralized baselines while substantially reducing cloud overhead.
format Preprint
id arxiv_https___arxiv_org_abs_2505_19847
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DGRAG: Distributed Graph-based Retrieval-Augmented Generation in Edge-Cloud Systems
Zhou, Wenqing
Yan, Yuxuan
Yang, Qianqian
Artificial Intelligence
Distributed, Parallel, and Cluster Computing
Retrieval-Augmented Generation (RAG) improves factuality by grounding LLMs in external knowledge, yet conventional centralized RAG requires aggregating distributed data, raising privacy risks and incurring high retrieval latency and cost. We present DGRAG, a distributed graph-driven RAG framework for edge-cloud collaborative systems. Each edge device organizes local documents into a knowledge graph and periodically uploads subgraph-level summaries to the cloud for lightweight global indexing without exposing raw data. At inference time, queries are first answered on the edge; a gate mechanism assesses the confidence and consistency of multiple local generations to decide whether to return a local answer or escalate the query. For escalated queries, the cloud performs summary-based matching to identify relevant edges, retrieves supporting evidence from them, and generates the final response with a cloud LLM. Experiments on distributed question answering show that DGRAG consistently outperforms decentralized baselines while substantially reducing cloud overhead.
title DGRAG: Distributed Graph-based Retrieval-Augmented Generation in Edge-Cloud Systems
topic Artificial Intelligence
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2505.19847