arrow
返回

Practical Near-Data-Processing Architecture for Large-Scale Distributed Graph Neural Network

delete2022-01-01
delete1
delete
OA
AI
L
Linyong Huang *
Z
Zhe Zhang
S
Shuangchen Li
D
Dimin Niu
Y
Yijin Guan
H
Hongzhong Zheng
Y
Yuan Xie
DOI:10.1109/ACCESS.2022.3169423delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

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

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

A
alibaba group
学者数:
1.1K
论文数: 789
被引数: 0
Z
zhejiang university
学者数:
17.7W
论文数: 12.1W
被引数: 152
引用论文

引用论文

Mercury isotopic composition of igneous rocks from an accretionary orogen: Implications for lithospheric recycling
err2022-05-27
err0
errOAAI
errChangzhou Deng; Jun Gou; Deyou Sun; Guangyi Sun; Zhendong Tian; Bernd Lehmann; Frédéric Moynier; Runsheng Yin
err分享
err收藏
学者 查看更多内容