一种神经网络加速方法、装置、设备及存储介质

基本信息

申请号 CN202110989298.3 申请日 -
公开(公告)号 CN113688976A 公开(公告)日 2021-11-23
申请公布号 CN113688976A 申请公布日 2021-11-23
分类号 G06N3/04;G06N3/08;G06N3/063 分类 计算;推算;计数;
发明人 孙炜;祝叶华 申请(专利权)人 哲库科技(上海)有限公司
代理机构 北京派特恩知识产权代理有限公司 代理人 吴薇薇;张颖玲
地址 200120 上海市浦东新区自由贸易试验区盛夏路61弄1号8层(房产登记楼层7层)01室
法律状态 -

摘要

摘要 本申请实施例公开了一种神经网络加速方法、装置、设备及存储介质,该方法包括:确定神经网络结构的第一卷积核;其中,第一卷积核中各个输入通道的非零权重数据的个数相同;将第一卷积核中各个输入通道的非零权重数据进行集中存储,得到神经网络结构的第二卷积核;利用第二卷积核与特征图像进行卷积操作,得到目标卷积结果。这样,由于本申请确定的第一卷积核仅需求各个输入通道的非零权重数据的个数相同,增加了算法剪枝的自由度,而且减小了剪枝所带来的效果损失;另外,对于非零权重数据采用集中存储策略,不仅可以提高硬件资源的利用率,而且在保证算法效果的同时,还可以提高算法实现的能效比,减小算法实现的延时。