arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.09079cs.DCcs.AIcs.LG
收藏

D3-GNN:面向流式图神经网络的动态分布式数据流

D3-GNN: Dynamic Distributed Dataflow for Streaming Graph Neural Networks

University of Warwick(华威大学);Fujitsu Research of Europe(富士通欧洲研究院);Amazon Web Services(亚马逊网络服务)

Rustam Guliyev, Aparajita Haldar, Hakan Ferhatosmanoglu

AI总结:

针对流式图神经网络的算法与系统挑战,提出分布式混合并行系统D3-GNN,通过流式聚合器、分布式计算图及层间层内加窗方案处理更新,相较DGL吞吐量提升约76倍。

摘要:

流式图上的图神经网络(GNN)模型既面临持续捕捉动态状态的算法挑战,也存在推理与训练阶段优化延迟、内存和吞吐量的系统挑战。我们提出D3-GNN,这是首个分布式混合并行流式GNN系统,专为在线查询场景下处理实时图更新而设计。该系统解决了数据管理、算法与系统层面的多重挑战,能够持续捕捉图的动态状态并更新节点表示,同时具备容错能力,实现了最优的延迟、负载均衡与吞吐量。D3-GNN利用流式GNN聚合器以及展开式分布式计算图架构来处理级联图更新。为应对数据倾斜和邻域爆炸问题,我们引入了层间与层内加窗前向传播方案。在大规模图流上的实验表明,D3-GNN具备高效性与可扩展性。与DGL相比,D3-GNN在流式工作负载下的吞吐量提升约76倍。加窗优化在更高并行度下进一步将运行时间降低约10倍,消息量最多减少15倍。

英文摘要:

Graph Neural Network (GNN) models on streaming graphs entail algorithmic challenges to continuously capture its dynamic state, as well as systems challenges to optimize latency, memory, and throughput during both inference and training. We present D3-GNN, the first distributed, hybrid-parallel, streaming GNN system designed to handle real-time graph updates under online query setting. Our system addresses data management, algorithmic, and systems challenges, enabling continuous capturing of the dynamic state of the graph and updating node representations with fault-tolerance and optimal latency, load-balance, and throughput. D3-GNN utilizes streaming GNN aggregators and an unrolled, distributed computation graph architecture to handle cascading graph updates. To counteract data skew and neighborhood explosion issues, we introduce inter-layer and intra-layer windowed forward pass solutions. Experiments on large-scale graph streams demonstrate that D3-GNN achieves high efficiency and scalability. Compared to DGL, D3-GNN achieves a significant throughput improvement of about 76x for streaming workloads. The windowed enhancement further reduces running times by around 10x and message volumes by up to 15x at higher parallelism.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑