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 条
  • [1] Remote Sensing Image Scene Classification Using Multiscale Feature Fusion Covariance Network With Octave Convolution
    Bai, Lin
    Liu, Qingxin
    Li, Cuiling
    Ye, Zhen
    Hui, Meng
    Jia, Xiuping
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [2] Vision Transformers for Remote Sensing Image Classification
    Bazi, Yakoub
    Bashmal, Laila
    Rahhal, Mohamad M. Al
    Dayil, Reham Al
    Ajlan, Naif Al
    [J]. REMOTE SENSING, 2021, 13 (03) : 1 - 20
  • [3] Change Detection in Optical Aerial Images by a Multilayer Conditional Mixed Markov Model
    Benedek, Csaba
    Sziranyi, Tamas
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2009, 47 (10): : 3416 - 3430
  • [4] Local Semantic Enhanced ConvNet for Aerial Scene Recognition
    Bi, Qi
    Qin, Kun
    Zhang, Han
    Xia, Gui-Song
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 : 6498 - 6511
  • [5] Training Small Networks for Scene Classification of Remote Sensing Images via Knowledge Distillation
    Chen, Guanzhou
    Zhang, Xiaodong
    Tan, Xiaoliang
    Cheng, Yufeng
    Dai, Fan
    Zhu, Kun
    Gong, Yuanfu
    Wang, Qing
    [J]. REMOTE SENSING, 2018, 10 (05)
  • [6] Remote Sensing Image Scene Classification Meets Deep Learning: Challenges, Methods, Benchmarks, and Opportunities
    Cheng, Gong
    Xie, Xingxing
    Han, Junwei
    Guo, Lei
    Xia, Gui-Song
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2020, 13 : 3735 - 3756
  • [7] Remote Sensing Image Scene Classification: Benchmark and State of the Art
    Cheng, Gong
    Han, Junwei
    Lu, Xiaoqiang
    [J]. PROCEEDINGS OF THE IEEE, 2017, 105 (10) : 1865 - 1883
  • [8] Learning Rotation-Invariant Convolutional Neural Networks for Object Detection in VHR Optical Remote Sensing Images
    Cheng, Gong
    Zhou, Peicheng
    Han, Junwei
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (12): : 7405 - 7415
  • [9] Histograms of oriented gradients for human detection
    Dalal, N
    Triggs, B
    [J]. 2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, : 886 - 893
  • [10] Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171