Lightweight Remote Sensing Scene Classification Based on Knowledge Distillation

Chong-Yang Zhang,Bin Wang
DOI: https://doi.org/10.11972/j.issn.1001-9014.2024.05.012
2024-01-01
JOURNAL OF INFRARED AND MILLIMETER WAVES
Abstract:Remote sensing image scene classification aims to automatically assign a semantic label to each remote sens- ing image according to its content, and has become one of the hot topics in the field of remote sensing image process- ing. Methods based on convolutional neural networks (CNNs) and methods based on self-attention mechanism are two mainstream methods in remote sensing image scene classification. However, the former is less effective in exploring long-range contextual information, and the latter has limitations in learning local information and has a large number of parameters and calculations. In order to address these issues, a lightweight method based on knowledge distillation is proposed to solve the problem of scene classification for remote sensing images. The proposed method uses Swin Trans- former and lightweight CNNs as the teacher model and the student models, respectively, and integrates the advantages of the two kinds of models by means of knowledge distillation. Furthermore, a novel distillation loss function is pro- posed to enable the student models to focus on both inter- and intra-class potential information of remote sensing images simultaneously. The experimental results on two large-scale remote sensing image datasets demonstrate that the pro- posed method not only achieves high classification accuracy compared to existing methods but also has a significantly re- duced number of parameters and calculations.
What problem does this paper attempt to address?