DSGAU: Dual-Scale Graph Attention U-Nets for Hyperspectral Image Classification With Limited Samples
Graph convolutional networks (GCNs) exhibit remarkable capabilities in hyperspectral image (HSI) classification tasks, primarily due to their ability to establish long-range pixel correlations. This enables the simultaneous learning of local spectral features and global contextual patterns within HSI data. However, the convolutional operations in traditional GCNs require the inclusion of all data points during graph construction, leading to significant computational overhead, particularly for large-scale datasets. To this challenge, graph pooling effectively mitigates the high computational costs of GCNs through hierarchical downsampling, adaptive node selection, and feature preservation mechanisms. Nevertheless, prevalent graph pooling techniques often employ single-scale strategies that inadequately capture multi-scale features, potentially leading to information loss or redundancy. To address this issue, we propose a dualscale graph attention U-Nets (DSGAU) for HSI classification with limited samples. First, we design a cross-scale and self-attention (CSSA) module to reduce the graph structure while extracting detailed information from HSI for graph construction, where a cross-scale attention branch establishes inter-level feature correlations while a self-attention branch enhances intra-level contextual learning. Second, we design a dual-scale constrained graph U-Nets encoder, and use an attention feature fusion (AFF) module dynamically weights these multi-scale representations using channel-wise attention coefficients, effectively resolving feature redundancy issues. Finally, we introduce the graph attention network with contrastive normalization layer (GAT-CNL) module to replace traditional GCNs, enabling dynamic graph structure updating during propagation alleviating oversmoothing through differential feature enhancement. The classification performance of the proposed method is evaluated on four benchmark datasets. Experimental results show that, the proposed method outperforms existing cutting-edge methods, with improvements in terms of overall accuracy around 2.37%˜17.8% (Indian Pines), 1.05%˜10.3% (Salinas), 1.54%˜9.64% (WHU-Hi-LongKou) under limited training sample conditions.
Authors
- Zhaohui Xue (ORCID: https://orcid.org/0000-0001-6253-2967)
- Hongjun Su (ORCID: https://orcid.org/0000-0002-8991-8568)
- Hua Ji (ORCID: https://orcid.org/0000-0001-9782-7992)
- Leying Song
Institutions
- Hohai University (CN)
- Qinghai Meteorological Bureau (CN)
Publication Details
- Journal
- IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1109/jstars.2025.3580769
- Citations
- 1
- Primary Topic
- Remote-Sensing Image Classification
- Type
- article
- Field-Weighted Citation Impact
- 1.14
Funders
- National Natural Science Foundation of China
- Natural Science Foundation of Jiangsu Province