CSA-MSO3DCNN: Multiscale Octave 3D CNN with Channel and Spatial Attention for Hyperspectral Image Classification

被引:68
作者
Xu, Qin [1 ]
Xiao, Yong [1 ]
Wang, Dongyue [1 ]
Luo, Bin [1 ]
机构
[1] Anhui Univ, Sch Comp Sci & Technol, Minist Educ, Key Lab Intelligent Comp & Signal Proc, Hefei 230601, Peoples R China
基金
中国国家自然科学基金;
关键词
hypersctral image classification; octave convolution; feature extraction; channel and spatial attention; REPRESENTATION; NETWORK;
D O I
10.3390/rs12010188
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
3D convolutional neural networks (CNNs) have been demonstrated to be a powerful tool in hyperspectral images (HSIs) classification. However, using the conventional 3D CNNs to extract the spectral-spatial feature for HSIs results in too many parameters as HSIs have plenty of spatial redundancy. To address this issue, in this paper, we first design multiscale convolution to extract the contextual feature of different scales for HSIs and then propose to employ the octave 3D CNN which factorizes the mixed feature maps by their frequency to replace the normal 3D CNN in order to reduce the spatial redundancy and enlarge the receptive field. To further explore the discriminative features, a channel attention module and a spatial attention module are adopted to optimize the feature maps and improve the classification performance. The experiments on four hyperspectral image data sets demonstrate that the proposed method outperforms other state-of-the-art deep learning methods.
引用
收藏
页数:24
相关论文
共 51 条
[1]   Spatial Prior Fuzziness Pool-Based Interactive Classification of Hyperspectral Images [J].
Ahmad, Muhammad ;
Khan, Asad ;
Khan, Adil Mehmood ;
Mazzara, Manuel ;
Distefano, Salvatore ;
Sohaib, Ahmed ;
Nibouche, Omar .
REMOTE SENSING, 2019, 11 (09)
[2]  
[Anonymous], 2017, P NEUR INF PROC SYST
[3]  
[Anonymous], GRSS DAT FUS CONT
[4]  
[Anonymous], PROC CVPR IEEE
[5]  
[Anonymous], ADV NEURAL INFORM PR
[6]  
[Anonymous], 2019, IEEE T GEOSCI REMOTE, DOI DOI 10.1109/TGRS.2018.2862899
[7]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[8]  
[Anonymous], P IEEE C COMP VIS PA
[9]  
[Anonymous], 2014, Neural Information Processing Systems
[10]  
Ba J., 2014, Multiple object recognition with visual attention