架空线停电预测神经网络压缩方法及系统
摘要:
本发明公开了一种架空线停电预测神经网络压缩方法及系统,属于配电网大数据技术领域,包括获取架空线停电数据集;利用架空线停电数据集对基模型进行训练,得到停电预测基模型;利用神经网络预设尺寸约束剪枝Mask对停电预测基模型的网络参数进行剪枝处理,得到剪枝后的网络模型;基于预设尺寸约束正则化损失函数,对停电预测基模型的输出、剪枝后的网络模型的输出以及剪枝Mask进行迭代训练,在迭代训练结束后,得到停电预测神经网络压缩模型。本发明可高效地得到最终可实际部署的架空线停电预测神经网络模型,并可以较好的平衡算法性能与模型规模。
公开/授权文献
0/0