Graph Sample and Aggregate-Attention Network for Hyperspectral Image Classification
Yao Ding, Xiaofeng Zhao, Zhili Zhang, Wei Cai, Nengjun Yang
Abstract
Graph convolutional network (GCN) has shown potential in hyperspectral image (HSI) classification. However, GCN is a transductive learning method, which is difficult to aggregate the new node. The available GCN-based methods fail to understand the global and contextual information of the graph. To address this deficiency, a novel semisupervised network based on graph sample and aggregate-attention (SAGE-A) for HSIs’ classification is proposed. Different from the GCN-based method, SAGE-A adopts a multilevel graph sample and aggregate (graphSAGE) network, as it can flexibly aggregate the new neighbor node among arbitrarily structured non-Euclidean data and capture long-range contextual relations. Inspired by the convolution neural network (CNN) self-attention mechanism, the proposed network uses the graph attention mechanism to characterize the importance among spatially neighboring regions, so the deep contextual and global information of the graph can be learned automatically by focusing on important spatial targets. Extensive experimental results on different real hyperspectral data sets demonstrate the performances of our proposed method compared with the state-of-the-art methods.