Hyperspectral Image Classification Based on Deep Attention Graph Convolutional Network

被引:103
作者
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 条
[41]   Remote sensing for mineral exploration [J].
Sabins, FF .
ORE GEOLOGY REVIEWS, 1999, 14 (3-4) :157-183
[42]   Dynamic Edge-Conditioned Filters in Convolutional Neural Networks on Graphs [J].
Simonovsky, Martin ;
Komodakis, Nikos .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :29-38
[43]  
Thekumparampil Kiran K, 2018, Agent Reinforcement Learning Based on Graph Attention Network for
[44]   Progress in Hyperspectral Remote Sensing Science and Technology in China Over the Past Three Decades [J].
Tong, Qingxi ;
Xue, Yongqi ;
Zhang, Lifu .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (01) :70-91
[45]   Active Learning Methods for Remote Sensing Image Classification [J].
Tuia, Devis ;
Ratle, Frederic ;
Pacifici, Fabio ;
Kanevski, Mikhail F. ;
Emery, William J. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2009, 47 (07) :2218-2232
[46]  
Vaswani A, 2017, ADV NEUR IN, V30
[47]   Hyperspectral Image Classification With Context-Aware Dynamic Graph Convolutional Network [J].
Wan, Sheng ;
Gong, Chen ;
Zhong, Ping ;
Pan, Shirui ;
Li, Guangyu ;
Yang, Jian .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (01) :597-612
[48]   Multiscale Dynamic Graph Convolutional Network for Hyperspectral Image Classification [J].
Wan, Sheng ;
Gong, Chen ;
Zhong, Ping ;
Du, Bo ;
Zhang, Lefei ;
Yang, Jian .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (05) :3162-3177
[49]   Data interpretation for spectral sensors with correlated bands [J].
Wang, Zhipeng ;
Tyo, J. Scott ;
Hayat, Majeed A. .
JOURNAL OF THE OPTICAL SOCIETY OF AMERICA A-OPTICS IMAGE SCIENCE AND VISION, 2007, 24 (09) :2864-2870
[50]  
Xu KYL, 2018, PR MACH LEARN RES, V80