[发明专利]基于向量处理器的深度神经网络多核加速实现方法有效
申请号: | 201710384961.0 | 申请日: | 2017-05-26 |
公开(公告)号: | CN107301456B | 公开(公告)日: | 2020-05-12 |
发明(设计)人: | 郭阳;张军阳;扈啸;王慧丽;胡敏慧 | 申请(专利权)人: | 中国人民解放军国防科学技术大学 |
主分类号: | G06N3/063 | 分类号: | G06N3/063;G06N3/04;G06F15/80 |
代理公司: | 湖南兆弘专利事务所(普通合伙) 43008 | 代理人: | 周长清;胡君 |
地址: | 410073 *** | 国省代码: | 湖南;43 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 向量 处理器 深度 神经网络 多核 加速 实现 方法 | ||
1.一种基于向量处理器的深度神经网络多核加速实现方法,其特征在于步骤包括:
S1.将待处理数据按一维向量输出,并作为输入值输入至目标深度神经网络;
S2.由向量处理器中各个核依次计算目标深度神经网络中相邻两个隐层的权值矩阵,每次计算时,将输入值广播至各个核内的标量存储体中,同时加载相邻两个隐层的权值矩阵,将加载的所述权值矩阵进行划分后分别传输至各个核内的向量存储体中,启动各个核并行计算后得到多个向量计算结果并作为下一次计算的输入值;
所述步骤S2中每次计算时,具体将加载的所述权值矩阵按列平均划分成多份,每份划分后权值矩阵通过DMA传输至各个核内的向量存储体AM中,以将权值矩阵平均分配给各个计算核;
所述步骤S2中每次计算前,还包括判断当前次所需计算的权值矩阵的状态,若判断到满足预设条件时,直接由指定核执行当前权值矩阵的计算,所述预设条件具体为权值矩阵不能平均划分或权值矩阵规模小于预设值。
2.根据权利要求1所述的基于向量处理器的深度神经网络多核加速实现方法,其特征在于:所述步骤S2中每次计算时,具体通过启动DMA的广播传输模式,将输入值广播至各个核内的标量存储体中。
3.根据权利要求2所述的基于向量处理器的深度神经网络多核加速实现方法,其特征在于:所述步骤S2中每次计算时,将加载的所述权值矩阵进行划分后,具体通过启动各个核DMA点对点传输方式,将划分后权值矩阵分别传输至各个核内的向量存储体中。
4.根据权利要求3所述的基于向量处理器的深度神经网络多核加速实现方法,其特征在于:所述步骤S2中每次计算时,得到多个向量计算结果后,具体通过启动各个核的DMA将得到的多个向量计算结果输出至外部DDR中。
5.根据权利要求1~4中任意一项所述的基于向量处理器的深度神经网络多核加速实现方法,其特征在于:所述步骤S2中每次计算时,得到多个向量计算结果后还包括由各个核对得到的所述向量计算结果进行激活函数处理,得到最终的向量计算结果步骤。
6.根据权利要求5所述的基于向量处理器的深度神经网络多核加速实现方法,其特征在于:所述激活函数为sigmoid函数、ReLU函数、tanh函数中的一种。
7.根据权利要求1~4中任意一项所述的基于向量处理器的深度神经网络多核加速实现方法,其特征在于,所述步骤S1中具体按列或行输出一维的向量数据作为目标深度神经网络的输入值。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中国人民解放军国防科学技术大学,未经中国人民解放军国防科学技术大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201710384961.0/1.html,转载请声明来源钻瓜专利网。