G3: When Graph Neural Networks Meet Parallel Graph Processing Systems on GPUs
本文展示了G3,一种用于图神经网络(GNN)训练的框架,由图形处理单元(gpu)上的图处理系统定制。G3旨在通过使用并行图处理系统支持图结构操作来提高GNN的训练效率。 G3允许用户通过以下两种方式利用gpu的大规模并行和其他架构特性:通过编写一组灵活的api(应用程序编程接口)顺序的C/ c++代码构建GNN层;使用G3中提供的基本GNN操作和层创建GNN模型。G3的运行时系统在GPU上自动执行用户定义的gnn,并启用一系列以图为中心的优化。展示了用G3开发一些流行GNN模型的步骤,以及G3相对于现有GNN训练系统(即PyTorch和TensorFlow)的优越性能。