A FPGA-HBM-Based Hardware Streaming Accelerator for GNN Sampling

Yuchen Gui,Qizhe Wu,Wei Yuan,Huawen Liang,Xiaotian Wang,Xi Jin
DOI: https://doi.org/10.1109/asap61560.2024.00026
2024-01-01
Abstract:Sampling takes a long time during GNN training and inference, especially in large-scale graph datasets, so accelerating the sampling process is of great value. Present GNN samplers are mainly focused on the CPU or GPU side, with fewer sampling accelerator implementations on hardware platforms such as FPGAs, and most of the existing accelerators focus on the entire GNN computation process rather than on sampling. Sampling GNNs on the FPGA side is difficult because of the intensive random memory access. In this work, we propose a FPGA-HBM-based streaming sampler working in the time domain to accelerate traditional FPGA sampling, which realizes GNN sampling while reading data in bus bursts and can satisfy both playback and non-playback sampling modes. In addition, the fast loading of node feature vectors is also realized by combining the features of HBM. The proposed hardware achieves a speedup of 2x to 20x relative to traditional FPGA-based node index sampling and 5x to 18x relative to feature vector loading at the CPU side.
What problem does this paper attempt to address?