Automatic Pruning and Lightweight Deployment of Reinforcement Learning-based Deep Diagnostic Model for Partial Discharge
Various discharge identification models based on local discharge signal sampling data developed on PC com-puter or server side with high positive judgment rate are almost all deep models,which are large in volume and raise higher requirements for computer resources,and cannot be run on the local discharge detection device.For this reason,this paper proposes a reinforcement learning-based automatic pruning method for partial discharge depth diagnostic mod-els with a model lightweight deployment scheme.On the server side,deep reinforcement learning is used for intelligent body training,and automatic search is performed by interacting with the original partial discharge diagnostic model,which in turn determines the pruning rate of each layer;and then the filter pruning according to the filter pruning by geo-metric median(FPGM)method is used to discriminate the importance of the filter and realize parameter pruning.Simulation experimental results show that the method achieves more than 85％parameter compression effect on the lightweight series models MobileNetV1 and V2 as well as the classic series neural networks of ResNet50.The com-pressed lightweight model is converted into lightweight ONNX format,saved on a portable computer,and implanted into a Raspberry Pi smart terminal through wireless transmission,which can realize the diagnostic experimental simulation of partial discharge on the smart terminal.The test results show that the local discharge diagnostic model which is deployed after pruning using this method has great improvements in performance indexes such as memory occupation,power con-sumption,and inference time.
partial dischargereinforcement learning pruningneural networkfilter pruning via geometric medianedge deployment of models
万方数据
维普
