Hyperspectral Image Classification Based on Dual-Scale Dense Network with Efficient Channel Attentional Feature Fusion

被引:2
作者
Shi, Zhongyang [1 ]
Chen, Ming [1 ]
Wu, Zhigao [1 ]
机构
[1] Shanghai Ocean Univ, Coll Informat Sci, Key Lab Fisheries Informat, Minist Agr & Rural Affairs, Shanghai 201306, Peoples R China
关键词
hyperspectral image classification; dense network; separable convolution; efficient channel attention; feature fusion; NEURAL-NETWORKS;
D O I
10.3390/electronics12132991
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Hyperspectral images (HSIs) have abundant spectral and spatial information, which shows bright prospects in the application industry of urban-rural. Thus, HSI classification has drawn much attention from researchers. However, the spectral and spatial information-extracting method is one of the research difficulties in HSI classification tasks. To meet this tough challenge, we propose an efficient channel attentional feature fusion dense network (CA-FFDN). Our network has two structures. In the feature extraction structure, we utilized a novel bottleneck based on separable convolution (SC-bottleneck) and efficient channel attention (ECA) to simultaneously fuse spatial-spectral features from different depths, which can make full use of the dual-scale shallow and deep spatial-spectral features of the HSI and also significantly reduce the parameters. In the feature enhancement structure, we used 3D convolution and average pooling to further integrate spatial-spectral features. Many experiments on Indian Pines (IP), University of Pavia (UP), and Kennedy Space Center (KSC) datasets demonstrated that our CA-FFDN outperformed the other five state-of-the-art networks, even with small training samples. Meanwhile, our CA-FFDN achieved classification accuracies of 99.51%, 99.91%, and 99.89%, respectively, in the case where the ratio of the IP, UP, and KSC datasets was 2:1:7, 1:1:8, and 2:1:7. It provided the best classification performance with the highest accuracy, fastest convergence, and slightest training and validation loss fluctuations.
引用
收藏
页数:20
相关论文
共 58 条
[1]   Hyperparameter Optimization: Comparing Genetic Algorithm against Grid Search and Bayesian Optimization [J].
Alibrahim, Hussain ;
Ludwig, Simone A. .
2021 IEEE CONGRESS ON EVOLUTIONARY COMPUTATION (CEC 2021), 2021, :1551-1559
[2]   Pruning Multi-Scale Multi-Branch Network for Small-Sample Hyperspectral Image Classification [J].
Bai, Yu ;
Xu, Meng ;
Zhang, Lili ;
Liu, Yuxuan .
ELECTRONICS, 2023, 12 (03)
[3]   Hyperspectral Imagery Detects Water Deficit and Salinity Effects on Photosynthesis and Antioxidant Enzyme Activity of Three Greek Olive Varieties [J].
Boshkovski, Blagoja ;
Doupis, Georgios ;
Zapolska, Anhelina ;
Kalaitzidis, Chariton ;
Koubouris, Georgios .
SUSTAINABILITY, 2022, 14 (03)
[4]   Composite kernels for hyperspectral image classification [J].
Camps-Valls, G ;
Gomez-Chova, L ;
Muñoz-Marí, J ;
Vila-Francés, J ;
Calpe-Maravilla, J .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2006, 3 (01) :93-97
[5]   Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks [J].
Chen, Yushi ;
Jiang, Hanlu ;
Li, Chunyang ;
Jia, Xiuping ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10) :6232-6251
[6]   Attentional Feature Fusion [J].
Dai, Yimian ;
Gieseke, Fabian ;
Oehmcke, Stefan ;
Wu, Yiquan ;
Barnard, Kobus .
2021 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION WACV 2021, 2021, :3559-3568
[7]   Hyperspectral Images Classification Based on Dense Convolutional Networks with Spectral-Wise Attention Mechanism [J].
Fang, Bei ;
Li, Ying ;
Zhang, Haokui ;
Chan, Jonathan Cheung-Wai .
REMOTE SENSING, 2019, 11 (02)
[8]   Deep Hashing Neural Networks for Hyperspectral Image Feature Extraction [J].
Fang, Leyuan ;
Liu, Zhiliang ;
Song, Weiwei .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2019, 16 (09) :1412-1416
[9]   Dual Attention Network for Scene Segmentation [J].
Fu, Jun ;
Liu, Jing ;
Tian, Haijie ;
Li, Yong ;
Bao, Yongjun ;
Fang, Zhiwei ;
Lu, Hanqing .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :3141-3149
[10]  
Howard AG, 2017, Arxiv, DOI [arXiv:1704.04861, DOI 10.48550/ARXIV.1704.04861, 10.48550/arXiv.1704.04861]