Self-Supervised Feature Learning Method for Hyperspectral Images Based on Mixed Convolutional Networks

被引:0
|
作者
Feng, Fan [1 ]
Zhang, Yongsheng [1 ]
Zhang, Jin [1 ]
Liu, Bing [2 ]
Yu, Ying [1 ]
机构
[1] PLA Strateg Support Force Informat Engn Univ, Inst Geospatial Informat, Zhengzhou 450001, Henan, Peoples R China
[2] PLA Strateg Support Force Informat Engn Univ, Inst Data & Target Engn, Zhengzhou 450001, Henan, Peoples R China
关键词
hyperspectral image classification; self- supervised learning; contrastive learning; mixed convolutional network; second- order pooling;
D O I
10.3788/AOS231776
中图分类号
O43 [光学];
学科分类号
070207 ; 0803 ;
摘要
Objective Hyperspectral images record the reflectance of ground objects in hundreds of narrow bands, forming a unified three-dimensional data cube. Accurate hyperspectral image classification results exhibit a detailed distribution of ground objects, making it the cornerstone of many remote sensing applications. Recently, hyperspectral images with high spatial resolution have promoted the application of hyperspectral technology in various fine-grained tasks. Since hyperspectral images feature high nonlinearity, feature extraction serves as a key to accurate classification. Learning robust spatial- spectral features in real- world complex scenes with insufficient labeled samples has been a long-standing problem. We propose a self- supervised feature learning method for hyperspectral images based on mixed convolutional networks and contrastive learning. This method can make full use of abundant spatial- spectral information in hyperspectral images and automatically learn to extract features suitable for classification tasks in a self- supervised manner. We hope that our findings can help the study of small sample hyperspectral classification, and promote the generalization and practicability of deep learning methods in complex hyperspectral scenes. Methods We propose a self- supervised mixed feature fusion network, which is based on mixed convolutional networks and contrastive learning. Firstly, the dimensionality of hyperspectral images is reduced by a factor analysis (FA) algorithm, and the neighborhood information of image pixels is extracted to form image patches. Positive and negative sample pairs are then generated through random spatial and spectral augmentation. Secondly, an efficient cascade feature fusion encoder is constructed by 3D convolution layers and 2D depth- separable convolutional layers. Multi- scale spatial- spectral features are extracted and fine-grained embeddings are calculated by a second- order pooling (SOP) layer. By calculating the contrastive loss on the extracted features for positive and negative sample pairs, the encoder can be trained in a self- supervised manner. Finally, the trained encoder will be fine-tuned using few labeled samples, producing the classification results of hyperspectral images. Results and Discussions To validate the proposed method, extensive experiments are conducted on four hyperspectral datasets with distinct spatial- spectral features, namely Indian Pines, Houston, Longkou, and Hanchuan. Indian Pines and Houston are conventional hyperspectral datasets for algorithm verification. Longkou and Hanchuan are recently released datasets that feature extremely high spatial resolution. The contrast methods include attention- based methods, transformer- based methods, and the contrastive learning method that have been proposed recently. Only five supervised samples from each type of ground object are utilized for fine-tuning, and the overall accuracy of the proposed method stands at 79.46 degrees o, 84.32 degrees o, 92.97 degrees o, and 82.31 degrees o, respectively, which outperform the above contrast methods (Tables 2-5). The classification maps of the four datasets also demonstrate fewer misclassifications of this method (Figs. 3 -6). Targeted ablation experiments are carried out with the results confirming the efficacy of FA, SOP, and contrastive learning method designed in this paper (Table 6). Further experiments on contrastive learning- related settings reveal three key points. First, spatial and spectral enhancement is indispensable. Second, the batch normalization (BN) layer in the projection head plays a crucial role in contrastive learning. Third, the full-finetune approach is more suitable than the linear probe method in hyperspectral image classification tasks (Table 7). Additionally, operational efficiency has been considered and the proposed method can realize the balance between classification accuracy and operation efficiency (Table 8). Conclusions We propose a self- supervised classification framework for hyperspectral image classification based on mixed convolutional networks and contrastive learning. Our method combines self- supervised pretext task design and encoder design. The abundant spatial- spectral information of hyperspectral images can be systematically investigated and the features suitable for classification tasks can be extracted in a self- supervised manner. Firstly, spatial and spectral enhancement is used to add random perturbations to hyperspectral image patches, forming positive and negative sample pairs. Then, a mixed convolutional network- based encoder is utilized to extract multi- scale features. The mixed convolutional network consists of a cascade feature fusion structure and a SOP layer, which can extract robust fine-grained spatial- spectral features from disturbed sample pairs. Lastly, the contrastive loss is calculated using the extracted features, enabling the encoder parameters to be optimized in a self- supervised way. Experiments are carried out on four hyperspectral datasets with distinct differences in spatial- spectral features. The classification accuracy of the proposed method is superior to those of contrast methods, and the ablation experimental results show the effectiveness of FA, SOP, and the proposed contrastive learning method. In addition, this method is designed to reduce parameter redundancy and improve parameter utilization efficiency for a balance between operating efficiency and classification accuracy. We explore the combination of model design and self- supervised learning. In the future, we hope that the proposed method will be used in various hyperspectral datasets and it will be further improved for greater generalization ability.
引用
收藏
页数:12
相关论文
共 32 条
  • [1] Deep learning techniques to classify agricultural crops through UAV imagery: a review
    Bouguettaya, Abdelmalek
    Zarzour, Hafed
    Kechida, Ahmed
    Taberkit, Amine Mohammed
    [J]. NEURAL COMPUTING & APPLICATIONS, 2022, 34 (12): : 9511 - 9536
  • [2] Chen T, 2020, PR MACH LEARN RES, V119
  • [3] Exploring Simple Siamese Representation Learning
    Chen, Xinlei
    He, Kaiming
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 15745 - 15753
  • [4] Lightweight Residual Network Based on Depthwise Separable Convolution for Hyperspectral Image Classification
    Cheng Rongjie
    Yang Yun
    Li Longwei
    Wang Yanting
    Wang Jiayu
    [J]. ACTA OPTICA SINICA, 2023, 43 (12)
  • [5] Small Sample Hyperspectral Image Classification Based on Cascade Fusion of Mixed Spatial-Spectral Features and Second-Order Pooling
    Feng, Fan
    Zhang, Yongsheng
    Zhang, Jin
    Liu, Bing
    [J]. REMOTE SENSING, 2022, 14 (03)
  • [6] Hyperspectral Images Classification Based on Multi-Feature Fusion and Hybrid Convolutional Neural Networks
    Feng Fan
    Wang Shuangting
    Zhang Jin
    Wang Chunyang
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (08)
  • [7] [杭仁龙 Hang Renlong], 2023, [测绘学报, Acta Geodetica et Cartographica Sinica], V52, P1164
  • [8] Masked Autoencoders Are Scalable Vision Learners
    He, Kaiming
    Chen, Xinlei
    Xie, Saining
    Li, Yanghao
    Dollar, Piotr
    Girshick, Ross
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 15979 - 15988
  • [9] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [10] Hyperspectral Imagery Classification Based on Contrastive Learning
    Hou, Sikang
    Shi, Hongye
    Cao, Xianghai
    Zhang, Xiaohua
    Jiao, Licheng
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60