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

被引:3
|
作者
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
相关论文
共 50 条
  • [41] Self-supervised learning with self-distillation on COVID-19 medical image classification
    Tan, Zhiyong
    Yu, Yuhai
    Meng, Jiana
    Liu, Shuang
    Li, Wei
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2024, 243
  • [42] Class-Aware Self-Distillation for Remote Sensing Image Scene Classification
    Wu, Bin
    Hao, Siyuan
    Wang, Wei
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 2173 - 2188
  • [43] Self-supervised Learning for Astronomical Image Classification
    Martinazzo, Ana
    Espadoto, Mateus
    Hirata, Nina S. T.
    2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 4176 - 4182
  • [44] Self-supervised Learning for Sonar Image Classification
    Preciado-Grijalva, Alan
    Wehbe, Bilal
    Firvida, Miguel Bande
    Valdenegro-Toro, Matias
    2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2022, 2022, : 1498 - 1507
  • [45] Self-supervised learning with randomised layers for remote sensing
    Jung, Heechul
    Jeon, Taegyun
    ELECTRONICS LETTERS, 2021, 57 (06) : 249 - 251
  • [46] Variational Self-Distillation for Remote Sensing Scene Classification
    Hu, Yutao
    Huang, Xin
    Luo, Xiaoyan
    Han, Jungong
    Cao, Xianbin
    Zhang, Jun
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [47] Unsupervised remote sensing image scene classification based on semi-supervised learning
    Bai, Kun
    Mu, Xiaodong
    Chen, Xuebing
    Zhu, Yongqing
    You, Xuanang
    Cehui Xuebao/Acta Geodaetica et Cartographica Sinica, 2022, 51 (05): : 691 - 702
  • [48] Distill on the Go: Online knowledge distillation in self-supervised learning
    Bhat, Prashant
    Arani, Elahe
    Zonooz, Bahram
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2021, 2021, : 2672 - 2681
  • [49] Tree Species Classification Based on Self-Supervised Learning with Multisource Remote Sensing Images
    Wang, Xueliang
    Yang, Nan
    Liu, Enjun
    Gu, Wencheng
    Zhang, Jinglin
    Zhao, Shuo
    Sun, Guijiang
    Wang, Jian
    APPLIED SCIENCES-BASEL, 2023, 13 (03):
  • [50] A CNN-TRANSFORMER KNOWLEDGE DISTILLATION FOR REMOTE SENSING SCENE CLASSIFICATION
    Nabi, Mostaan
    Maggiolo, Luca
    Moser, Gabriele
    Serpico, Sebastiano B.
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 663 - 666