NasmamSR: a fast image super-resolution network based on neural architecture search and multiple attention mechanism

被引:7
|
作者
Yang, Xin [1 ]
Fan, Jiangfeng [1 ]
Wu, Chenhuan [1 ]
Zhou, Dake [1 ]
Li, Tao [1 ]
机构
[1] Nanjing Univ Aeronaut & Astronaut, Coll Automat Engn, Nanjing 210016, Peoples R China
基金
中国国家自然科学基金;
关键词
Super-resolution; Neural architecture search; Attention mechanism; Deep learning; CONVOLUTIONAL NETWORK;
D O I
10.1007/s00530-021-00841-2
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Although the current super-resolution model based on deep learning has achieved excellent reconstruction results, the increasing depth of the model results in huge parameters, limiting the further application of the super-resolution deep model. To solve this problem, we propose an efficient super-resolution model based on neural architecture search and attention mechanism. First, we use global residual learning to limit the search to the non-linear mapping part of the network and add a down-sampling to this part to reduce the feature map's size and computation. Second, we establish a lightweight search space and joint rewards for searching the optimal network structure. The model divides the search into macro search and micro search, which are used to search for the optimal down-sampling position and the optimal cell structure, respectively. In addition, we introduce the Bayesian algorithm for hyper-parameter tuning and further improve the model's performance based on the optimal sub-network searched out. Detailed experiments show that our model achieves excellent super-resolution performance and high computational efficiency compared with some state-of-the-art models.
引用
收藏
页码:321 / 334
页数:14
相关论文
共 50 条
  • [1] NasmamSR: a fast image super-resolution network based on neural architecture search and multiple attention mechanism
    Xin Yang
    Jiangfeng Fan
    Chenhuan Wu
    Dake Zhou
    Tao Li
    [J]. Multimedia Systems, 2022, 28 : 321 - 334
  • [2] Image super -resolution based on deep neural network of multiple attention mechanism *
    Yang, Xin
    Li, Xiaochuan
    Li, Zhiqiang
    Zhou, Dake
    [J]. JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2021, 75
  • [3] Attention mechanism feedback network for image super-resolution
    Chen, Xiao
    Jing, Ruyun
    Suna, Chaowen
    [J]. JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (04)
  • [4] Fast, Accurate and Lightweight Super-Resolution with Neural Architecture Search
    Chu, Xiangxiang
    Zhang, Bo
    Ma, Hailong
    Xu, Ruijun
    Li, Qingyuan
    [J]. 2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 59 - 64
  • [5] Hierarchical Neural Architecture Search for Single Image Super-Resolution
    Guo, Yong
    Luo, Yongsheng
    He, Zhenhao
    Huang, Jin
    Chen, Jian
    [J]. IEEE SIGNAL PROCESSING LETTERS, 2020, 27 : 1255 - 1259
  • [6] Super-Resolution Generative Adversarial Network Based on the Dual Dimension Attention Mechanism for Biometric Image Super-Resolution
    Huang, Chi-En
    Li, Yung-Hui
    Aslam, Muhammad Saqlain
    Chang, Ching-Chun
    [J]. SENSORS, 2021, 21 (23)
  • [7] Image super-resolution network based on a multi-branch attention mechanism
    Xin Yang
    Yingqing Guo
    Zhiqiang Li
    Dake Zhou
    [J]. Signal, Image and Video Processing, 2021, 15 : 1397 - 1405
  • [8] Image super-resolution network based on a multi-branch attention mechanism
    Yang, Xin
    Guo, Yingqing
    Li, Zhiqiang
    Zhou, Dake
    [J]. SIGNAL IMAGE AND VIDEO PROCESSING, 2021, 15 (07) : 1397 - 1405
  • [9] A Frequency Domain Neural Network for Fast Image Super-resolution
    Li, Junxuan
    You, Shaodi
    Robles-Kelly, Antonio
    [J]. 2018 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2018,
  • [10] Differentiable Neural Architecture Search for Extremely Lightweight Image Super-Resolution
    Huang, Han
    Shen, Li
    He, Chaoyang
    Dong, Weisheng
    Liu, Wei
    [J]. IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2023, 33 (06) : 2672 - 2682