Remote Sensing Image Scene Classification via Self-Supervised Learning and Knowledge Distillation

被引:10
作者
Zhao, Yibo [1 ]
Liu, Jianjun [1 ]
Yang, Jinlong [1 ]
Wu, Zebin [2 ]
机构
[1] Jiangnan Univ, Sch Artificial Intelligence & Comp Sci, Jiangsu Prov Engn Lab Pattern Recognit & Computat, Wuxi 214122, Jiangsu, Peoples R China
[2] Nanjing Univ Sci & Technol, Sch Comp Sci & Engn, Nanjing 210094, Peoples R China
基金
中国国家自然科学基金;
关键词
remote sensing images; scene classification; knowledge distillation; attention; feature fusion; self-supervised learning;
D O I
10.3390/rs14194813
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
The main challenges of remote sensing image scene classification are extracting discriminative features and making full use of the training data. The current mainstream deep learning methods usually only use the hard labels of the samples, ignoring the potential soft labels and natural labels. Self-supervised learning can take full advantage of natural labels. However, it is difficult to train a self-supervised network due to the limitations of the dataset and computing resources. We propose a self-supervised knowledge distillation network (SSKDNet) to solve the aforementioned challenges. Specifically, the feature maps of the backbone are used as supervision signals, and the branch learns to restore the low-level feature maps after background masking and shuffling. The "dark knowledge" of the branch is transferred to the backbone through knowledge distillation (KD). The backbone and branch are optimized together in the KD process without independent pre-training. Moreover, we propose a feature fusion module to fuse feature maps dynamically. In general, SSKDNet can make full use of soft labels and has excellent discriminative feature extraction capabilities. Experimental results conducted on three datasets demonstrate the effectiveness of the proposed approach.
引用
收藏
页数:25
相关论文
共 72 条
  • [31] Model Inspired Autoencoder for Unsupervised Hyperspectral Image Super-Resolution
    Liu, Jianjun
    Wu, Zebin
    Xiao, Liang
    Wu, Xiao-Jun
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [32] Path Aggregation Network for Instance Segmentation
    Liu, Shu
    Qi, Lu
    Qin, Haifang
    Shi, Jianping
    Jia, Jiaya
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 8759 - 8768
  • [33] Swin Transformer: Hierarchical Vision Transformer using Shifted Windows
    Liu, Ze
    Lin, Yutong
    Cao, Yue
    Hu, Han
    Wei, Yixuan
    Zhang, Zheng
    Lin, Stephen
    Guo, Baining
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9992 - 10002
  • [34] Very High Resolution Multiangle Urban Classification Analysis
    Longbotham, Nathan
    Chaapel, Chuck
    Bleiler, Laurence
    Padwick, Chris
    Emery, William J.
    Pacifici, Fabio
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2012, 50 (04): : 1155 - 1170
  • [35] Loshchilov Ilya, 2017, CoRR, abs/1711.05101
  • [36] Distinctive image features from scale-invariant keypoints
    Lowe, DG
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2004, 60 (02) : 91 - 110
  • [37] SCViT: A Spatial-Channel Feature Preserving Vision Transformer for Remote Sensing Image Scene Classification
    Lv, Pengyuan
    Wu, Wenjun
    Zhong, Yanfei
    Du, Fang
    Zhang, Liangpei
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [38] Unsupervised Adversarial Domain Adaptation with Error-Correcting Boundaries and Feature Adaption Metric for Remote-Sensing Scene Classification
    Ma, Chenhui
    Sha, Dexuan
    Mu, Xiaodong
    [J]. REMOTE SENSING, 2021, 13 (07)
  • [39] Maggiori E, 2017, INT GEOSCI REMOTE SE, P3226, DOI 10.1109/IGARSS.2017.8127684
  • [40] Segment Optimization and Data-Driven Thresholding for Knowledge-Based Landslide Detection by Object-Based Image Analysis
    Martha, Tapas Ranjan
    Kerle, Norman
    van Westen, Cees J.
    Jetten, Victor
    Kumar, K. Vinod
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2011, 49 (12): : 4928 - 4943