基于RISC-V的图卷积神经网络加速器设计

作     者：周理 赵祉乔 潘国腾 铁俊波 赵王 ZHOU Li;ZHAO Zhi-qiao;PAN Guo-teng;TIE Jun-bo;ZHAO Wang

作者机构：国防科技大学计算机学院湖南长沙410073 

出 版 物：《计算机工程与科学》 (Computer Engineering & Science)

年 卷 期：2023年第45卷第12期

页      码：2113-2120页

摘      要：图卷积神经网络GCN当前主要在PyTorch等深度学习框架上基于GPU实现加速。然而GCN的运算过程包含多层嵌套的矩阵乘法和数据访存操作,使用GPU虽然可以满足实时性需求,但是部署代价大、能效比低。为了提高GCN算法的计算性能并保持软件灵活性,提出一种基于RSIC-V SoC的定制GCN加速器,在蜂鸟E203的SoC平台中通过点积运算扩展指令和硬件加速器软硬件协同的方法实现了针对GCN的加速,通过神经网络参数分析确定了从浮点数到32位定点数的硬件量化方案。实验结果表明,在Cora数据集上运行GCN算法时,该加速器没有精度损失,速度最高提高了6.88倍。

主 题 词：RISC-V 图卷积神经网络 硬件加速器 指令集 

学科分类：12[管理学] 1201[管理学-管理科学与工程类] 08[工学] 081201[081201] 0812[工学-测绘类] 

D　O　I：10.3969/j.issn.1007-130X.2023.12.003

馆 藏 号：203125223...