Single Image Super-Resolution via a Holistic Attention Network

被引:568
作者
Niu, Ben [1 ]
Wen, Weilei [2 ,3 ]
Ren, Wenqi [3 ]
Zhang, Xiangde [1 ]
Yang, Lianping [1 ]
Wang, Shuzhen [2 ]
Zhang, Kaihao [4 ]
Cao, Xiaochun [3 ]
Shen, Haifeng [5 ,6 ]
机构
[1] Northeastern Univ, Shenyang, Peoples R China
[2] Xidian Univ, Xian, Peoples R China
[3] Chinese Acad Sci, SKLOIS, IIE, Beijing, Peoples R China
[4] Peng Cheng Lab, Cyberspace Secur Res Ctr, Shenzhen, Peoples R China
[5] Australian Natl Univ, Canberra, ACT, Australia
[6] AI Labs, Didi Chuxing, Beijing, Peoples R China
来源
COMPUTER VISION - ECCV 2020, PT XII | 2020年 / 12357卷
基金
北京市自然科学基金; 国家重点研发计划; 中国国家自然科学基金;
关键词
Super-resolution; Holistic attention; Layer attention; Channel-spatial attention;
D O I
10.1007/978-3-030-58610-2_12
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Informative features play a crucial role in the single image super-resolution task. Channel attention has been demonstrated to be effective for preserving information-rich features in each layer. However, channel attention treats each convolution layer as a separate process that misses the correlation among different layers. To address this problem, we propose a new holistic attention network (HAN), which consists of a layer attention module (LAM) and a channel-spatial attention module (CSAM), to model the holistic interdependencies among layers, channels, and positions. Specifically, the proposed LAM adaptively emphasizes hierarchical features by considering correlations among layers. Meanwhile, CSAM learns the confidence at all the positions of each channel to selectively capture more informative features. Extensive experiments demonstrate that the proposed HAN performs favorably against the state-of-the-art single image super-resolution approaches.
引用
收藏
页码:191 / 207
页数:17
相关论文
共 41 条
[1]   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,
[2]   Second-order Attention Network for Single Image Super-Resolution [J].
Dai, Tao ;
Cai, Jianrui ;
Zhang, Yongbing ;
Xia, Shu-Tao ;
Zhang, Lei .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :11057-11066
[3]   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
[4]   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
[5]  
Goodfellow IJ, 2014, ADV NEUR IN, V27, P2672
[6]   Deep Back-Projection Networks For Super-Resolution [J].
Haris, Muhammad ;
Shakhnarovich, Greg ;
Ukita, Norimichi .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1664-1673
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
[9]   Channel-Wise and Spatial Feature Modulation Network for Single Image Super-Resolution [J].
Hu, Yanting ;
Li, Jie ;
Huang, Yuanfei ;
Gao, Xinbo .
IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2020, 30 (11) :3911-3927
[10]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269