Graph Convolutional Network with Local and Global Feature Fusion for Hyperspectral Image Classification

Yufan Wang,Xiaodong Yu,Hongbin Dong,Shuying Zang
DOI: https://doi.org/10.1109/tgrs.2024.3487296
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:In Hyperspectral Image (HSI) classification tasks, extracting surface features is crucial but the complexity of HSI brings considerable challenges. Traditional Convolutional Neural Networks (CNNs) and (Vision Transformer) ViT perform well but have large parameters and are time consuming. In contrast, Graph Convolutional Neural Networks (GCNs) are fast to train with few parameters. The SLIC segmentation algorithm generates pixel-level graph data and super-pixel-level graph data. Single-branch GCNs learn only one type of graph data resulting in insufficient feature richness. In two-branch GCNs, one branch extracts local spatial-spectral features from pixel-level data, another branch extracts global spatial-spectral features from super-pixel-level data, and finally the two branches are combined. But the existing two-branch GCNs do not focus on the close fusion of local and global features. Therefore, in this paper, we propose a new two-branch GCNs with local and global feature fusion (LG-GCN). The LG-GCN model proposed in this study not only has a short training time, but also has a high accuracy. The LG-GCN can capture more comprehensively the local and global spatial-spectral semantic information of HSI. Experimental results on three publicly available HSI datasets show that LG-GCN outperforms other state-of-the-art models in terms of classification accuracy and robustness. In addition, it has a shorter training time and is more resistant to noise. The source code of the LG-GCN model is available at https://github.com/Dreamvai/LG-GCN.
What problem does this paper attempt to address?