REVIEW 2 major objections 5 minor 1 cited by
Demystifying Distributed Training of Graph Neural Networks for Link Prediction
T0 review · 2 major / 5 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read The paper argues that distributed GNN link prediction degrades because partitioning breaks neighbor information and negative sampling is local, then fixes both with sparsification.
desk verdict Useful system insight and a plausible diagnostic claim, but the training/eval asymmetry between positive and negative neighborhoods could inflate the headline accuracy; a symmetric-neighborhood ablation should gate the claim. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The central mechanism is negative sampling over sparsified remote subgraphs. Each worker draws positive samples from its own unsimplified partition, while negative destinations are chosen from the entire graph, with remote partitions stored in shared memory in sparsified form. The sparsification is built on the effective-resistance identity: an edge is sampled with probability $p_{u,v}\propto 1/d_u+1/d_v$, a degree-only approximation of effective resistance that the paper justifies through a spectral sparsification theorem and a degree-based bound on resistance. Because a global negative sample's endpoint in a remote partition now has far fewer $k$-hop neighbors in the computational graph, the bytes transferred per negative sample drop sharply, while the sample space for choosing negative destinations remains the entire graph.
What would settle it
Train exactly the SpLPG procedure, then evaluate on a test set where every negative pair is represented with full $k$-hop neighborhoods for both endpoints (or, symmetrically, train with sparsified neighborhoods for positives too); if Hits@100 drops well below the centralized baseline, the accuracy gain is an artifact of the density difference between positive and negative samples rather than better link prediction.
Extended reading notes
Core claim
On its own terms, the paper's discovery is that the performance degradation in distributed GNN link prediction is not mainly a data-distribution mismatch introduced by partitioning; it is the combination of partitioned neighbor loss and a shrunken negative-sample space. The authors show that giving every worker access to the whole graph restores centralized-level accuracy, then propose SpLPG to obtain the same effect more cheaply: each worker retains full neighborhoods for its own partition, while remote partitions are sparsified by sampling edges with probability proportional to a degree-based approximation of effective resistance and are used only when drawing negative samples. The experimental claim is that this design recovers centralized-level Hits@100 on most datasets and reduces communication traffic by up to about 80% compared to full data sharing, with the residual accuracy gap concentrated on small graphs such as Citeseer and Cora.
Load-bearing premise
The accuracy claim rests on the assumption that a model trained with full neighborhoods for positive samples but sparsified neighborhoods for negative samples does not learn to exploit that density difference as a shortcut; if it does, the reported Hits@100 would overstate true link-prediction ability.
Editorial extensions
If this is right
- If the reported results hold, distributed GNN link prediction no longer needs full remote neighborhoods on every worker; sparsified copies suffice for accurate global negative sampling.
- The ablation showing that both full local neighbors and global negative samples are needed gives a concrete diagnostic: a distributed system that fixes only one of the two will still see degraded accuracy.
- The communication saving is tunable: at about 85% edge removal, the paper measures roughly 68% cost savings, and more aggressive sparsification trades some accuracy for larger savings.
- Because the experiments cover GCN, GraphSAGE, GAT, and GATv2, the method is presented as architecture-agnostic rather than tied to one GNN variant.
Reading between the lines
- A natural next experiment is to test SpLPG on graphs where remote $k$-hop neighborhoods are precomputed and cached; part of the reported communication saving may shift to storage rather than disappear.
- The positive/negative asymmetry in neighborhood density is a confound worth checking directly: if positive samples always see full neighborhoods while negative samples see sparsified ones, Hits@100 could partly reflect a density cue. Evaluating test samples with both endpoints expanded to full neighborhoods would settle whether the score is genuine link prediction.
- The fixed global sparsification level $\alpha=0.15$ leaves small graphs behind; an adaptive per-partition sparsity based on local density is a testable extension that might close the remaining gap on Cora and Citeseer.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies distributed training of GNNs for link prediction and identifies two root causes of accuracy degradation: information loss from graph partitioning and the restricted negative-sample space when workers only access local subgraphs. The authors propose SpLPG, which keeps full local neighborhoods for positive-sample computation, sparsifies remote subgraphs used only for negative sampling, and shares these sparsified subgraphs across workers to reduce communication cost. Experiments on nine datasets with GCN, GraphSAGE, GAT, and GATv2 report that SpLPG reduces communication overhead by up to about 80% while roughly preserving centralized link-prediction accuracy.
Significance. If the accuracy-preservation claim holds, SpLPG is a practical contribution to distributed GNN training for link prediction, where existing frameworks are mainly optimized for node classification. The paper provides code, extensive experiments across multiple models and datasets, and a useful ablation (Fig. 12) isolating the effects of full-neighbor information and global negative sampling. However, the central claim is weakened by a potential confound in the training/evaluation protocol: positive and negative samples have asymmetric neighborhood densities, and the test-time graph is unspecified. The theoretical part (Theorems 1–2) is taken from prior work, and the proposed sparsification uses only a degree-based approximation, so the contribution is primarily empirical. Because of the confound, the significance is conditional on the outcome of a symmetric-neighborhood test.
major comments (2)
- [Algorithm 1, Lines 20–21; §V-B; footnote 2] Same as above.
- [§V-B, Fig. 11] Same as above.
minor comments (5)
- [Figure 4] Same as above.
- [§V-A] Same as above.
- [Abstract; Table III] Same as above.
- [§V-A, Fig. 11] Same as above.
- [§IV-A, Fig. 6] Same as above.
Circularity Check
No circularity found; the sparsification theory is imported from external mathematical results, the root-cause claims are empirical and benchmarked against centralized training, and no fitted quantity is renamed as a prediction.
full rationale
I walked the derivation chain rather than the narrative. SpLPG partitions the graph with METIS, keeps each worker's full local neighborhood, sparsifies remote subgraphs by effective-resistance sampling, and draws global negative samples from sparsified remote subgraphs. The spectral guarantee (Theorem 1) is quoted from Spielman-Srivastava [34] and the degree-based resistance bound (Theorem 2) from Lovasz [39]; both are external, machine-independent results that do not depend on SpLPG's own accuracy claims. The paper's main claims are empirical: the performance drop is diagnosed by comparing distributed variants with centralized training (Figs. 3, 4, 12), and the accuracy/communication trade-off is measured against complete-data-sharing baselines and centralized models (Figs. 8, 10, 11). No equation is defined in terms of the quantity it is said to predict, and no fitted parameter is relabeled as a prediction; the sparsification level alpha is a hyperparameter explored in Table III, not a parameter fitted to the reported Hits@100 values. The only self-citations ([8], [10]) appear as introductory examples of GNN-related work and are not load-bearing. The nearest passage to a caveat is in Sec. IV-A and footnote 2, where the paper concedes that each global negative sample from outside the assigned subgraph 'comes with a much fewer number of k-hop neighbors' after sparsification. That asymmetry between positive and negative neighborhoods could confound the accuracy comparison if test-time embeddings are computed in the same way, but it is an evaluation-validity concern, not a circularity: it does not make any derivation equivalent to its own inputs. I therefore find no significant circularity.
Assumptions & free parameters
free parameters (1)
- sparsification level alpha =
0.15
assumptions (3)
- domain assumption Sparsifying remote subgraphs by sampling edges with probability proportional to 1/d_u + 1/d_v preserves enough graph structure for negative-sample neighbor aggregation to yield an unbiased training signal.
- domain assumption Random 80/10/10 splits produce validation/test sets whose negative sampling matches the training distribution under the same distributed protocol.
- domain assumption The communication cost metric (total bytes transferred from master shared memory to workers per epoch) is a faithful proxy for real distributed communication overhead.
Cite this review
Pith. "Pith review of Demystifying Distributed Training of Graph Neural Networks for Link Prediction." pith.science (2026). https://pith.science/paper/3U2DPN36
@misc{pith2026250620818,
author = {Pith},
title = {Pith review of: Demystifying Distributed Training of Graph Neural Networks for Link Prediction},
year = {2026},
howpublished = {\url{https://pith.science/paper/3U2DPN36}},
note = {Machine review of arXiv:2506.20818}
}
read the original abstract
Graph neural networks (GNNs) are powerful tools for solving graph-related problems. Distributed GNN frameworks and systems enhance the scalability of GNNs and accelerate model training, yet most are optimized for node classification. Their performance on link prediction remains underexplored. This paper demystifies distributed training of GNNs for link prediction by investigating the issue of performance degradation when each worker trains a GNN on its assigned partitioned subgraph without having access to the entire graph. We discover that the main sources of the issue come from not only the information loss caused by graph partitioning but also the ways of drawing negative samples during model training. While sharing the complete graph information with each worker resolves the issue and preserves link prediction accuracy, it incurs a high communication cost. We propose SpLPG, which effectively leverages graph sparsification to mitigate the issue of performance degradation at a reduced communication cost. Experiment results on several public real-world datasets demonstrate the effectiveness of SpLPG, which reduces the communication overhead by up to about 80% while mostly preserving link prediction accuracy.
Figures
Figures from the paper (10 more)
Forward citations
Cited by 1 Pith paper
-
Graph Neural Network for Product Recommendation on the Amazon Co-purchase Graph
A benchmark of four GNN architectures (LightGCN, GraphSAGE, GAT, PinSAGE) for link prediction on the Amazon co-purchase graph, reporting trade-offs between accuracy, training cost, and scalability.
Reference graph
Works this paper leans on
-
[1]
The link prediction problem for social networks,
D. Liben-Nowell and J. Kleinberg, “The link prediction problem for social networks,” inACM CIKM, 2003, pp. 556–559
work page 2003
-
[2]
Learning entity and relation embeddings for knowledge graph completion,
Y . Lin, Z. Liu, M. Sun, Y . Liu, and X. Zhu, “Learning entity and relation embeddings for knowledge graph completion,” inAAAI, 2015
work page 2015
-
[3]
Link prediction in complex networks: A survey,
L. L ¨u and T. Zhou, “Link prediction in complex networks: A survey,” Physica A: Statistical Mechanics and its Applications, pp. 1150–1170, 2011
work page 2011
-
[4]
Link prediction tech- niques, applications, and performance: A survey,
A. Kumar, S. S. Singh, K. Singh, and B. Biswas, “Link prediction tech- niques, applications, and performance: A survey,”Physica A: Statistical Mechanics and its Applications, p. 124289, 2020
work page 2020
-
[5]
Graph neural networks: foundation, frontiers and applications,
L. Wu, P. Cui, J. Pei, L. Zhao, and X. Guo, “Graph neural networks: foundation, frontiers and applications,” inACM KDD, 2022, pp. 4840– 4841
work page 2022
- [6]
-
[7]
Graph representation learning,
W. L. Hamilton, “Graph representation learning,”Synthesis Lectures on Artificial Intelligence and Machine Learning, pp. 1–159, 2020
work page 2020
-
[8]
Characterizing the efficiency of graph neural network frameworks with a magnifying glass,
X. Huang, J. Kim, B. Rees, and C.-H. Lee, “Characterizing the efficiency of graph neural network frameworks with a magnifying glass,” inIEEE IISWC, 2022, pp. 160–170
work page 2022
Show all 44 references
-
[9]
Heterogeneous spatio- temporal graph convolution network for traffic forecasting with missing values,
W. Zhong, Q. Suo, X. Jia, A. Zhang, and L. Su, “Heterogeneous spatio- temporal graph convolution network for traffic forecasting with missing values,” inIEEE ICDCS, 2021, pp. 707–717
2021
-
[10]
GRAFICS: Graph embedding-based floor identification using crowdsourced RF signals,
W. Zhuo, Z. Zhao, K. H. Chiu, S. Li, S. Ha, C.-H. Lee, and S.-H. G. Chan, “GRAFICS: Graph embedding-based floor identification using crowdsourced RF signals,” inIEEE ICDCS, 2022, pp. 1051–1061
2022
-
[11]
Embedding communication for federated graph neural networks with privacy guarantees,
X. Wu, Z. Ji, and C.-L. Wang, “Embedding communication for federated graph neural networks with privacy guarantees,” inIEEE ICDCS, 2023, pp. 305–315
2023
-
[12]
Mega: More efficient graph attention for GNNs,
W. Deng and J. Rao, “Mega: More efficient graph attention for GNNs,” inIEEE ICDCS, 2024, pp. 71–81
2024
-
[13]
Semi-supervised classification with graph convolutional networks,
T. N. Kipf and M. Welling, “Semi-supervised classification with graph convolutional networks,” inICLR, 2017
2017
-
[14]
Graph attention networks,
P. Veli ˇckovi´c, G. Cucurull, A. Casanova, A. Romero, P. Li `o, and Y . Bengio, “Graph attention networks,” inICLR, 2018
2018
-
[15]
Inductive representation learning on large graphs,
W. Hamilton, Z. Ying, and J. Leskovec, “Inductive representation learning on large graphs,” inNeurIPS, 2017, pp. 1025–1035
2017
-
[16]
How powerful are graph neural networks?
K. Xu, W. Hu, J. Leskovec, and S. Jegelka, “How powerful are graph neural networks?” inICLR, 2018
2018
-
[17]
Position-aware graph neural net- works,
J. You, R. Ying, and J. Leskovec, “Position-aware graph neural net- works,” inICML, 2019, pp. 7134–7143
2019
-
[18]
Graph convolutional neural networks for web-scale rec- ommender systems,
R. Ying, R. He, K. Chen, P. Eksombatchai, W. L. Hamilton, and J. Leskovec, “Graph convolutional neural networks for web-scale rec- ommender systems,” inACM KDD, 2018, pp. 974–983
2018
-
[19]
Modeling relational data with graph convolutional networks,
M. Schlichtkrull, T. N. Kipf, P. Bloem, R. Van Den Berg, I. Titov, and M. Welling, “Modeling relational data with graph convolutional networks,” inExtended Semantic Web Conference, 2018, pp. 593–607
2018
-
[20]
Fast graph representation learning with PyTorch Geometric,
M. Fey and J. E. Lenssen, “Fast graph representation learning with PyTorch Geometric,” inICLR Workshop on Representation Learning on Graphs and Manifolds, 2019
2019
-
[21]
DistDGL: Distributed graph neural network training for billion-scale graphs,
D. Zheng, C. Ma, M. Wang, J. Zhou, Q. Su, X. Song, Q. Gan, Z. Zhang, and G. Karypis, “DistDGL: Distributed graph neural network training for billion-scale graphs,” inIEEE/ACM IA3, 2020, pp. 36–44
2020
-
[22]
Distributed hybrid CPU and GPU training for graph neural networks on billion-scale heterogeneous graphs,
D. Zheng, X. Song, C. Yang, D. LaSalle, and G. Karypis, “Distributed hybrid CPU and GPU training for graph neural networks on billion-scale heterogeneous graphs,” inACM KDD, 2022, pp. 4582–4591
2022
-
[23]
DistGNN: Scalable distributed training for large-scale graph neural networks,
V . Md, S. Misra, G. Ma, R. Mohanty, E. Georganas, A. Heinecke, D. Kalamkar, N. K. Ahmed, and S. Avancha, “DistGNN: Scalable distributed training for large-scale graph neural networks,” inACM/IEEE SC, 2021, pp. 1–14
2021
-
[24]
BGL: GPU-Efficient GNN training by optimizing graph data I/O and preprocessing,
T. Liu, Y . Chen, D. Li, C. Wu, Y . Zhu, J. He, Y . Peng, H. Chen, H. Chen, and C. Guo, “BGL: GPU-Efficient GNN training by optimizing graph data I/O and preprocessing,” inUSENIX NSDI, 2023, pp. 103–118
2023
-
[25]
Adaptive message quantization and parallelization for distributed full-graph GNN training,
B. Wan, J. Zhao, and C. Wu, “Adaptive message quantization and parallelization for distributed full-graph GNN training,”MLSys, 2023
2023
-
[26]
Simplifying distributed neural network training on massive graphs: Randomized partitions improve model aggregation,
J. Zhu, A. A. Reganti, E. Huang, C. Dickens, N. Rao, K. Subbian, and D. Koutra, “Simplifying distributed neural network training on massive graphs: Randomized partitions improve model aggregation,” inICML Workshop on Localized Learning, 2023
2023
-
[27]
Multilevel k-way partitioning scheme for irregular graphs,
G. Karypis and V . Kumar, “Multilevel k-way partitioning scheme for irregular graphs,” inJ. Parallel Distrib. Comput., 1998, pp. 96–129
1998
-
[28]
DeepWalk: Online learning of social representations,
B. Perozzi, R. Al-Rfou, and S. Skiena, “DeepWalk: Online learning of social representations,” inACM KDD, 2014, pp. 701–710
2014
-
[29]
node2vec: Scalable feature learning for networks,
A. Grover and J. Leskovec, “node2vec: Scalable feature learning for networks,” inACM KDD, 2016, pp. 855–864
2016
-
[30]
Does negative sampling matter? a review with insights into its theory and applications,
Z. Yang, M. Ding, T. Huang, Y . Cen, J. Song, B. Xu, Y . Dong, and J. Tang, “Does negative sampling matter? a review with insights into its theory and applications,”IEEE Trans. Pattern Anal. Mach. Intell., 2024
2024
-
[31]
Under- standing negative sampling in graph representation learning,
Z. Yang, M. Ding, C. Zhou, H. Yang, J. Zhou, and J. Tang, “Under- standing negative sampling in graph representation learning,” inACM KDD, 2020, pp. 1666–1676
2020
-
[32]
Learn locally, correct globally: A distributed algorithm for training graph neural networks,
M. Ramezani, W. Cong, M. Mahdavi, M. Kandemir, and A. Sivasub- ramaniam, “Learn locally, correct globally: A distributed algorithm for training graph neural networks,” inICLR, 2022
2022
-
[33]
Demystifying graph sparsification algorithms in graph properties preservation,
Y . Chen, H. Ye, S. Vedula, A. Bronstein, R. Dreslinski, T. Mudge, and N. Talati, “Demystifying graph sparsification algorithms in graph properties preservation,”Proceedings of the VLDB Endowment, pp. 427– 440, 2023
2023
-
[34]
Graph sparsification by effective resistances,
D. A. Spielman and N. Srivastava, “Graph sparsification by effective resistances,” inACM STOC, 2008, pp. 563–568
2008
-
[35]
LightNE: A lightweight graph processing system for network embedding,
J. Qiu, L. Dhulipala, J. Tang, R. Peng, and C. Wang, “LightNE: A lightweight graph processing system for network embedding,” inACM SIGMOD, 2021, pp. 2281–2289
2021
-
[36]
DSpar: An embarrassingly simple strategy for efficient GNN training and inference via degree-based sparsification,
Z. Liu, K. Zhou, Z. Jiang, L. Li, R. Chen, S.-H. Choi, and X. Hu, “DSpar: An embarrassingly simple strategy for efficient GNN training and inference via degree-based sparsification,”Transactions on Machine Learning Research, 2023
2023
-
[37]
Sur- vey on graph neural network acceleration: An algorithmic perspective,
X. Liu, M. Yan, L. Deng, G. Li, X. Ye, D. Fan, S. Pan, and Y . Xie, “Sur- vey on graph neural network acceleration: An algorithmic perspective,” inIJCAI, 2022, pp. 5521–5529
2022
-
[38]
Open Graph Benchmark: Datasets for machine learning on graphs,
W. Hu, M. Fey, M. Zitnik, Y . Dong, H. Ren, B. Liu, M. Catasta, and J. Leskovec, “Open Graph Benchmark: Datasets for machine learning on graphs,” inNeurIPS, 2020, pp. 22 118–22 133
2020
-
[39]
Random walks on graphs: A survey,
L. Lov ´asz, “Random walks on graphs: A survey,”Combinatorics, Paul Erd˝os Is Eighty, pp. 1–46, 1993
1993
-
[40]
On the convergence of FedAvg on Non-IID data,
X. Li, K. Huang, W. Yang, S. Wang, and Z. Zhang, “On the convergence of FedAvg on Non-IID data,” inICLR, 2019
2019
-
[41]
PyTorch: An imperative style, high-performance deep learning library,
A. Paszke, S. Gross, F. Massa, A. Lerer, J. Bradbury, G. Chanan, T. Killeen, Z. Lin, N. Gimelshein, L. Antigaet al., “PyTorch: An imperative style, high-performance deep learning library,” inNeurIPS, 2019, pp. 8024–8035
2019
-
[42]
Deep Graph Library: A graph-centric, highly-performant package for graph neural networks,
M. Wang, D. Zheng, Z. Ye, Q. Gan, M. Li, X. Song, J. Zhou, C. Ma, L. Yu, Y . Gai, T. Xiao, T. He, G. Karypis, J. Li, and Z. Zhang, “Deep Graph Library: A graph-centric, highly-performant package for graph neural networks,”arXiv preprint arXiv:1909.01315, 2019
1909 arXiv
-
[43]
PyTorch distributed: Experi- ences on accelerating data parallel draining,
S. Li, Y . Zhao, R. Varma, O. Salpekar, P. Noordhuis, T. Li, A. Paszke, J. Smith, B. Vaughan, P. Damaniaet al., “PyTorch distributed: Experi- ences on accelerating data parallel draining,”Proceedings of the VLDB Endowment, pp. 3005–3018, 2020
2020
-
[44]
How attentive are graph attention networks?
S. Brody, U. Alon, and E. Yahav, “How attentive are graph attention networks?” inICLR, 2022
2022
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.