[发明专利]一种基于SOC的数据复用卷积神经网络加速器

权利要求书

1.一种基于SOC的数据复用卷积神经网络加速器，其特征在于包括图像地址生成器、图像缓存区、移位逻辑、权重地址生成器、权重缓存区、偏置地址生成器、偏置缓存区、控制单元和计算单元阵列，控制单元接收外部输入的启动控制信号，之后，根据预设时序，控制偏置地址生成器、权重地址生成器和图像地址生成器产生偏置写控制信号、权重写控制信号和图像写控制信号，将偏置、权重和图像数据分块存入相应的缓冲区，之后，再控制偏置地址生成器、权重地址生成器和图像地址生成器产生相应缓冲区的读写地址，权重缓存区、偏置缓存区分别将相应地址内的权重、偏置数据输出至计算单元阵列；图像缓冲区将相应地址内的图像数据输出至移位逻辑，移位逻辑根据控制单元发送的移位控制信号和层运算序号，将图像数据进行移位处理之后输出至计算单元阵列，计算单元阵列，根据权重数据、偏置数据、图像数据，采用分块运算的方法，对图像数据进行多层卷积、池化以及多层全连接运算；

所述控制单元包括主控制模块、权重控制状态机，偏置控制状态机、图像控制状态机和写控制状态机，其中：

主控制模块，接收外部输入的启动控制信号，之后，根据预设的卷积神经网络加速器各层卷积运算的时间和全连接运算时间，按照预设时序，将卷积、池化和全连接运算过程都分成多个层运算，每个层运算分为多个块运算，在层运算开始之前，向写控制状态机发送写控制启动指令，当前层运算所需数据全部写入相应的缓冲区后，向写控制状态机发出写控制停止信号；在层运算开始时刻，向权重读控制状态机、偏置读控制状态机、图像读控制状态机发送权重读控制启动信号，向移位逻辑发送移位控制信号和层运算序号；在层运算结束时刻和块运算结束时刻，向权重读控制状态机、偏置读控制状态机、图像读控制状态机，发送层运算结束标志和块运算结束标志信号，向移位逻辑发送的移位控制信号和层运算序号；

权重控制状态机、偏置控制状态机、图像控制状态机和写控制状态机，在主控制模块的控制下，分别向权重缓存区、偏置缓存区、图像缓存区输出相应的读使能信号、写使能信号和片选信号，向权重地址生成器、偏置地址生成器、图像地址生成器输出相应的地址控制信号，权重地址生成器、偏置地址生成器、图像地址生成器根据地址控制信号产生相应的读写地址。

2.根据权利要求1所述的一种基于SOC的数据复用卷积神经网络加速器，其特征在于所述图像缓存以及权重缓存均为分组存储结构，图像缓存区和权重缓存区分成M个子缓存区，M个子缓存区的片选控制端、读写使能端并联连接，地址线相互独立，各子缓存区相对应地址用来存储一个一次块运算所需要的图像数据或者权重数据，同时写入或读出，M为一次块运算所对应的最大图像数据大小。

3.根据权利要求1所述的一种基于SOC的数据复用卷积神经网络加速器，其特征在于权重地址生成器、偏置地址生成器包括计数器，计数器的计数值作为地址输出至相应的缓冲区，当地址复位信号“有效”时，计数器的计数值清零；当地址保持信号有效时，计数器的计数值不变；当地址递增信号有效时，计数器的计数值加1。

4.根据权利要求1所述的一种基于SOC的数据复用卷积神经网络加速器，其特征在于图像地址生成器，包括读地址生成模块、写地址生成模块、读写地址选通模块；

写地址生成模块，包括计数器，计数器的计数值作为图像写地址输出至读写地址选通模块，当地址复位信号“有效”时，计数器的计数值清零；当地址保持信号有效时，计数器的计数值不变；当地址递增信号有效时，计数器的计数值加1；

读地址生成模块，包括R个读地址生成子模块，所述R为层数； 每个读地址生成子模块用来控制产生一次层运算中的所有块运算所需要的地址，根据层序号选通相应层的读地址生成子模块，对于某一层的处理，输入为三维图像数据，首先沿图像通道方向递增地址，当图像通道方向数据读取完成，按图像二维平面换列方向递增地址后，继续沿图像通道方向递增地址，当图像二维平面换列方向和图像通道方向数据均读取完成，按图像二维平面换行方向递增地址，继续沿图像通道方向递增地址，直到整块数据处理完成。