Global-local Feature Adaptive Fusion Мethod for Small Sample Classification of Нyperspectral Images

被引:0
|
作者
Zuo X. [1 ]
Liu Z. [1 ]
Jin F. [1 ]
Lin Y. [1 ]
Wang S. [1 ]
Liu X. [1 ]
Li M. [1 ]
机构
[1] Institute of Geospatial Information, Information Engineering University, Zhengzhou
关键词
adaptive feature fusion; deep learning; depth wise separable convolutional network; dynamic graph convolutional network; hyperspectral image classification; polarized self-attention mechanism; small sample; super pixel segmentation;
D O I
10.12082/dqxxkx.2023.230058
中图分类号
学科分类号
摘要
Acquisition of labeled samples for hyperspectral image classification is usually a time- and labor-consuming task. How to effectively improve the classification accuracy using a small number of samples is one of the challenges in the field of hyperspectral image classification. Most of existing classification methods for hyperspectral images lack sufficient multi-scale information mining, which leads to unsatisfactory classification performance due to small sample numbers. To address the aforementioned issue, this paper designed an adaptive fusion method by integrating global and local features for hyperspectral image classification with small sample numbers. Based on the dynamic graph convolutional network and the depth wise separable convolutional network, a two- branch network structure was constructed to mine the potential information of hyperspectral images from the global and local scales, which realizes the effective usage of labeled samples. Furthermore, the polarization self-attention mechanism was introduced to further improve the expression of intermediate features in the network while cutting down the loss of feature information, and the adaptive feature fusion mechanism was adopted to carry out adaptive fusion of global and local features. Finally, the fusion features flow into the full- connection layer and are manipulated by softmax to obtain prediction labels for each pixel of the hyperspectral image. In order to verify the effectiveness of the proposed method, classification experiments were carried out on four hyperspectral image benchmark data sets including University of Pavia, Salinas, WHU-Hi-LongKou, and WHU-Hi-HanChuan. We discussed and analyzed the influence of model parameters and different modules on the classification accuracy. Subsequently, a comprehensive comparison with seven existing advanced classification methods was conducted in terms of classification visualization, classification accuracy, number of labeled samples, and execution efficiency. The experimental results show that the dynamic graph convolutional network, depth wise separable convolutional network, the polarization self- attention mechanism, and the adaptive feature fusion mechanism all contributed to the improvement of hyperspectral image classification accuracy. In addition, compared with traditional classifiers and advanced deep learning models, the proposed method considered both execution efficiency and classification accuracy, and can achieve better classification performance under the condition of small sample numbers. Specifically, on these four data sets (i.e., University of Pavia, Salinas, WHU- Hi- LongKou and WHU- Hi- HanChuan). The overall classification accuracy was 99.01%, 99.42%, 99.18% and 95.84%, respectively; the average classification accuracy was 99.31%, 99.65%, 98.89% and 95.49%, respectively; and the Kappa coefficient was 98.69%, 99.35%, 98.93% and 95.14%, respectively. © 2023 Journal of Geo-Information Science. All rights reserved.
引用
收藏
页码:1699 / 1716
页数:17
相关论文
共 35 条
  • [21] Wu Z H, Pan S R, Chen F W, Et al., A comprehensive survey on graph neural networks[J], IEEE Transactions on Neural Networks and Learning Systems, 32, 1, pp. 4-24, (2021)
  • [22] Kipf T N, Welling M., Semi-supervised classification with graph convolutional networks, International Conference on Learning Representations, (2017)
  • [23] Zuo X B, Liu B, Yu X C, Et al., Graph convolutional network method for small sample classification of hyperspectral images, Acta Geodaetica et Cartographica sinica, 50, 10, pp. 1358-1369, (2021)
  • [24] Sha A S, Wang B, Wu X F, Et al., Semisupervised classification for hyperspectral images using graph attention networks[J], IEEE Geoscience and Remote Sensing Letters, 18, 1, pp. 157-161, (2021)
  • [25] Ding Y, Zhao X F, Zhang Z L, Et al., Multiscale graph sample and aggregate network with context- aware learning for hyperspectral image classification[J], IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 14, pp. 4561-4572, (2021)
  • [26] Hong D F, Gao L R, Yao J, Et al., Graph convolutional networks for hyperspectral image classification[J], IEEE Transactions on Geoscience and Remote Sensing, 59, 7, pp. 5966-5978, (2021)
  • [27] Liu Q C, Xiao L, Yang J X, Et al., CNN-enhanced graph convolutional network with pixel- and superpixel- level feature fusion for hyperspectral image classification[J], IEEE Transactions on Geoscience and Remote Sensing, 59, 10, pp. 8657-8671, (2021)
  • [28] Chollet F., Xception: deep learning with depthwise separable convolutions[C], 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1800-1807, (2017)
  • [29] Liu H J, Liu F Q, Fan X Y, Et al., Polarized self-attention: Towards high- quality pixel- wise regression, (2021)
  • [30] Dai Y M, Gieseke F, Oehmcke S, Et al., Attentional feature fusion[C], 2021 IEEE Winter Conference on Applications of Computer Vision (WACV), pp. 3559-3568, (2021)