Pre-propagation GNNs, whose training is bottlenecked by data loading, can be sped up roughly 15x with custom loaders, GPU double buffering, chunk reshuffling, and direct storage access, beating sampling-based GNNs by up to two orders of magnitude in throughput at comparable accuracy.
DSP: Efficient GNN Training with Multiple GPUs
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
cs.LG 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Graph Learning at Scale: Characterizing and Optimizing Pre-Propagation GNNs
Pre-propagation GNNs, whose training is bottlenecked by data loading, can be sped up roughly 15x with custom loaders, GPU double buffering, chunk reshuffling, and direct storage access, beating sampling-based GNNs by up to two orders of magnitude in throughput at comparable accuracy.