Computer Engineering & Science ›› 2023, Vol. 45 ›› Issue (12): 2113-2120.
• High Performance Computing • Previous Articles Next Articles
ZHOU Li,ZHAO Zhi-qiao,PAN Guo-teng,TIE Jun-bo,ZHAO Wang
Received:
Revised:
Accepted:
Online:
Published:
Abstract: Graph Convolutional Networks (GCN), an algorithm for processing non-Euclidean data, is currently mainly implemented on deep learning frameworks such as PyTorch for GPU acceleration. GCN's computation process involves nested matrix multiplication and data access operations, which can be satisfied by GPU in real-time but have high deployment costs and low energy efficiency. To improve the computational performance of GCN algorithm while maintaining software flexibility, this paper proposes a custom GCN accelerator based on RSIC-V SoC, which extends the dot product operation and hardware accelerator through hardware-software co-design in the hummingbird E203 SoC platform. The neural network parameter analysis determines the hardware quantization scheme from floating point to 32-bit fixed point. Experimental results show that the proposed accelerator has no accuracy loss and can achieve a maximum speedup of 6.88 times when running GCN algorithm on Cora dataset.
Key words: RISC-V, graph convolutional neural network, hardware accelerator, instruction set
ZHOU Li, ZHAO Zhi-qiao, PAN Guo-teng, TIE Jun-bo, ZHAO Wang. RISC-V based design of graph convolutional neural network accelerator[J]. Computer Engineering & Science, 2023, 45(12): 2113-2120.
0 / / Recommend
Add to citation manager EndNote|Ris|BibTeX
URL: http://joces.nudt.edu.cn/EN/
http://joces.nudt.edu.cn/EN/Y2023/V45/I12/2113