一种基于位掩码的卷积压缩计算方法及系统

基本信息

申请号 CN202110134844.5 申请日 -
公开(公告)号 CN112949851A 公开(公告)日 2021-06-11
申请公布号 CN112949851A 申请公布日 2021-06-11
分类号 G06N3/08;G06F7/544;G06F16/901;G06F17/16 分类 计算;推算;计数;
发明人 张笑;林广栋;黄光红;顾大晔;刘国成 申请(专利权)人 安徽芯纪元科技有限公司
代理机构 合肥律众知识产权代理有限公司 代理人 邓盛花
地址 230000 安徽省合肥市高新区习友路3366号博微产业园系统协同中心7楼
法律状态 -

摘要

摘要 本发明提出一种基于位掩码的卷积压缩计算方法及系统,输入矩阵F大小为h*w,卷积核K大小为k1h*k2w,卷积压缩计算方法包括以下步骤:1、提取输入矩阵F的非0数据形成行矩阵Df,提取输入矩阵F的位掩码形成矩阵Mf,提取卷积核K的非0数据形成行矩阵Dk,提取卷积核K的位掩码形成矩阵Mk;2、将行矩阵Dk中的每个元素与行矩阵Df中的元素逐一相乘,得到多个乘法结果矩阵;3、利用矩阵Mk和矩阵Mf将每个乘法结果矩阵恢复成卷积核K大小的恢复矩阵;4、对每个乘法结果矩阵得到的恢复矩阵进行对位累加,得到卷积计算结果。本发明可以避免卷积运算过程中0值参与乘法计算,降低了算力和功耗的浪费,提高了卷积计算速度。