DesU-NetAM: optimized DenseU-Net with attention mechanism for hyperspectral image classification

被引:2
作者
Balaji K. [1 ]
Nirosha V. [1 ]
Yallamandaiah S. [2 ]
Karthik S. [3 ]
Prasad V.S. [4 ]
Prathyusha G. [1 ]
机构
[1] Department of Computer Science and Engineering, B V Raju Institute of Technology, Vishnupur, Telangana, Narsapur
[2] Department of Electronics and Communication Engineering, Vignan’s Nirula Institute of Technology and Science for Women, Peda Palakaluru, Andhra Pradesh, Guntur
[3] Department of Computer Science and Engineering, R. V. R & J. C College of Engineering, Chowdavaram, Andhra Pradesh, Guntur
[4] Department of Computer Science and Engineering, Sree Vidyanikethan Engineering College, Sree Sainath Nagar, Andhra Pradesh, Tirupati
关键词
Attention mechanism; DenseU-Net; Hyperspectral images (HSI); Spectral and spatial data; Tuna swarm optimization (TSO) algorithm;
D O I
10.1007/s41870-023-01386-5
中图分类号
学科分类号
摘要
The utilization of hyperspectral images (HSI) is expanding rapidly with the advancement of remote sensing technology. Accurately categorizing ground features using HSI is a crucial research topic that has garnered considerable interest. The high dimensional space, numerous spectral bands, and lack of labeled training data make categorizing hyperspectral images difficult. We provide a novel hyperspectral image categorization approach based on DenseU-Net to address these issues. The data is first normalized by separating the maximum value of the entire set of data by the average intensity of each pixel. Then we also propose an attention mechanism network. Because spectral and spatial data are extracted independently, there may be less interference between the two types of features in this network. The retrieved spectral and spatial data are integrated to categorize the data. Finally, DenseU-Net method parameters are adjusted using the Tuna Swarm Optimization (TSO) algorithm, which has a global search capability. This optimized DenseU-Net is then utilized to manage a hyperspectral image categorization approach effectively. Three standard HSI datasets utilized for experiments were acquired by various sensors at different acquisition times and used for classification studies. The comparison findings show that the proposed approach outperforms other deep learning methods models in classification effectiveness. © 2023, The Author(s), under exclusive licence to Bharati Vidyapeeth's Institute of Computer Applications and Management.
引用
收藏
页码:3761 / 3777
页数:16
相关论文
共 29 条
[1]  
Sun L., Zhao G., Zheng Y., Wu Z., Spectral–spatial feature tokenization transformer for hyperspectral image classification, IEEE Trans Geosci Remote Sens, 60, pp. 1-14, (2022)
[2]  
Hou S., Shi H., Cao X., Zhang X., Jiao L., Hyperspectral imagery classification based on contrastive learning, IEEE Trans Geosci Remote Sens, 60, pp. 1-13, (2021)
[3]  
Ding Y., Zhang Z., Zhao X., Hong D., Cai W., Yu C., Cai W., Multi-feature fusion: graph neural network and CNN combining for hyperspectral image classification, Neurocomputing, 501, pp. 246-257, (2022)
[4]  
Cao X., Yao J., Xu Z., Meng D., Hyperspectral image classification with convolutional neural network and active learning, IEEE Trans Geosci Remote Sens, 58, 7, pp. 4604-4616, (2020)
[5]  
Qing Y., Liu W., Feng L., Gao W., Improved transformer net for hyperspectral image classification, Remote Sens, 13, 11, (2021)
[6]  
Ghaderizadeh S., Abbasi-Moghadam D., Sharifi A., Zhao N., Tariq A., Hyperspectral image classification using a hybrid 3D–2D convolutional neural networks, IEEE J Sel Top Appl Earth Obs Remote Sens, 14, pp. 7570-7588, (2021)
[7]  
Ge Z., Cao G., Li X., Fu P., Hyperspectral image classification method based on 2D–3D CNN and multibranch feature fusion, IEEE J Sel Top Appl Earth Obs Remote Sens, 13, pp. 5776-5788, (2020)
[8]  
Dong Y., Liu Q., Du B., Zhang L., Weighted feature fusion of convolutional neural network and graph attention network for hyperspectral image classification, IEEE Trans Image Process, 31, pp. 1559-1572, (2022)
[9]  
Hang R., Liu Q., Hong D., Ghamisi P., Cascaded recurrent neural networks for hyperspectral image classification, IEEE Trans Geosci Remote Sens, 57, 8, pp. 5384-5394, (2019)
[10]  
Subba Reddy T., Harikiran J., Enduri M.K., Hajarathaiah K., Almakdi S., Alshehri M., Rahman M.H., Hyperspectral image classification with optimized compressed synergic deep convolution neural network with aquila optimization, Comput Intell Neurosci, (2022)