买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：上海交通大学

摘要：本发明涉及一种基于Crossbar架构的剪枝方法及系统，该方法包括：对神经网络当前层待处理的权值矩阵进行不考虑架构因素的结构化剪枝，得到第一权值矩阵；权值矩阵为神经网络的权值矩阵；根据第一权值矩阵的尺寸与crossbar的尺寸，确定剪枝方案；剪枝方案包括：基于区块的结构化剪枝、交叉阵列行剪枝和交叉阵列列剪枝；根据剪枝方案对第一权值矩阵进行剪枝，得到第二权值矩阵；第二权值矩阵用于映射至加速器中的crossbar阵列，通过加速器对神经网络进行加速。本发明可以降低Crossbar的资源占用，降低硬件资源的的浪费。

主权项：1.一种基于Crossbar架构的剪枝方法，其特征在于，包括：对神经网络当前层待处理的权值矩阵进行不考虑架构因素的结构化剪枝，得到第一权值矩阵；所述权值矩阵为神经网络的权值矩阵；根据所述第一权值矩阵的尺寸与crossbar的尺寸，确定剪枝方案；所述剪枝方案包括：基于区块的结构化剪枝、交叉阵列行剪枝和交叉阵列列剪枝；根据所述剪枝方案对所述第一权值矩阵进行剪枝，得到第二权值矩阵，具体包括：确定所述第一权值矩阵映射至加速器中的crossbar阵列时每个crossbar对应的权值分布矩阵；当剪枝方案为基于区块的结构化剪枝时，将所述第一权值矩阵均匀划分为x*y个区块；所述区块的尺寸大于所述crossbar的尺寸，所述加速器中的crossbar阵列中被映射crossbar的个数为x*y；按照所述crossbar的尺寸，对每个所述区块进行结构化剪枝，得到所述第二权值矩阵；当剪枝方案为交叉阵列列剪枝时，确定每个crossbar对应的权值分布矩阵中每一列权值的列重要性度量因子；将列重要性度量因子小于列阈值的列权值删除，并将所述权值分布矩阵中被删除列右侧的所有列左移至删除列所在的空白位置，得到所述第二权值矩阵；所述列阈值可调节；当剪枝方案为交叉阵列行剪枝时，确定每个crossbar对应的权值分布矩阵中每一行权值的行重要性度量因子；将行重要性度量因子小于行阈值的行权值删除，并将所述权值分布矩阵中被删除行下方的所有行上移至删除行所在的空白位置，得到所述第二权值矩阵；所述行阈值可调节；所述第二权值矩阵用于映射至加速器中的crossbar阵列，通过所述加速器对所述神经网络进行加速。

全文数据：

权利要求：

百度查询： 上海交通大学 一种基于Crossbar架构的剪枝方法及系统