English

Single-GPU GNN Systems: Traps and Pitfalls

Machine Learning 2024-02-07 v1 Distributed, Parallel, and Cluster Computing

Abstract

The current graph neural network (GNN) systems have established a clear trend of not showing training accuracy results, and directly or indirectly relying on smaller datasets for evaluations majorly. Our in-depth analysis shows that it leads to a chain of pitfalls in the system design and evaluation process, questioning the practicality of many of the proposed system optimizations, and affecting conclusions and lessons learned. We analyze many single-GPU systems and show the fundamental impact of these pitfalls. We further develop hypotheses, recommendations, and evaluation methodologies, and provide future directions. Finally, a new reference system is developed to establish a new line of optimizations rooted in solving the system-design pitfalls efficiently and practically. The proposed design can productively be integrated into prior works, thereby truly advancing the state-of-the-art.

Keywords

Cite

@article{arxiv.2402.03548,
  title  = {Single-GPU GNN Systems: Traps and Pitfalls},
  author = {Yidong Gong and Arnab Tarafder and Saima Afrin and Pradeep Kumar},
  journal= {arXiv preprint arXiv:2402.03548},
  year   = {2024}
}
R2 v1 2026-06-28T14:39:23.894Z