FastGL通过优化采样三阶段,实现超大规模图神经网络训练的高效加速。
FastGL: A GPU-Efficient Framework for Accelerating Sampling-Based GNN Training at Large Scale
- 利用图结构重叠性设计匹配重排策略,降低内存访问开销。
- 基于GPU内存层级特性优化计算,缓解不规则数据访问问题。
- 融合映射技术减少采样同步开销,提升整体训练效率。
图神经网络(GNN)在非欧几里得图数据上表现出色,在各类图任务中取得突破性成果。为在包含数十亿节点与边的大规模图上训练GNN,现有框架普遍采用采样训练。然而,深入分析发现,现有采样训练框架效率受限于采样、内存IO和计算三个阶段的瓶颈。为此,我们提出FastGL,一个面向大规模图神经网络采样训练的GPU高效框架,通过同时优化上述三阶段,结合GPU特性和图结构特征实现加速。具体而言,通过挖掘图结构内在重叠性,FastGL提出匹配重排策略,在不增加GPU内存开销的前提下减少数据传输量;进一步,采用内存感知计算方法,利用GPU内存分层特性缓解计算过程中的不规则数据访问;此外,引入融合映射机制以降低采样阶段的同步开销。大量实验表明,FastGL相比当前最先进的PyG、DGL和GNNLab框架,平均分别提速11.8倍、2.2倍和1.5倍。代码已开源:https://github.com/a1bc2def6g/fastgl-ae。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have shown great superiority on non-Euclidean graph data, achieving ground-breaking performance on various graph-related tasks. As a practical solution to train GNN on large graphs with billions of nodes and edges, the sampling-based training is widely adopted by existing training frameworks. However, through an in-depth analysis, we observe that the efficiency of existing sampling-based training frameworks is still limited due to the key bottlenecks lying in all three phases of sampling-based training, i.e., subgraph sample, memory IO, and computation. To this end, we propose FastGL, a GPU-efficient Framework for accelerating sampling-based training of GNN at Large scale by simultaneously optimizing all above three phases, taking into account both GPU characteristics and graph structure. Specifically, by exploiting the inherent overlap within graph structures, FastGL develops the Match-Reorder strategy to reduce the data traffic, which accelerates the memory IO without incurring any GPU memory overhead. Additionally, FastGL leverages a Memory-Aware computation method, harnessing the GPU memory's hierarchical nature to mitigate irregular data access during computation. FastGL further incorporates the Fused-Map approach aimed at diminishing the synchronization overhead during sampling. Extensive experiments demonstrate that FastGL can achieve an average speedup of 11.8x, 2.2x and 1.5x over the state-of-the-art frameworks PyG, DGL, and GNNLab, respectively.Our code is available at https://github.com/a1bc2def6g/fastgl-ae.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。