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 条
[11]   Dual Graph Convolutional Network for Hyperspectral Image Classification With Limited Training Samples [J].
He, Xin ;
Chen, Yushi ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[12]   Graph Convolutional Networks for Hyperspectral Image Classification [J].
Hong, Danfeng ;
Gao, Lianru ;
Yao, Jing ;
Zhang, Bing ;
Plaza, Antonio ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (07) :5966-5978
[13]   Remote Sensing of 3-D Geometry and Surface Moisture of a Peat Production Area Using Hyperspectral Frame Cameras in Visible to Short-Wave Infrared Spectral Ranges Onboard a Small Unmanned Airborne Vehicle (UAV) [J].
Honkavaara, Eija ;
Eskelinen, Matti A. ;
Polonen, Ilkka ;
Saari, Heikki ;
Ojanen, Harri ;
Mannila, Rami ;
Holmlund, Christer ;
Hakala, Teemu ;
Litkey, Paula ;
Rosnell, Tomi ;
Viljanen, Niko ;
Pulkkanen, Merja .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (09) :5440-5454
[14]   Graph Neural Network via Edge Convolution for Hyperspectral Image Classification [J].
Hu, Haojie ;
Yao, Minli ;
He, Fang ;
Zhang, Fenggan .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[15]   Spectral-Spatial Masked Transformer With Supervised and Contrastive Learning for Hyperspectral Image Classification [J].
Huang, Lingbo ;
Chen, Yushi ;
He, Xin .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[16]  
Kipf T. N., 2017, INT C LEARN REPR, P1, DOI DOI 10.48550/ARXIV.1609.02907
[17]  
Lafferty J., 2001, P 18 INT C MACH LEAR, P282
[18]   Hyperspectral image classification via nonlocal joint kernel sparse representation based on local covariance [J].
Li, Dan ;
Kong, Fanqiang ;
Wang, Qiang .
SIGNAL PROCESSING, 2021, 180
[19]  
Li Guangfei, 2023, IEEE Trans. Circuits Syst. Video Technol.
[20]   Classification of Hyperspectral Image Based on Double-Branch Dual-Attention Mechanism Network [J].
Li, Rui ;
Zheng, Shunyi ;
Duan, Chenxi ;
Yang, Yang ;
Wang, Xiqi .
REMOTE SENSING, 2020, 12 (03)