Dual-Branch Spectral–Spatial Attention Network for Hyperspectral Image Classification

被引:9
作者
Zhao, Jinling [1 ]
Wang, Jiajie [1 ]
Ruan, Chao [1 ]
Dong, Yingying [2 ]
Huang, Linsheng [1 ]
机构
[1] Anhui Univ, Hefei 230601, Peoples R China
[2] Chinese Acad Sci, Aerosp Informat Res Inst, Key Lab Digital Earth Sci, Beijing 100094, Peoples R China
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2024年 / 62卷
基金
中国国家自然科学基金;
关键词
Feature extraction; Convolutional neural networks; Data mining; Transformers; Hyperspectral imaging; Computational modeling; Principal component analysis; Central pixels; hyperspectral classification; multilayer perceptron (MLP); spectral-spatial attention; Transformer;
D O I
10.1109/TGRS.2024.3351997
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
In order to achieve accurate hyperspectral image (HSI) classification, the convolutional neural network (CNN) has been extensively utilized. However, most existing patch-based CNN methods overlook the relationship between central pixels and their surroundings. A novel dual-branch spectral-spatial attention network (DBSSAN) is proposed, which helps suppress the impact from interference elements and enhances effective feature extraction from complex features in HSI data. The global and local spatial features are fully integrated through the proposed spatial self-attention module. More specifically, it measures the relationship between the central and surrounding pixels based on cosine similarity and Gaussian-Euclidean similarity to extract global features, while the scale information extraction (SIE) model captures the local features. Furthermore, the inclusion of Transformer model enables the extraction of spectral information from a global perspective, facilitating the capture of long-distance dependencies and nonlinear correlations in HSI. The extracted spectral and spatial features are subsequently classified using a multilayer perceptron (MLP). Five publicly available hyperspectral datasets were used to present experimental evaluations, namely, Indian Pines, Kennedy Space Center, Pavia University, Houston2013, and Houston2018. The comparative results demonstrate the superior performance of the proposed network compared to several state-of-the-art methods.
引用
收藏
页码:1 / 18
页数:18
相关论文
共 49 条
[1]   Classification of hyperspectral data from urban areas based on extended morphological profiles [J].
Benediktsson, JA ;
Palmason, JA ;
Sveinsson, JR .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2005, 43 (03) :480-491
[2]   A NEW K-NEAREST NEIGHBOR DENSITY-BASED CLUSTERING METHOD AND ITS APPLICATION TO HYPERSPECTRAL IMAGES [J].
Cariou, Claude ;
Chehdi, Kacem .
2016 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2016, :6161-6164
[3]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[4]   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
[5]   Exploring Hierarchical Convolutional Features for Hyperspectral Image Classification [J].
Cheng, Gong ;
Li, Zhenpeng ;
Han, Junwei ;
Yao, Xiwen ;
Guo, Lei .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (11) :6712-6722
[6]   Lightweight Spectral-Spatial Attention Network for Hyperspectral Image Classification [J].
Cui, Ying ;
Xia, Jinbiao ;
Wang, Zhiteng ;
Gao, Shan ;
Wang, Liguo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[7]   Classification of Hyperspectral Images by Using Extended Morphological Attribute Profiles and Independent Component Analysis [J].
Dalla Mura, Mauro ;
Villa, Alberto ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Bruzzone, Lorenzo .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2011, 8 (03) :542-546
[8]   Classification via Structure-Preserved Hypergraph Convolution Network for Hyperspectral Image [J].
Duan, Yule ;
Luo, Fulin ;
Fu, Maixia ;
Niu, Yingying ;
Gong, Xiuwen .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[9]   Crop Type Classification by DESIS Hyperspectral Imagery and Machine Learning Algorithms [J].
Farmonov, Nizom ;
Amankulova, Khilola ;
Szatmari, Jozsef ;
Sharifi, Alireza ;
Abbasi-Moghadam, Dariush ;
Nejad, Seyed Mahdi Mirhoseini ;
Mucsi, Laszlo .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 :1576-1588
[10]   Spectral and Spatial Classification of Hyperspectral Data Using SVMs and Morphological Profiles [J].
Fauvel, Mathieu ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Sveinsson, Johannes R. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2008, 46 (11) :3804-3814