首页/ 南方电网数字企业科技（广东）有限公司/ 专利详情

神经网络压缩方法、装置、电子设备及存储介质

基本信息

申请号 CN202110281889.5 申请日 -
公开（公告）号 CN113128664A 公开（公告）日 2021-07-16
申请公布号 CN113128664A 申请公布日 2021-07-16
分类号 G06N3/04（2006.01）I 分类 计算；推算；计数;
发明人 林嘉鑫;赖蔚蔚;吴广财;郑杰生;郑颖龙;周昉昉;刘佳木;陈颖璇 申请（专利权）人 南方电网数字企业科技（广东）有限公司
代理机构 深圳市华优知识产权代理事务所（普通合伙） 代理人 余薇
地址 510000广东省广州市越秀区东风东路808号509号房
法律状态 -

摘要

摘要 本申请提供了一种神经网络压缩方法、装置、电子设备及存储介质，所述方法包括：获取神经网络中各个网络层的参数，其中，所述神经网络为已训练好的神经网络；根据所述各个网络层的参数，计算所述各个网络层的自适应剪枝阈值；根据所述自适应剪枝阈值，将对应网络层的参数进行剪枝，得到剪枝神经网络；对所述剪枝神经网络进行重训练，得到目标压缩神经网络。通过各个网络层的参数情况，对各个网络层进行参数剪枝，完成对神经网络的结构压缩，使得神经网络结构规模减小，降低神经网络的计算复杂度，利于神经网络部署在实时性要求较高的场景以及利于小型化的部署在嵌入式设备中。