Two-Way Assistant: A Knowledge Distillation Object Detection Method for Remote Sensing Images

Xi Yang,Sheng Zhang,Weichao Yang
DOI: https://doi.org/10.1109/tgrs.2024.3371681
IF: 8.2
2024-03-12
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Due to resource constraints on edge devices, lightweight detection models are gaining popularity in remote sensing. However, achieving efficient performance with these models is challenging compared with traditional detection models. Knowledge distillation (KD) is a promising solution to this issue. However, previous KD methods in remote sensing often suffer from background noise and fail to address feature disparities among detectors. To address the above issues, we introduce the two-way assistant (TWA) distillation method for remote sensing object detection. TWA comprises two crucial modules: the compression assistant module (CPAM) and the multiscale adaptive assistant module (MAAM). CPAM reduces the background information and category interference by compressing and redistributing teacher model features to the student model. MAAM enhances feature knowledge through multiscale fusion, addressing feature disparities. Through extensive experiments conducted on two distinct types of remote sensing datasets, optical learning, vision and remote sensing laboratory (LEVIR) and SAR SSDD, our TWA demonstrates favorable performance across both single-stage and two-stage detectors. Especially, it achieves a performance of 82.5% (optical LEVIR dataset) and 95.4% (SAR SSDD dataset) in the AP50 metric, superior to existing state-of-the-art methods.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?