We propose a framework that automatically transforms non-scalable GNNs into precomputation-based GNNs which are efficient and scalable for large-scale graphs. The advantages of our framework are two-fold; 1) it transforms various non-scalable GNNs to scale well to large-scale graphs by separating local feature aggregation from weight learning in their graph convolution, 2) it efficiently executes precomputation on GPU for large-scale graphs by decomposing their edges into small disjoint and balanced sets. Through extensive experiments with large-scale graphs, we demonstrate that the transformed GNNs run faster in training time than existing GNNs while achieving competitive accuracy to the state-of-the-art GNNs. Consequently, our transformation framework provides simple and efficient baselines for future research on scalable GNNs.
@article{arxiv.2207.12000,
title = {GNN Transformation Framework for Improving Efficiency and Scalability},
author = {Seiji Maekawa and Yuya Sasaki and George Fletcher and Makoto Onizuka},
journal= {arXiv preprint arXiv:2207.12000},
year = {2022}
}