返回
Practical Near-Data-Processing Architecture for Large-Scale Distributed Graph Neural Network
DOI:10.1109/ACCESS.2022.3169423.png)
摘要
En 中文
Graph Neural Networks have drawn tremendous attention in the past few years due to their convincing performance and high interpretability in various graph-based tasks like link prediction and node classification. With the ever-growing graph size in the real world, especially for industrial graphs at a billion-level, the storage of graphs can easily consume Terabytes so that the process of GNNs has to be processed in a distributed manner. As a result, the execution could be inefficient due to the expensive cross-node communication and irregular memory access. Various GNN accelerators have been proposed for efficient GNN processing. They, however, mainly focused on small and medium-size graphs, which is not applicable to large-scale distributed graphs. In this paper, we present a practical Near-Data-Processing architecture based on a memory-pool system for large-scale distributed GNNs. We propose a customized memory fabric interface to construct the memory pool for low-latency and high throughput cross-node communication, which can provide flexible memory allocation and strong scalability. A practical Near-Data-Processing design is proposed for efficient work offloading and bandwidth utilization improvement. Moreover, we also introduce a partition and scheduling scheme to further improve performance and achieve workload balance. Comprehensive evaluations demonstrate that the proposed architecture can achieve up to 27 x and 8 x higher training speed compared to two state-of-the-art distributed GNN frameworks: Deep Graph Library and P-3, respectively.
Keyword:
Graph neural network
large-scale graph processing
memory pool
near data processing
期刊
IF:
3.6
论文数:
9.8W
被引数:
29.4W
机构
引用论文
The association of CNTNAP2 rs2710102 and ENGRAILED-2 rs1861972 genes polymorphism and autism in Iranian population伊朗人群中CNTNAP2基因rs2710102和ENGRAILED-2基因rs1861972多态性与自闭症的相关性
Meta Gene
IF0
Conjugated heat transfer and temperature distributions in a gas turbine combustion liner under base-load operation基本负荷运行下燃气轮机燃烧衬里中的共轭传热和温度分布
Effects of Deletion of Mutant Huntingtin in Steroidogenic Factor 1 Neurons on the Psychiatric and Metabolic Phenotype in the BACHD Mouse Model of Huntington Disease
PLoS ONE
IF0
Mercury isotopic composition of igneous rocks from an accretionary orogen: Implications for lithospheric recycling
Geology
IF0

