Statistical Guarantees for Link Prediction using Graph Neural Networks

This paper derives statistical guarantees for the performance of Graph Neural Networks (GNNs) in link prediction tasks on graphs generated by a graphon. We propose a linear GNN architecture (LG-GNN) that produces consistent estimators for the underlying edge probabilities. We establish a bound on the mean squared error and give guarantees on the ability of LG-GNN to detect high-probability edges. Our guarantees hold for both sparse and dense graphs. Finally, we demonstrate some of the shortcomings of the classical GCN …

architecture consistent cs.lg edge error generated gnn gnns graph graph neural networks graphs linear link prediction math.st mean networks neural networks paper performance prediction statistical stat.ml stat.th tasks