Comprehensive Evaluation of GNN Training Systems: A Data Management Perspective

Hao Yuan, Yajiong Liu, Yanfeng Zhang, Xin Ai, Qiange Wang +3 more
11/22/2023
cs.LGcs.DC

Official code: iDC-NEU/NeutronBench (15 stars, Apache-2.0, last commit 2025-04-21).

Abstract

Many Graph Neural Network (GNN) training systems have emerged recently to support efficient GNN training. Since GNNs embody complex data dependencies between training samples, the training of GNNs should address distinct challenges different from DNN training in data management, such as data partitioning, batch preparation for mini-batch training, and data transferring between CPUs and GPUs. These factors, which take up a large proportion of training time, make data management in GNN training more significant. This paper reviews GNN training from a data management perspective and provides a comprehensive analysis and evaluation of the representative approaches. We conduct extensive experiments on various benchmark datasets and show many interesting and valuable results. We also provide some practical tips learned from these experiments, which are helpful for designing GNN training systems in the future.

View on arXivView PDF

Code Implementations(1)

Apache-2.0
151C, C++Sep 29, 20231 years agoApache-2.0

Cite this paper

@article{yuan2023comprehensive,
  title  = {Comprehensive Evaluation of GNN Training Systems: A Data Management Perspective},
  author = {Hao Yuan and Yajiong Liu and Yanfeng Zhang and Xin Ai and Qiange Wang and Chaoyi Chen and Yu Gu and Ge Yu},
  year   = {2023},
  eprint = {2311.13279},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2311.13279v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2311.13279.svg)](https://codeofpaper.com/paper/2311.13279)

Discussion