[发明专利]模型剪枝方法、装置、设备和存储介质

权利要求书

1.一种模型剪枝方法，其特征在于，包括：

获取待剪枝模型中多个卷积层组合，其中，所述待剪枝模型包括多个依次连接的卷积层，每个所述卷积层组合均包括第一卷积层和第二卷积层，所述第二卷积层为所述第一卷积层的下一卷积层；

确定所述卷积层组合中所述第一卷积层对应的滤波器和所述第二卷积层对应的通道，其中，所述第一卷积层中的滤波器与所述第二卷积层中的通道一一对应，所述第一卷积层包括至少一个所述滤波器；

获取所述滤波器的第一矩阵数据和所述通道的第二矩阵数据，根据所述第一矩阵数据和所述第二矩阵数据，计算得到所述滤波器对应的剪枝指标；

根据所述剪枝指标对所述待剪枝模型进行剪枝处理，得到剪枝模型。

2.根据权利要求1所述的模型剪枝方法，其特征在于，所述根据所述第一矩阵数据和所述第二矩阵数据，计算得到所述滤波器对应的剪枝指标，包括：

将所述第一矩阵数据和所述第二矩阵数据进行乘积计算，得到乘积矩阵数据；

对所述乘积矩阵数据中的矩阵元素分别取绝对值后进行累加求和计算，得到所述滤波器对应的剪枝指标。

3.根据权利要求2所述的模型剪枝方法，其特征在于，所述将所述第一矩阵数据和所述第二矩阵数据进行乘积计算，得到乘积矩阵数据，包括：

将所述第一矩阵数据中的第一矩阵元素和所述第二矩阵数据中的第二矩阵元素分别进行对应元素的乘积计算，得到所述乘积矩阵数据。

4.根据权利要求3所述的模型剪枝方法，其特征在于，所述第一矩阵数据中的第一矩阵元素和第二矩阵数据中的第二矩阵元素一一对应。

5.根据权利要求1所述的模型剪枝方法，其特征在于，所述根据所述剪枝指标对所述待剪枝模型进行剪枝处理，得到剪枝模型，包括：

根据所述剪枝指标对所述滤波器进行筛选处理，得到符合预设剪枝率的剪枝滤波器；

根据所述剪枝滤波器，对所述待剪枝模型进行剪枝处理，得到剪枝模型。

6.根据权利要求5所述的模型剪枝方法，其特征在于，所述根据所述剪枝指标对所述滤波器进行筛选处理，得到符合预设剪枝率的剪枝滤波器，包括：

对所述剪枝指标进行排序，得到排序数据；

根据所述预设剪枝率，对所述排序数据对应的所述滤波器进行筛选处理，得到所述剪枝滤波器。

7.根据权利要求5或6所述的模型剪枝方法，其特征在于，在所述得到剪枝模型之后，包括：

对所述剪枝模型进行重新训练，以更新所述剪枝指标。

8.一种模型剪枝装置，其特征在于，包括：

数据获取模块，用于获取待剪枝模型中多个卷积层组合，其中，所述待剪枝模型包括多个依次连接的卷积层，每个所述卷积层组合均包括第一卷积层和第二卷积层，所述第二卷积层为所述第一卷积层的下一卷积层；

数据确定模块，用于确定所述卷积层组合中所述第一卷积层对应的滤波器和所述第二卷积层对应的通道，其中，所述第一卷积层中的滤波器与所述第二卷积层中的通道一一对应，所述第一卷积层包括至少一个所述滤波器；

指标计算模块，用于获取所述滤波器的第一矩阵数据和所述通道的第二矩阵数据，根据所述第一矩阵数据和所述第二矩阵数据，计算得到所述滤波器对应的剪枝指标；

剪枝处理模块，用于根据所述剪枝指标对所述待剪枝模型进行剪枝处理，得到剪枝模型。

9.一种模型剪枝设备，其特征在于，包括：存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如权利要求1至7中任意一项所述的模型剪枝方法。

10.一种计算机可读存储介质，其特征在于，存储有计算机可执行指令，所述计算机可执行指令用于执行权利要求1至7中任意一项所述的模型剪枝方法。