A Multiscale Dual-Branch Feature Fusion and Attention Network for Hyperspectral Images Classification

被引:38
作者
Gao, Hongmin [1 ]
Zhang, Yiyan [1 ]
Chen, Zhonghao [1 ]
Li, Chenming [1 ]
机构
[1] Hohai Univ, Coll Comp & Informat, Nanjing 211100, Peoples R China
基金
中国博士后科学基金; 中国国家自然科学基金;
关键词
Feature extraction; Image classification; Hyperspectral imaging; Convolutional neural networks; Training; Kernel; Deep learning; Convolutional neural network (CNN); dual-branch feature fusion (DBFM); hyperspectral image (HSI) classification; multiscale feature extraction (MSFE) module; shuffle attention block; CONVOLUTIONAL NETWORKS; SPARSE; CNN;
D O I
10.1109/JSTARS.2021.3103176
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Recently, hyperspectral image classification based on deep learning has achieved considerable attention. Many convolutional neural network classification methods have emerged and exhibited superior classification performance. However, most methods focus on extracting features by using fixed convolution kernels and layer-wise representation, resulting in feature extraction singleness. Additionally, the feature fusion process is rough and simple. Numerous methods get accustomed to fusing different levels of features by stacking modules hierarchically, which ignore the combination of shallow and deep spectral-spatial features. In order to overcome the preceding issues, a novel multiscale dual-branch feature fusion and attention network is proposed. Specifically, we design a multiscale feature extraction (MSFE) module to extract spatial-spectral features at a granular level and expand the range of receptive fields, thereby enhancing the MSFE ability. Subsequently, we develop a dual-branch feature fusion interactive module that integrates the residual connection's feature reuse property and the dense connection's feature exploration capability, obtaining more discriminative features in both spatial and spectral branches. Additionally, we introduce a novel shuffle attention mechanism that allows for adaptive weighting of spatial and spectral features, further improving classification performance. Experimental results on three benchmark datasets demonstrate that our model outperforms other state-of-the-art methods while incurring the lower computational cost.
引用
收藏
页码:8180 / 8192
页数:13
相关论文
共 56 条
[1]   Atmospheric Column Water Vapor Retrieval From Hyperspectral VNIR Data Based on Low-Rank Subspace Projection [J].
Acito, N. ;
Diani, M. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (07) :3924-3940
[2]  
Alonso M.C., 2011, Proc. ASPRS annual conference, P1
[3]  
[Anonymous], 2018, IEEE T GEOSCI ELECT
[4]   Classification of Hyperspectral Images With Regularized Linear Discriminant Analysis [J].
Bandos, Tatyana V. ;
Bruzzone, Lorenzo ;
Camps-Valls, Gustavo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2009, 47 (03) :862-873
[5]   Detection of early plant stress responses in hyperspectral images [J].
Behmann, Jan ;
Steinruecken, Joerg ;
Pluemer, Lutz .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2014, 93 :98-111
[6]  
Brook A., 2010, Hyperspectral Image and Signal Processing: Evolution in Remote Sensing (WHISPERS), Reykjavik, Iceland, 14-16 June 2010, P1
[7]   Hyperspectral Image Classification With Convolutional Neural Network and Active Learning [J].
Cao, Xiangyong ;
Yao, Jing ;
Xu, Zongben ;
Meng, Deyu .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (07) :4604-4616
[8]   GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond [J].
Cao, Yue ;
Xu, Jiarui ;
Lin, Stephen ;
Wei, Fangyun ;
Hu, Han .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, :1971-1980
[9]   Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network [J].
Chen, Yushi ;
Zhao, Xing ;
Jia, Xiuping .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) :2381-2392
[10]   Deep Learning-Based Classification of Hyperspectral Data [J].
Chen, Yushi ;
Lin, Zhouhan ;
Zhao, Xing ;
Wang, Gang ;
Gu, Yanfeng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (06) :2094-2107