Hyperspectral Image Classification Based on Deep Attention Graph Convolutional Network

被引:95
作者
Bai, Jing [1 ]
Ding, Bixiu [1 ]
Xiao, Zhu [2 ]
Jiao, Licheng [1 ]
Chen, Hongyang [3 ]
Regan, Amelia C. [4 ,5 ]
机构
[1] Xidian Univ, Sch Artificial Intelligence, Key Lab Intelligent Percept & Image Understanding, Minist Educ, Xian 710071, Peoples R China
[2] Hunan Univ, Coll Comp Sci & Elect Engn, Changsha 410082, Hunan, Peoples R China
[3] Zhejiang Lab, Hangzhou 311121, Peoples R China
[4] Univ Calif Irvine, Dept Comp Sci, Irvine, CA 92697 USA
[5] Univ Calif Irvine, Inst Transportat Studies, Irvine, CA 92697 USA
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2022年 / 60卷
基金
中国国家自然科学基金;
关键词
Feature extraction; Hyperspectral imaging; Convolution; Data mining; Correlation; Kernel; Spatial resolution; Attention mechanism; graph convolution network (GCN); hyperspectral image classification (HIC); similarity measurement; NEURAL-NETWORKS;
D O I
10.1109/TGRS.2021.3066485
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Hyperspectral images (HSIs) have gained high spectral resolution due to recent advances in spectral imaging technologies. This incurs problems, such as an increased data scale and an increased number of bands for HSIs, which results in a complex correlation between different bands. In the applications of remote sensing and earth observation, ground objects represented by each HSI pixel are composed of physical and chemical non-Euclidean structures, and HSI classification (HIC) is becoming a more challenging task. To solve the above problems, we propose a framework based on a deep attention graph convolutional network (DAGCN). Specifically, we first integrate an attention mechanism into the spectral similarity measurement to aggregate similar spectra. Therefore, we propose a new similarity measurement method, i.e., the mixed measurement of a kernel spectral angle mapper and spectral information divergence (KSAM-SID), to aggregate similar spectra. Considering the non-Euclidean structural characteristics of HSIs, we design deep graph convolutional networks (DeepGCNs) as a feature extraction method to extract deep abstract features and explore the internal relationship between HSI data. Finally, we dynamically update the attention graph adjacency matrix to adapt to the changes in each feature graph. Experiments on three standard HSI data sets, namely, the Indian Pines, Pavia University, and Salinas data sets, demonstrate that the DAGCN outperforms the baselines in terms of various evaluation criteria. For example, on the Indian Pines data set, the overall accuracy of the proposed method achieves 98.61x0025; when the training sample is 10x0025;.
引用
收藏
页数:16
相关论文
共 52 条
[1]   Deep Learning With Attribute Profiles for Hyperspectral Image Classification [J].
Aptoula, Erchan ;
Ozdemir, Murat Can ;
Yanikoglu, Berrin .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2016, 13 (12) :1970-1974
[2]   Vector Attribute Profiles for Hyperspectral Image Classification [J].
Aptoula, Erchan ;
Dalla Mura, Mauro ;
Lefevre, Sebastien .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (06) :3208-3220
[3]   Class Incremental Learning With Few-Shots Based on Linear Programming for Hyperspectral Image Classification [J].
Bai, Jing ;
Yuan, Anran ;
Xiao, Zhu ;
Zhou, Huaji ;
Wang, Dingchen ;
Jiang, Hongbo ;
Jiao, Licheng .
IEEE TRANSACTIONS ON CYBERNETICS, 2022, 52 (06) :5474-5485
[4]  
Camps-Valls G, 2014, IEEE SIGNAL PROC MAG, V31, P45, DOI 10.1109/MSP.2013.2279179
[5]   Integration of 3-dimensional discrete wavelet transform and Markov random field for hyperspectral image classification [J].
Cao, Xiangyong ;
Xu, Lin ;
Meng, Deyu ;
Zhao, Qian ;
Xu, Zongben .
NEUROCOMPUTING, 2017, 226 :90-100
[6]  
Chein-I Chang, 1999, IEEE 1999 International Geoscience and Remote Sensing Symposium. IGARSS'99 (Cat. No.99CH36293), P509, DOI 10.1109/IGARSS.1999.773549
[7]   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
[8]   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
[9]   Semisupervised Self-Learning for Hyperspectral Image Classification [J].
Dopido, Inmaculada ;
Li, Jun ;
Marpu, Prashanth Reddy ;
Plaza, Antonio ;
Bioucas Dias, Jose M. ;
Benediktsson, Jon Atli .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2013, 51 (07) :4032-4044
[10]   New hyperspectral discrimination measure for spectral characterization [J].
Du, YZ ;
Chang, CI ;
Ren, H ;
Chang, CC ;
Jensen, JO ;
D'Amico, FM .
OPTICAL ENGINEERING, 2004, 43 (08) :1777-1786