Back to Search Start Over

CEGAT: A CNN and enhanced-GAT based on key sample selection strategy for hyperspectral image classification.

Authors :
Shi, Cuiping
Wu, Haiyang
Wang, Liguo
Source :
Neural Networks. Nov2023, Vol. 168, p105-122. 18p.
Publication Year :
2023

Abstract

In recent years, the application of convolutional neural networks (CNNs) and graph convolutional networks (GCNs) in hyperspectral image classification (HSIC) has achieved remarkable results. However, the limited label samples are still a major challenge when using CNN and GCN to classify hyperspectral images. In order to alleviate this problem, a double branch fusion network of CNN and enhanced graph attention network (CEGAT) based on key sample selection strategy is proposed. First, a linear discrimination of spectral inter-class slices (LD_SICS) module is designed to eliminate spectral redundancy of HSIs. Then, a spatial spectral correlation attention (SSCA) module is proposed, which can extract and assign attention weight to the spatial and spectral correlation features. On the graph attention (GAT) branch, the HSI is segmented into some super pixels as input to reduce the amount of network parameters. In addition, an enhanced graph attention (EGAT) module is constructed to enhance the relationship between nodes. Finally, a key sample selection (KSS) strategy is proposed to enable the network to achieve better classification performance with few labeled samples. Compared with other state-of-the-art methods, CEGAT has better classification performance under limited label samples. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
08936080
Volume :
168
Database :
Academic Search Index
Journal :
Neural Networks
Publication Type :
Academic Journal
Accession number :
173474629
Full Text :
https://doi.org/10.1016/j.neunet.2023.08.059