MRDN: A lightweight Multi-stage residual distillation network for image Super-Resolution

Xin Yang,Yingqing Guo,Zhiqiang Li,Dake Zhou,Tao Li
DOI: https://doi.org/10.1016/j.eswa.2022.117594
IF: 8.5
2022-10-15
Expert Systems with Applications
Abstract:For the deep learning based super-resolution (SR) reconstruction method, researchers try to expand the receptive field to improve the reconstruction quality. With the increase of network depth, the consumption of computing resources is impressive. Therefore, SR technology is challenging to apply to small mobile clients. To deal with this defect, we propose a lightweight multi-stage residual distillation network (MRDN) for the SR task in this paper. The model has made the following two main improvements: First, we design a multi-stage residual distillation block (MRDB). It combines the channel separation and the skip connection to reduce the parameter number and guarantee the network’s performance by residual learning. Secondly, we propose efficient pixel attention (EPA) module, which weights different channels according to their importance so that the network can learn more details of pictures. Experiments show that the proposed MRDN is superior to the state-of-the-art models in subjective visual effects and objective evaluation criteria such as PSNR, SSIM, and computational complexity. Taking the basic data set manga109 as an example, the PSNR of this algorithm on scale ×4 reaches 30.66, and the average inference time is 0.021 s. The performance surpasses RFDN, the champion of AIM 2020 Challenge on Efficient Super-Resolution.
computer science, artificial intelligence,engineering, electrical & electronic,operations research & management science
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: **How to design a lightweight multi - stage residual distillation network (MRDN) to significantly reduce the consumption of computational resources while ensuring the quality of image super - resolution reconstruction, so that it can be applied to devices with limited computing power such as mobile clients?** Specifically, the paper addresses the following problems in the image super - resolution (SR) task in deep learning: 1. **High consumption of computational resources**: As the network depth increases, traditional SR models require a large amount of computational resources, which limits their application on small clients such as mobile devices. 2. **High model complexity**: Many high - performance SR models (such as RCAN, EDSR, etc.) have a large number of parameters and complex structures, resulting in long training and inference times. 3. **Balancing reconstruction quality and model complexity**: How to reduce the complexity and computational cost of the model while maintaining high - quality image reconstruction effects. To solve these problems, the paper proposes a lightweight multi - stage residual distillation network (MRDN), which achieves the goal mainly through the following innovations: - Design a multi - stage residual distillation block (MRDB), combine channel separation and skip connections to reduce the number of parameters and ensure network performance. - Propose an efficient pixel - attention module (EPA), weight pixels according to their importance, highlight important features, and maintain a low computational overhead at the same time. The experimental results show that the proposed MRDN model is superior to the existing lightweight SR models in terms of subjective visual effects and objective evaluation metrics (such as PSNR, SSIM, and computational complexity), and achieves a PSNR value of 30.66 at a ×4 scale on the manga109 dataset, with an average inference time of 0.021 seconds, outperforming the champion model RFDN of the AIM 2020 Efficient Super - Resolution Challenge.