Graph Neural Networks (GNNs) have been widely adopted due to their strong performance. However, GNN training often relies on expensive, high-performance computing platforms, limiting accessibility for many tasks. Profiling of representative GNN workloads indicates that substantial efficiency gains are possible on resource-constrained devices by fully exploiting available resources. This paper introduces A3GNN, a framework for affordable, adaptive, and automatic GNN training on heterogeneous CPU-GPU platforms. It improves resource usage through locality-aware sampling and fine-grained parallelism scheduling. Moreover, it leverages reinforcement learning to explore the design space and achieve pareto-optimal trade-offs among throughput, memory footprint, and accuracy. Experiments show that A3GNN can bridge the performance gap, allowing seven Nvidia 2080Ti GPUs to outperform two A100 GPUs by up to 1.8X in throughput with minimal accuracy loss.
@article{arxiv.2511.07421,
title = {Towards Affordable, Adaptive and Automatic GNN Training on CPU-GPU Heterogeneous Platforms},
author = {Tong Qiao and Ao Zhou and Yingjie Qi and Yiou Wang and Han Wan and Jianlei Yang and Chunming Hu},
journal= {arXiv preprint arXiv:2511.07421},
year = {2025}
}
Comments
Accepted by The 43rd IEEE International Conference on Computer Design, ICCD'25