Bottleneck Analysis of Dynamic Graph Neural Network Inference on CPU and GPU
School of Electrical and Computer Engineering, Georgia Institute of Technology
arXiv:2210.03900 [cs.AR], (8 Oct 2022)
@misc{https://doi.org/10.48550/arxiv.2210.03900,
doi={10.48550/ARXIV.2210.03900},
url={https://arxiv.org/abs/2210.03900},
author={Chen, Hanqiu and Alhinai, Yahya and Jiang, Yihan and Na, Eunjee and Hao, Cong},
keywords={Hardware Architecture (cs.AR), Machine Learning (cs.LG), FOS: Computer and information sciences, FOS: Computer and information sciences},
title={Bottleneck Analysis of Dynamic Graph Neural Network Inference on CPU and GPU},
publisher={arXiv},
year={2022},
copyright={arXiv.org perpetual, non-exclusive license}
}
Dynamic graph neural network (DGNN) is becoming increasingly popular because of its widespread use in capturing dynamic features in the real world. A variety of dynamic graph neural networks designed from algorithmic perspectives have succeeded in incorporating temporal information into graph processing. Despite the promising algorithmic performance, deploying DGNNs on hardware presents additional challenges due to the model complexity, diversity, and the nature of the time dependency. Meanwhile, the differences between DGNNs and static graph neural networks make hardware-related optimizations for static graph neural networks unsuitable for DGNNs. In this paper, we select eight prevailing DGNNs with different characteristics and profile them on both CPU and GPU. The profiling results are summarized and analyzed, providing in-depth insights into the bottlenecks of DGNNs on hardware and identifying potential optimization opportunities for future DGNN acceleration. Followed by a comprehensive survey, we provide a detailed analysis of DGNN performance bottlenecks on hardware, including temporal data dependency, workload imbalance, data movement, and GPU warm-up. We suggest several optimizations from both software and hardware perspectives. This paper is the first to provide an in-depth analysis of the hardware performance of DGNN Code is available.
October 16, 2022 by hgpu
