arXiv · 2207.12000
GNN Transformation Framework for Improving Efficiency and Scalability
Abstract
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.
Explore related subjects
Keep this discovery
Seiji Maekawa, Yuya Sasaki, George Fletcher, Makoto Onizuka. 2022-07-25. GNN Transformation Framework for Improving Efficiency and Scalability. https://arxiv.org/abs/2207.12000
Cite the original work for its findings. Save a collection to share your selection of sources.