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

被引:9
作者
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
相关论文
共 46 条
[1]   Fast, Accurate, and Lightweight Super-Resolution with Cascading Residual Network [J].
Ahn, Namhyuk ;
Kang, Byungkon ;
Sohn, Kyung-Ah .
COMPUTER VISION - ECCV 2018, PT X, 2018, 11214 :256-272
[2]   Low-Complexity Single-Image Super-Resolution based on Nonnegative Neighbor Embedding [J].
Bevilacqua, Marco ;
Roumy, Aline ;
Guillemot, Christine ;
Morel, Marie-Line Alberi .
PROCEEDINGS OF THE BRITISH MACHINE VISION CONFERENCE 2012, 2012,
[3]  
Chu Xiangxiang, 2019, ARXIV190107261
[4]   Accelerating the Super-Resolution Convolutional Neural Network [J].
Dong, Chao ;
Loy, Chen Change ;
Tang, Xiaoou .
COMPUTER VISION - ECCV 2016, PT II, 2016, 9906 :391-407
[5]   Learning a Deep Convolutional Network for Image Super-Resolution [J].
Dong, Chao ;
Loy, Chen Change ;
He, Kaiming ;
Tang, Xiaoou .
COMPUTER VISION - ECCV 2014, PT IV, 2014, 8692 :184-199
[6]   Hierarchical Neural Architecture Search for Single Image Super-Resolution [J].
Guo, Yong ;
Luo, Yongsheng ;
He, Zhenhao ;
Huang, Jin ;
Chen, Jian .
IEEE SIGNAL PROCESSING LETTERS, 2020, 27 :1255-1259
[7]  
Huang JB, 2015, PROC CVPR IEEE, P5197, DOI 10.1109/CVPR.2015.7299156
[8]   Fast and Accurate Single Image Super-Resolution via Information Distillation Network [J].
Hui, Zheng ;
Wang, Xiumei ;
Gao, Xinbo .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :723-731
[9]   Real-World Super-Resolution via Kernel Estimation and Noise Injection [J].
Ji, Xiaozhong ;
Cao, Yun ;
Tai, Ying ;
Wang, Chengjie ;
Li, Jilin ;
Huang, Feiyue .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :1914-1923
[10]   Hierarchical dense recursive network for image super-resolution [J].
Jiang, Kui ;
Wang, Zhongyuan ;
Yi, Peng ;
Jiang, Junjun .
PATTERN RECOGNITION, 2020, 107