高光谱成像
计算机科学
模式识别(心理学)
人工智能
卷积神经网络
邻接矩阵
图形
平滑的
上下文图像分类
图像(数学)
计算机视觉
理论计算机科学
作者
Quanwei Liu,Yanni Dong,Yuxiang Zhang,Hui Luo
标识
DOI:10.1109/tgrs.2022.3179419
摘要
Deep learning has achieved impressive results on hyperspectral images (HSIs) classification. Among them, both convolutional neural networks (CNNs) and graph neural networks (GNNs) have great potential for hyperspectral image classification. Supervised CNNs can efficiently extract hierarchical spatial-spectral features of hyperspectral images, but these methods face the problem of high time complexity as the number of network layers increases. Semi-supervised GNNs can rapidly capture the structural information of HSIs, while they cannot be well extended to hyperspectral image applications because of the process of adjacency matrix consuming large amount of memory resources. In this paper, we propose a fast dynamic graph convolutional network (dynamic GCN) and CNN parallel network (FDGC) for HSI classification. We first obtain two classification features by flattening and pooling operations on the results of the convolution layers, which fully exploits the spatial-spectral information contained in the hyperspectral data cube. Then a dynamic graph convolution module is applied to extract the intrinsic structural information of each patch. Finally, we can obtain the HSI classification results based on these spatial, spectral and structural features. By using three branches, FDGC can parallel process multiple features of HSI in a supervised learning manner. In addition, regularization techniques such as DropBlock and label smoothing are applied to further improve the generalization capability of the model. Experimental results on three datasets show that our proposed algorithm is comparable with the state-of-the-art supervised learning models in terms of accuracy while also significantly outperforming in terms of training and inference time.
科研通智能强力驱动
Strongly Powered by AbleSci AI