DSGAU: Dual-Scale Graph Attention U-Nets for Hyperspectral Image Classification With Limited Samples

被引:0
作者
Ji, Hongzhuang [1 ,2 ,3 ]
Song, Leying [4 ]
Xue, Zhaohui [1 ,2 ,3 ]
Su, Hongjun [1 ,2 ,3 ]
机构
[1] Hohai Univ, Coll Geog & Remote Sensing, Nanjing 211100, Peoples R China
[2] Hohai Univ, Jiangsu Prov Engn Res Ctr Watershed Geospatial Int, Nanjing 211100, Peoples R China
[3] Hohai Univ, Key Lab Soil & Water Proc Watershed, Nanjing 211000, Peoples R China
[4] Qingzhou Bur Nat Resources & Planning, Qingdao 262500, Peoples R China
基金
中国国家自然科学基金;
关键词
Feature extraction; Convolution; Graph convolutional networks; Hyperspectral imaging; Deep learning; Correlation; Attention mechanisms; Accuracy; Image classification; Computational modeling; Convolutional neural network (CNN); graph attention network (GAT); graph U-Nets; hyperspectral image (HSI) classification; limited samples; NETWORKS;
D O I
10.1109/JSTARS.2025.3580769
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
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 multiscale features, potentially leading to information loss or redundancy. To address this issue, we propose a dual-scale graph attention U-Nets for HSI classification with limited samples. First, we design a cross-scale and self-attention 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 module dynamically weights these multiscale representations using channel-wise attention coefficients, effectively resolving feature redundancy issues. Finally, we introduce the graph attention network with contrastive normalization layer module to replace traditional GCNs, enabling dynamic graph structure updating during propagation alleviating over-smoothing 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 (OA) around 2.37-17.8% (Indian Pines), 1.05%-10.3% (Salinas), and 1.54%-9.64% (WHU-Hi-LongKou) under limited training sample conditions.
引用
收藏
页码:16134 / 16151
页数:18
相关论文
共 55 条
[1]   3-D Deep Learning Approach for Remote Sensing Image Classification [J].
Ben Hamida, Amina ;
Benoit, Alexandre ;
Lambert, Patrick ;
Ben Amar, Chokri .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (08) :4420-4434
[2]   SDFL-FC: Semisupervised Deep Feature Learning With Feature Consistency for Hyperspectral Image Classification [J].
Cao, Yun ;
Wang, Yuebin ;
Peng, Junhuan ;
Qiu, Chunping ;
Ding, Lei ;
Zhu, Xiao Xiang .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (12) :10488-10502
[3]   Automatic Graph Learning Convolutional Networks for Hyperspectral Image Classification [J].
Chen, Jie ;
Jiao, Licheng ;
Liu, Xu ;
Li, Lingling ;
Liu, Fang ;
Yang, Shuyuan .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[4]   Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks [J].
Chen, Yushi ;
Jiang, Hanlu ;
Li, Chunyang ;
Jia, Xiuping ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10) :6232-6251
[5]   Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network [J].
Chen, Yushi ;
Zhao, Xing ;
Jia, Xiuping .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) :2381-2392
[6]   Deep Learning-Based Classification of Hyperspectral Data [J].
Chen, Yushi ;
Lin, Zhouhan ;
Zhao, Xing ;
Wang, Gang ;
Gu, Yanfeng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (06) :2094-2107
[7]   Graph Sample and Aggregate-Attention Network for Hyperspectral Image Classification [J].
Ding, Yao ;
Zhao, Xiaofeng ;
Zhang, Zhili ;
Cai, Wei ;
Yang, Nengjun .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[8]   Multiscale Graph Sample and Aggregate Network With Context-Aware Learning for Hyperspectral Image Classification [J].
Ding, Yao ;
Zhao, Xiaofeng ;
Zhang, Zhili ;
Cai, Wei ;
Yang, Nengjun .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 :4561-4572
[9]   Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification [J].
Dong, Yanni ;
Liu, Quanwei ;
Du, Bo ;
Zhang, Liangpei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 :1559-1572
[10]   Graph U-Nets [J].
Gao, Hongyang ;
Ji, Shuiwang .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (09) :4948-4960