APP下载

一种支持卷积神经网络稀疏计算的方法

2021-04-11中兴通讯股份有限公司余金清贺小龙

电子世界 2021年6期
关键词:分类

中兴通讯股份有限公司 余金清 章 伟 贺小龙

本文提出一种对神经网络权重进行编排方法,根据量化的权重值的稀疏特性,通过只编排权重非0值,从而减少权重带宽;同时,提出一种硬件实现架构,结合编排好的权重,能够有效减少卷积网络的乘法运算,特别是在网络权重稀疏情况下,可以进一步减少卷积计算,从而提升神经网络处理能力。

1 背景分析

近些年来,人工智能技术在在全世界范围内得到了迅猛的发展,与此同时神经网络的规模不断增长,典型的神经网络动辄几十层甚至上百层,计算量不断的增加造成推理需要的时间也需来越长。

但实际应用中,在如此高的运算量中并不是所有的计算都是有效计算,特别是对于算法经过定点化或是稀疏化后,出现的权重为零的点,实际运行中,是不需要计算的。因此,神经网络的硬件加速器利用权重的稀疏特性减少计算量,且在此基础上进一步降低处理时间,提高效能,目前业界都在关注的重点技术问题。

2 设计思路

本设计采用两种方法来提升神经网络设计:

(1)权重编排:通过离线方式,对提前训练好的权重( weights)进行特别编排,去掉值为0的权重,完成对非0权重的编排。

(2)计算加速:卷积网络加速器,完成神经网络稀疏计算(只计算非0权重)提升网络处理能力。

2.1 权重编排

权重编排,通过编译器对每个积核中点化后的权重值重新编排,编排后的权重变为指示符表+映射表的形式。如图1所示……

登录APP查看全文

猜你喜欢

分类
分类算一算
垃圾分类的困惑你有吗
星星的分类
我给资源分分类
垃圾分类,你准备好了吗
按需分类
教你一招:数的分类
说说分类那些事
给塑料分分类吧