Method and apparatus for compressing topology recognition model, electronic device, and medium

Abstract

Provided are a method and apparatus for compressing a topology recognition model, an electronic device, and a medium. The method includes: pruning a to-be-compressed model to obtain a pruned model, and training the pruned model to obtain a trained pruned model; quantifying a bit quantity of a weight parameter of each neural network layer in the trained pruned model from a first bit quantity to a second bit quantity to obtain a highly-quantified pruned model, and training the highly-quantified pruned model to obtain a trained highly-quantified pruned model; using the trained highly-quantified pruned model as the to-be-compressed model, continuously pruning and quantifying the to-be-compressed model until a compressed topology recognition model is obtained, and deploying the compressed topology recognition model on an Internet of Things in Power Systems (IOTIPS) terminal, to reduce a size of the topology recognition model.