Hyperspectral Image Classification Framework Based on Multichannel Graph Convolutional Networks and Class-Guided Attention Mechanism

Hao Feng,Yongcheng Wang,Chi Chen,Dongdong Xu,Zhikang Zhao,Tianqi Zhao
DOI: https://doi.org/10.1109/tgrs.2024.3388429
IF: 8.2
2024-04-27
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Graph convolutional networks (GCNs) can extract features of samples in non-Euclidean space, which can be used for hyperspectral image (HSI) classification in collaboration with convolutional neural networks (CNNs). The features of GCNs and CNNs are incompatible to a certain extent, and traditional graph convolution methods use a single-channel 2-D matrix to extract features. As a result, it is difficult to explore the relationships fully and flexibly between samples. To further exploit the potential of these two networks for collaborative extraction of HSI features, we propose a fusion framework based on multichannel GCNs and class-guided attention mechanism (MG2A). Specifically, a multichannel GCN (MGCN) module is designed for batchwise network training, where the adjacent matrix of each channel contains different information between samples. In addition, we develop a class-guided attention mechanism to adaptively fuse the features of multiple MGCN modules and learn the transformation process of the features. Finally, the features of CNNs and MGCN modules are fused at multiple layers through a fusion framework. Experimental results on four benchmark HSI datasets show that MG2A achieves better classification performance compared to other state-of-the-art methods.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?