Lightweight image super-resolution with the adaptive weight learning network

被引:0
|
作者
Zhang Y. [1 ]
Cheng P. [1 ]
Zhang S. [1 ]
Wang X. [2 ]
机构
[1] School of Electro-Mechanical Engineering, Xidian University, Xi'an
[2] School of Electronic Engineering, Xidian University, Xi'an
关键词
Adaptive weight; Convolutional neural networks; Deep learning; Lightweight; Super-resolution;
D O I
10.19665/j.issn1001-2400.2021.05.003
中图分类号
学科分类号
摘要
In recent years, the single-image super-resolution (SISR) method using deep convolutional neural networks (CNN) has achieved remarkable results. The Pixel Attention Network(PAN) is one of the most advanced lightweight super-resolution methods, which can lead to a good reconstruction performance with a very small number of parameters. But the PAN is limited by the parameters of each module, resulting in slow model training and strict training conditions. To address these problems, this paper proposes a Lightweight Adaptive Weight learning Network (LAWN) for image super-resolution. The network uses multiple adaptive weight modules to form a non-linear mapping network, with each module extracting different levels of feature information. In each adaptive weight module, the network employs the attention branch and the non-attention branch to extract the corresponding information, and then the adaptive weight fusion branch is employed to integrate these two branches. Splitting and fusing the two branches with a specific convolutional layer greatly reduces the number of parameters of the attention branch and the non-attention branch, which helps the network to achieve a relative balance between the number of parameters and the performance. The quantitative evaluations on benchmark datasets demonstrate that the proposed LAWN reduces the number of model parameters and performs favorably against state-of-the-art methods in terms of both PSNR and SSIM. Experimental results show that this method can reconstruct more accurate texture details. The qualitative evaluations with better visual effects prove the effectiveness of the proposed method. © 2021, The Editorial Board of Journal of Xidian University. All right reserved.
引用
收藏
页码:15 / 22
页数:7
相关论文
共 37 条
  • [21] DAI T, CAI J, ZHANG Y, Et al., Second-Order Attention Network for SingleImage Su-per-Resolution, Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11065-11074, (2019)
  • [22] ZHANG Y, LI K, LI K, Et al., Residual Non-local Attention Networks for Image Restoration
  • [23] WANG X, GIRSHICK R, GUPTA A, Et al., Non-Local Neural Networks, Proceedings of the IEEE Confe-rence on Computer Vision and Pattern Recognition, pp. 7794-7803, (2018)
  • [24] NIU B, WEN W, REN W, Et al., Single Image Super-Resolution via a Holistic Attention Network, European Conference on Computer Vision, pp. 191-207, (2020)
  • [25] AHN N, KANG B, SOHN K A., Fast, Accurate, and Lightweight Super-Resolution with Cascading Residual Network, Proceedings of the European Conference on Computer Vision, pp. 252-268, (2018)
  • [26] HUI Z, WANG X, GAOX, Fast and Accurate Single Image Super-Resolution via Information Distillation Network, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 723-731, (2018)
  • [27] HUI Z, GAO X, YANG Y, Et al., Lightweight Image Super-Resolution with Information Multi-Distillation Network, Proceedings of the 27th ACM International Conference on Multimedia, pp. 2024-2032, (2019)
  • [28] TAI Y, YANG J, LIUX, Et al., Memnet: A Persistent Memory Network for Image Restoration, Proceedings of the IEEE International Conference on Computer Vision, pp. 4539-4547, (2017)
  • [29] ZHAO H, KONG X, HE J, Et al., Efficient Image Super-Resolution Using Pixel Attention, European Con-ference on Computer Vision, pp. 56-72, (2020)
  • [30] CHEN H, GU J, ZHANG Z., Attention in Attention Network for Image Super-Resolution