公开(公告)号：CN118798292A

公开(公告)日：2024-10-18

申请号：CN202311213764.4

申请日：2023-09-19

申请人： 中国移动通信集团黑龙江有限公司 ,  中国移动通信集团有限公司

发明人： 白洋 ,  孙忠贺 ,  武星宇 ,  王亮只 ,  李刚 ,  齐兵 ,  姜宝彦 ,  王兰 ,  王志民 ,  张士君

IPC分类号： G06N3/082 ,  G06N3/0495 ,  G06N3/0464

摘要： 本申请涉及人工智能技术领域，提供了一种神经网络滤波器剪枝方法、装置、电子设备和存储介质，方法包括获取训练数据和预先训练完成的神经网络模型；从神经网络模型的所有网络层中确定待剪枝的目标网络层；基于滤波器幅值和滤波器相似度中的至少一个指标以及训练数据，确定目标网络层中待剪枝的目标滤波器；对每一目标网络层中的目标滤波器进行剪枝处理，完成神经网络滤波器剪枝。本申请实施例通过滤波器幅值和滤波器相似度中的至少一个指标对每一目标网络层中的滤波器进行筛选，确定待剪枝的目标滤波器，使得每一网络层中待剪枝的滤波器不同，剪枝率不是固定的，提高了神经网络的精度。