ZHOU Renshuang, CHEN Yaosen, GUO Bing, SHEN Yan, LI Jie, WANG Wei. Module Similarity-Based Pruning for Image Super-Resolution Network[J]. Journal of University of Electronic Science and Technology of China, 2022, 51(1): 108-116. DOI: 10.12178/1001-0548.2021126
Citation: ZHOU Renshuang, CHEN Yaosen, GUO Bing, SHEN Yan, LI Jie, WANG Wei. Module Similarity-Based Pruning for Image Super-Resolution Network[J]. Journal of University of Electronic Science and Technology of China, 2022, 51(1): 108-116. DOI: 10.12178/1001-0548.2021126

Module Similarity-Based Pruning for Image Super-Resolution Network

  • This paper proposes a network pruning method for single image super-resolution network (SISR). This method evaluates the similarity of each module in the super-resolution network and uses a simple method to convert the similarity into the contribution degree of each module to the network, and find the relatively unimportant modules of the network to perform network pruning. Through the method of network pruning for the super-resolution network based on the module similarity, the super-resolution network with a huge amount of parameters is compressed, and the number of parameters and the amount of calculation are greatly reduced. Experiments show that the parameters of the super-resolution network after pruning can be reduced by more than 60%, while the accuracy is not reduced by more than 0.1%, which has great practical significance for the deployment of the super-resolution network to a low-performance platform.
  • loading

Catalog

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return