DSGAU: Dual-Scale Graph Attention U-Nets for Hyperspectral Image Classification With Limited Samples
Year: 2025
Type: article
Abstract: 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: Hongzhuang Ji, Leying Song, Zhaohui Xue, Hongjun Su
Institution: Hohai University
Language: English
Subfield: Media Technology
Field: Engineering
Domain: Physical Sciences
Open Access status: gold
Estimated APC paid: $1,800
Awards: G2659737875, G5116539810
OpenAlex ID: W4411408511