公开(公告)号：CN118036696A

公开(公告)日：2024-05-14

申请号：CN202410102719.X

申请日：2024-01-24

Applicant: 北京邮电大学

Inventor： 别红霞 ,  王嘉利 ,  别志松 ,  荆朝 ,  李晋鑫

IPC: G06N3/082 ,  G06N3/0495 ,  G06F18/213 ,  G06F17/15 ,  G06F17/16

Abstract: 本公开提供一种基于特征提取敏感度的网络剪枝方法及相关设备，包括：确定目标网络的目标层的滤波器矩阵的核范数；其中所述目标网络包括若干网络层；对所述滤波器矩阵中的目标滤波器进行遮蔽处理，并确定所述遮蔽处理后的滤波器矩阵的核范数；基于所述滤波器矩阵的核范数以及，所述遮蔽处理后的滤波器矩阵的核范数确定所述目标滤波器的敏感度；基于所述敏感度确定所述目标层的剪枝率；基于所述剪枝率对所述目标层中的滤波器进行剪枝处理。本公开中，首先基于目标网络中目标层的滤波器矩阵的核范数，确定了目标层中的滤波器的敏感度。之后基于敏感度确定了目标层的剪枝率，最后通过剪枝率以及前述的敏感度对目标层中的滤波器进行了剪枝处理。