[发明专利]剪枝模块的确定方法、装置及计算机可读存储介质

权利要求书

1.一种剪枝模块的确定方法，其特征在于，所述剪枝模块的确定方法包括以下步骤：

获取待剪枝神经网络的输入任务和任务数量，以及所述待剪枝神经网络的组成模块的输出信息；

根据所述输入任务、所述任务数量和所述输出信息确定所述组成模块的任务关联度；

根据所述任务数量和所述输出信息确定所述组成模块的输入信息保留度；

根据所述任务关联度和所述输入信息保留度确定所述组成模块的重要性指标值；

根据所述重要性指标值确定所述待剪枝神经网络中的剪枝模块。

2.如权利要求1所述的剪枝模块的确定方法，其特征在于，所述根据所述输入任务、所述任务数量和所述输出信息确定所述组成模块的任务关联度的步骤包括：

获取当前网络层级中所述组成模块的目标序号；

确定当前网络层级中序号小于或等于所述目标序号的目标组成模块；

根据各个所述目标组成模块的输出信息与所述输入任务确定各个所述目标组成模块的输入任务保留度和输入输出信息总和；

根据所述任务数量、所述输入任务保留度和所述输入输出信息总和确定所述组成模块的任务关联度。

3.如权利要求2所述的剪枝模块的确定方法，其特征在于，所述输入任务包括至少一张输入任务图，所述输出信息包括输出特征图，所述根据各个所述目标组成模块的输出信息与所述输入任务确定各个所述目标组成模块的输入任务保留度和输入输出信息总和的步骤包括：

对各个所述目标组成模块的输出特征图的分辨率进行调节，以使所述输出特征图的分辨率与所述输入任务图的分辨率一致；

对所述输入任务图和调节后的所述输出特征图进行二值化处理；

对二值化处理后的所述输入任务图和二值化处理后的所述输出特征图取交集，得到各个所述目标组成模块的输入任务保留度；

对二值化处理后的所述输入任务图和二值化处理后的所述输出特征图的取并集，得到各个所述目标组成模块的输入输出信息总和。

4.如权利要求2所述的剪枝模块的确定方法，其特征在于，所述根据所述任务数量、所述输入任务保留度和所述输入输出信息总和确定所述组成模块的任务关联度的步骤包括：

获取所述输入任务保留度和所述输入输出信息总和的比值，得到各个所述目标组成模块的任务保留占比；

对各个所述任务保留占比求和，得到任务保留占比和值；

将所述任务保留占比和值与所述任务数量的商值确定为所述组成模块的任务关联度。

5.如权利要求1所述的剪枝模块的确定方法，其特征在于，所述根据所述任务数量和所述输出信息确定所述组成模块的输入信息保留度的步骤包括：

获取所述组成模块在当前网络层级中的目标序号，以及所述输入任务包含的图像数量；

确定当前网络层级中序号小于或等于所述目标序号的目标组成模块；

根据各个所述目标组成模块的输出信息确定当前网络层级中各个所述目标组成模块的第一能量值，根据前一网络层级中各个组成模块的输出信息确定前一网络层级中的组成模块的第二能量值；

根据所述任务数量、所述第一能量值和所述第二能量值确定所述组成模块的输入信息保留度。

6.如权利要求5所述的剪枝模块的确定方法，其特征在于，所述根据各个所述目标组成模块的输出信息确定当前网络层级中各个所述目标组成模块的第一能量值，根据前一网络层级中各个组成模块的输出信息确定前一网络层级中的组成模块的第二能量值的步骤包括：

求取各个所述目标组成模块的输出信息的范数，将所述范数的平方作为各个所述目标组成模块的第一能量值；

求取前一网络层级中的各个组成模块的输出信息的方差，得到所述第二能量值。