Hyperspectral image classification based on mixed similarity graph convolutional network and pixel refinement

被引:0
作者
Shang, Ronghua [1 ]
Zhu, Keyao [1 ]
Chang, Huidong [1 ]
Zhang, Weitong [1 ]
Feng, Jie [1 ]
Xu, Songhua [2 ]
机构
[1] Xidian Univ, Sch Artificial Intelligence, Key Lab Intelligent Percept & Image Understanding, Minist Educ, Xian 710071, Shaanxi Provinc, Peoples R China
[2] Xi An Jiao Tong Univ, Dept Hlth Management & Inst Med Artificial Intelli, Affiliated Hosp 2, Xian 710004, Shaanxi Provinc, Peoples R China
基金
中国国家自然科学基金; 中国博士后科学基金;
关键词
Hyperspectral image classification; Mixed similarity; Conditional random fields; Graph convolutional networks; Superpixel; OF-THE-ART;
D O I
10.1016/j.asoc.2024.112657
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The traditional convolutional neural network cannot extract Non-Euclidean spatial information in hyperspectral image classification, while the superpixel-based graph convolutional network relies on node aggregation and superpixel segmentation accuracy. To solve these problems, a mixed similarity graph convolutional network and pixel refinement method (MSGCN-CRF) is proposed in this paper. Firstly, homogeneous superpixel-level features are obtained as nodes using a superpixel segmentation algorithm, and then a mixed similarity method is designed to aggregate nodes. This method combines spectral intensity similarity and spectral curve similarity to obtain more discriminant node features. Secondly, a two-layer convolutional layer is used to remove noise from the original hyperspectral image. Then node features are constructed and input into the two-layer graph convolutional network, which using the adjacency matrix generated by mixed similarity to guide superpixel node aggregation. Finally, pixel refinement is implemented by fully connect CRF, which is used to correct the false prediction caused by the segmentation error of superpixels, and to obtain more accurate classification results using spatial and spectral information between pixels. Experimental results with a small amount of training samples on three different datasets show that the proposed MSGCN-CRF can obtain better classification results than the seven state-of-the-art classification methods.
引用
收藏
页数:12
相关论文
共 43 条
[1]   SLIC Superpixels Compared to State-of-the-Art Superpixel Methods [J].
Achanta, Radhakrishna ;
Shaji, Appu ;
Smith, Kevin ;
Lucchi, Aurelien ;
Fua, Pascal ;
Suesstrunk, Sabine .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2012, 34 (11) :2274-2281
[2]   Hyperspectral Image Classification Based on Deep Attention Graph Convolutional Network [J].
Bai, Jing ;
Ding, Bixiu ;
Xiao, Zhu ;
Jiao, Licheng ;
Chen, Hongyang ;
Regan, Amelia C. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[3]   Kernel-based methods for hyperspectral image classification [J].
Camps-Valls, G ;
Bruzzone, L .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2005, 43 (06) :1351-1362
[4]   Automatic Graph Learning Convolutional Networks for Hyperspectral Image Classification [J].
Chen, Jie ;
Jiao, Licheng ;
Liu, Xu ;
Li, Lingling ;
Liu, Fang ;
Yang, Shuyuan .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[5]   Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network [J].
Chen, Yushi ;
Zhao, Xing ;
Jia, Xiuping .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) :2381-2392
[6]   Spatial-Spectral Unified Adaptive Probability Graph Convolutional Networks for Hyperspectral Image Classification [J].
Ding, Yun ;
Chong, Yanwen ;
Pan, Shaoming ;
Wang, Yujie ;
Nie, Congchong .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (07) :3650-3664
[7]   Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification [J].
Dong, Yanni ;
Liu, Quanwei ;
Du, Bo ;
Zhang, Liangpei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 :1559-1572
[8]   Target detection based on a dynamic subspace [J].
Du, Bo ;
Zhang, Liangpei .
PATTERN RECOGNITION, 2014, 47 (01) :344-358
[9]   A relative evaluation of multiclass image classification by support vector machines [J].
Foody, GM ;
Mathur, A .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2004, 42 (06) :1335-1343
[10]   Dual Graph U-Nets for Hyperspectral Image Classification [J].
Guo, Fangming ;
Li, Zhongwei ;
Xin, Ziqi ;
Zhu, Xue ;
Wang, Leiquan ;
Zhang, Jie .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 :8160-8170