发明公开
- 专利标题: 神经网络剪枝方法、装置、设备及存储介质
-
申请号: CN202110937759.2申请日: 2021-08-16
-
公开(公告)号: CN115936096A公开(公告)日: 2023-04-07
- 发明人: 马云驹 , 赵巍 , 李振齐 , 胡静 , 王飞
- 申请人: 广州视源电子科技股份有限公司 , 广州希科医疗器械科技有限公司 , 广州视源人工智能创新研究院有限公司
- 申请人地址: 广东省广州市黄埔区云埔四路6号; ;
- 专利权人: 广州视源电子科技股份有限公司,广州希科医疗器械科技有限公司,广州视源人工智能创新研究院有限公司
- 当前专利权人: 广州视源电子科技股份有限公司,广州希科医疗器械科技有限公司,广州视源人工智能创新研究院有限公司
- 当前专利权人地址: 广东省广州市黄埔区云埔四路6号; ;
- 代理机构: 北京泽方誉航专利代理事务所
- 代理商 陈照辉
- 主分类号: G06N3/082
- IPC分类号: G06N3/082 ; G06N3/0464
摘要:
本申请实施例公开了一种神经网络剪枝方法、装置、设备及存储介质,其包括:向原始网络模型和复制得到的剪枝网络模型输入同一心电信号,原始网络模型包括至少一个第一特征提取模块,剪枝网络模型包括至少一个第二特征提取模块;利用第二特征提取模块的第四卷积层中一通道内使用的卷积核参数处理第三卷积层输出的第一特征图对应的通道,利用处理后得到的第三特征图和第一特征提取模块的第二卷积层输出的第二特征图确定第二卷积层中对应通道的通道性能;根据通道性能确定第二卷积层中的待剪枝通道;根据待剪枝通道对第一卷积层和第二卷积层进行剪枝。采用上述方法可解决提高神经网络模型的性能时增加结构复杂度引起的推理时间变长的技术问题。