Abstract:As a new algorithm of graph embedding, graph neural networks (GNNs) have been widely used in many fields. However, GNN computing has the characteristics of both sparse graph processing and dense neural network, which make it difficult to be deployed efficiently on the existing graph processing accelerators or neural network accelerators. Recently, some GNN accelerators have been proposed, but the following challenges have not been fully solved: 1) the minibatch GNN inference scenario has the potential of software and hardware co-design, which can bring 30% computation amount reduction, and this is not well utilized. Besides, the cost of message flow graph construction is large and may account for more than 50% of the total delay; 2) the feature aggregation has a large amount of data access and relatively small amount of computation, which leads to low on-chip data reuse, only 10% of dense computing; and 3) without the optimization of sparse computing units, simple memory bank and cross bar architecture can easily lead to bank access conflict and load imbalance, reducing the utilization of computing units to less than 60%. In order to solve the above problems, we propose a algorithm-hardware co-design scheme to accelerate GNN inference, which includes three technologies: 1) a reuse-aware sampling method is proposed for minibatch inference scenarios, which reduces 30% of the calculation and improves the on-chip reusability of local data; 2) through the nodewise parallelism-aware quantization, the features and weights are quantized to integers with eight or four bits, which reduces the amount of memory access by at least four times; and 3) an accelerator supporting the above technologies is designed and evaluated, and different operations are supported by the sampling-inference integration architecture. The multibank on-chip memory pool is designed to support data reuse, and edge stream reordering is used to reduce data access conflicts, improving the utilization of computing units by 1.5x . Combined with the above technologies, the experiments show that our design achieves 9.2x speedup and 29x energy efficiency improvement compared with the Deep Graph Library framework running on servers equipped with CPU and GPU.

SDMA: an Efficient and Flexible Sparse-Dense Matrix-Multiplication Architecture for GNNs.

Algorithm/Hardware Co-Optimization for Sparsity-Aware SpMM Acceleration of GNNs

DaDianNao: A Machine-Learning Supercomputer

G-NMP: Accelerating Graph Neural Networks with DIMM-based Near-Memory Processing

GE-SpMM: General-Purpose Sparse Matrix-Matrix Multiplication on GPUs for Graph Neural Networks

MEGA: A Memory-Efficient GNN Accelerator Exploiting Degree-Aware Mixed-Precision Quantization

Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow.

Accelerating SpMM Kernel with Cache-First Edge Sampling for Graph Neural Networks

Efficient N:M Sparse DNN Training Using Algorithm, Architecture, and Dataflow Co-Design

Efficient Sparse Matrix Kernels based on Adaptive Workload-Balancing and Parallel-Reduction

A Fine-Grained Sparse Accelerator for Multi-Precision DNN.

Optimizing sparse general matrix–matrix multiplication for DCUs

Fast Sparse Deep Neural Network Inference with Flexible SpMM Optimization Space Exploration

VersaGNN: a Versatile accelerator for Graph neural networks

CoGNN: An Algorithm-Hardware Co-Design Approach to Accelerate GNN Inference With Minibatch Sampling

Structured Weight Matrices-Based Hardware Accelerators in Deep Neural Networks: FPGAs and ASICs

SDP: Co-Designing Algorithm, Dataflow, and Architecture for In-SRAM Sparse NN Acceleration

Accel-GCN: High-Performance GPU Accelerator Design for Graph Convolution Networks

NEM-GNN - DAC/ADC-less, scalable, reconfigurable, graph and sparsity-aware near-memory accelerator for graph neural networks

Optimizing Sparse Matrix Multiplications for Graph Neural Networks

GNNear: Accelerating Full-Batch Training of Graph Neural Networks with Near-Memory Processing