DyGNN: Algorithm and Architecture Support of Dynamic Pruning for Graph Neural Networks
Hosted in Virtual Platform
SoC, Heterogeneous, and Reconfigurable Architectures
DescriptionRecently, graph neural networks (GNNs) have achieved great success for graph representation learning tasks. Enlightened by the fact that numerous message passing redundancies exist in GNNs, we propose DyGNN, which speeds up GNNs by reducing redundancies. DyGNN is supported by an algorithm and architecture co-design. The proposed algorithm can dynamically prune vertices and edges during execution without accuracy loss. An architecture is designed to support dynamic pruning and transform it into performance improvement. DyGNN opens new directions for accelerating GNNs by pruning vertices and edges. DyGNN gains average 2× speedup with accuracy improvement of 4% compared with state-of-the-art GNN accelerators.