集微网 · 2020年11月16日

【专利解密】西井科技 通用AI引擎架构

【嘉德点评】西井科技发明的AI芯片结构,利用同一种卷积引擎硬件结构来适配多种神经网络结构,从而有效的提高了乘法器资源的利用率以及实现了数据的动态分布

集微网消息,西井科技成立于2015年,从研发类脑芯片起步,专注于AI芯片和算法的研发落地。2016年,通过将技术和产品相结合方式服务于智慧港口,利用人工智能技术来完成各种复杂任务,提高港口自动化程度。

而其解决港口客户对码头的智能化升级需求,并服务于集装箱数据智能识别、无人驾驶等项目,重点依靠的技术就是其自主研发的AI芯片。AI芯片主要应用于人工智能领域,其主要使用的算法就是神经网络算法。

卷积神经网络是一种前馈神经网络,它的人工神经元可以响应一部分覆盖范围内的周围单元,对于大型图像处理有出色表现。它主要包括卷积层和池化层,目前,卷积神经网络已广泛应用于图像分类、物体识别、目标追踪。

但是对于卷积神经网络的处理芯片,还存在着一系列的问题需要研究人员们去解决,例如如何通过乘法器和加法器的排布、设计提高乘法器资源的利用率,且根据不同的卷积核的尺寸实现不同的乘法器资源分配,以实现数据的动态分布。

为此,西井科技在2020年6月24日申请了一项名为“芯片结构及其乘加计算引擎”的发明专利(申请号:202010587029.X),申请人为上海西井信息科技有限公司。

根据该专利目前公开的资料,让我们一起来看看这项AI芯片技术吧。

image

由于乘加计算引擎中包括多个卷积引擎,而卷积引擎又包括多个乘法器和加法器,如上图,为卷积引擎应用于尺寸为33的卷积核的示意图。33的卷积大小是目前主流神经网络模型中经常使用的,每个卷积引擎包括15M4个乘法器,每个卷积组11包括3行4列乘法器且对应一个卷积核计算,因此5组卷积组同时可以支持5个不同的卷积核并行计算。

而上图只是一个以33卷积核为例的示例,在神经网络的实际应用中,还有55、7*7等常用卷积核,因此该专利对于其大小并没有进行限制,每个卷积引擎在应用于卷积核时,可以按照卷积核的尺寸划分为多个卷积组,每个卷积组的乘法器的行数与卷积核的行数一致,每个卷积组的乘法器的列数为N,这样就可以根据不同的卷积核的尺寸实现不同的乘法器资源分配,进一步实现数据的动态分布。

image

如上图,为这种卷积引擎的示意图,可以看到其具有多个输入特征图通道,这是为了当输入的特征图具有较高维度时可以进行并行加速。此外,这种卷积引擎还可以进行M倍于输入特征通道的并行加速,每个卷积组输出的N行部分累加值可以叠加到下一层的输入中,这与神经网络算法的设计也是相一致的。

考虑到常用的主流卷积神经网络模型,输入特征通道的数量通常以偶数出现,一般是2的n次方的形式。由此,可以利用16个卷积组,用于支持16个不同的输入特征图的通道进行计算。

图中所展示的16个不同的输入特征图,以一个输入像素点为例,一个输入像素点要进行16个通道的数据乘加,也就是16个乘法和M/2个加法,而在硬件的角度进行实现时,需要设计硬件加速方案,这种硬件加速方案由卷积引擎组成多级联结构,每组级联结构包括M/2个级联的处理单元,如下图所示。

image

如图,为处理单元的示意图,每个处理单元50包括第一输入接口511至第五输入接口515、第一触发器531至第五触发器535、两个乘法器(541和542)以及两个加法器(551和552),此外,该结构还具有输出接口、时钟信号和触发器等部件。

第一输入接口至第四输入接口分别与第一至第四触发器相连,时钟信号也与触发器相连,用于开启触发器,从而控制处理器进行乘加运算。

而在16个输入特征通道的例子中,可以使用8个级联的方式来完成一个像素点在16个通道内的乘加运算,具体的连接方式如下入所示:

image

可以看到级联的处理单元的输出结果为一个输入特征图像素点的16通道乘累加结果,在这个架构下,以3*3的卷积核为例,9个输入像素点的16个输入特征通道值进行相加,在6个时钟周期下生成最终的卷积操作的运算结果。

以上就是西井科技发明的AI芯片结构,这种硬件结构的特点就在于,利用不变的卷积引擎,应用在不同神经网络、不同大小的卷积核中。从而可以有效的提高乘法器资源的利用率,并且根据不同的卷积核的尺寸可以实现不同的乘法器资源分配,也就可以做到数据的动态分布,达到需要的计算方式。

关于嘉德

image

深圳市嘉德知识产权服务有限公司由曾在华为等世界500强企业工作多年的知识产权专家、律师、专利代理人组成,熟悉中欧美知识产权法律理论和实务,在全球知识产权申请、布局、诉讼、许可谈判、交易、运营、标准专利协同创造、专利池建设、展会知识产权、跨境电商知识产权、知识产权海关保护等方面拥有丰富的经验。

(校对/holly)

推荐阅读
关注数
12739
内容数
1029
从专利出发,浅析一切关于柔性屏、折叠屏、10倍光学变焦技术等有趣的前沿技术
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息