Graph Neural Network via Edge Convolution for Hyperspectral Image Classification
Haojie Hu, Minli Yao, Fang He, Fenggan Zhang
Abstract
Graph neural network (GNN) has recently gained increasing attention in the hyperspectral image (HSI) classification. Compared with convolutional neural network (CNN), GNN can effectively relieve the scarcity of labeled data. In our method, we first perform feature learning on large-scale irregular regions through GNN and then extract local spatial–spectral features at the pixel level. Besides, we incorporate edge convolution (EdgeConv) into GNN to adaptively capture the interrelationship of the representative descriptors and fully exploit the discriminative features on graph. Experiments on several HSI datasets show that our method can achieve better classification performance compared with the state-of-the-art HSI classification methods.