Hyperspectral Image Classification Based on Graph Transformer Network and Graph Attention Mechanism
Xiaofeng Zhao, Jiahui Niu, Chuntong Liu, Yao Ding, Danfeng Hong
Abstract
Graph convolutional networks (GCN) have begun to show their potential in hyperspectral image classification in recent years. However, most of the current GCN methods are designed to learn node features on fixed and homogeneous graphs, and it is difficult for them to learn effective node features on heterogeneous graphs. The limitation is particularly evident in hyperspectral classification because of the different types of nodes and edges. The Graph Transformer Network with the graph attention mechanism (GTN-A) is proposed to address this shortcoming in this paper. It can generate a new graph structure, which is represented by a more useful meta-path, so that node features can be better aggregated. The experiments conducted on two benchmark datasets illustrate the effectiveness of our method.