Graph Coarsening via Supervised Granular-Ball for Scalable Graph Neural Network Training

TL;DR

This paper introduces a novel graph coarsening method using granular-ball computing that adaptively compresses graphs, significantly boosting GNN training efficiency and scalability while maintaining accuracy and robustness.

Contribution

The proposed method is the first to use granular-ball computing for adaptive graph coarsening, eliminating the need for predefined coarsening rates and improving scalability.

Findings

Achieves up to 20x graph size reduction without accuracy loss

Maintains comparable accuracy to original graph training

Demonstrates robustness against noise injections

Abstract

Graph Neural Networks (GNNs) have demonstrated significant achievements in processing graph data, yet scalability remains a substantial challenge. To address this, numerous graph coarsening methods have been developed. However, most existing coarsening methods are training-dependent, leading to lower efficiency, and they all require a predefined coarsening rate, lacking an adaptive approach. In this paper, we employ granular-ball computing to effectively compress graph data. We construct a coarsened graph network by iteratively splitting the graph into granular-balls based on a purity threshold and using these granular-balls as super vertices. This granulation process significantly reduces the size of the original graph, thereby greatly enhancing the training efficiency and scalability of GNNs. Additionally, our algorithm can adaptively perform splitting without requiring a predefined coarsening rate. Experimental results demonstrate that our method achieves accuracy comparable to training on the original graph. Noise injection experiments further indicate that our method exhibits robust performance. Moreover, our approach can reduce the graph size by up to 20 times without compromising test accuracy, substantially enhancing the scalability of GNNs.