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

被引:9
|
作者
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] CMID: A Unified Self-Supervised Learning Framework for Remote Sensing Image Understanding
    Muhtar, Dilxat
    Zhang, Xueliang
    Xiao, Pengfeng
    Li, Zhenshi
    Gu, Feng
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [42] TOV: The Original Vision Model for Optical Remote Sensing Image Understanding via Self-Supervised Learning
    Tao, Chao
    Qi, Ji
    Zhang, Guo
    Zhu, Qing
    Lu, Weipeng
    Li, Haifeng
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 4916 - 4930
  • [43] 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
  • [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 for Astronomical Image Classification
    Martinazzo, Ana
    Espadoto, Mateus
    Hirata, Nina S. T.
    2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 4176 - 4182
  • [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] Self-supervised learning with randomised layers for remote sensing
    Jung, Heechul
    Jeon, Taegyun
    ELECTRONICS LETTERS, 2021, 57 (06) : 249 - 251
  • [48] 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
  • [49] An Inverted Residual Cross Head Knowledge Distillation Network for Remote Sensing Scene Image Classification
    Shi, Cuiping
    Ding, Mengxiang
    Wang, Liguo
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2025, 18 : 4881 - 4894
  • [50] 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