Slicing Input Features to Accelerate Deep Learning: A Case Study with Graph Neural Networks

Zhengjia Xu, Dingyang Lyu, Jinghui Zhang

As graphs grow larger, full-batch GNN training becomes hard for single GPU memory. Therefore, to enhance the scalability of GNN training, some studies have proposed sampling-based…