Satellite Optical Target Edge Detection Based on Knowledge Distillation

Edge detection of space targets is vital in aerospace applications, such as satellite monitoring and analysis, yet it faces challenges due to diverse target shapes and complex backgrounds. While deep learning-based edge detection methods dominate due to their powerful feature representation capabilities, they often suffer from large parameter sizes and lack explicit geometric prior constraints for space targets. This paper proposes a novel edge detection method for satellite targets based on knowledge distillation, namely STED-KD. Firstly, a multi-stage distillation strategy is proposed to guide a lightweight, fully convolutional network with fewer parameters to learn key features and decision boundaries from a complex teacher model, achieving model efficiency. Next, a shape prior guidance module is integrated into the student branch, incorporating geometric shape information through shape prior model construction, similarity metric calculation, and feature reconstruction, enhancing adaptability to space targets and improving detection accuracy. Additionally, a curvature-guided edge loss function is designed to ensure continuous and complete edges, minimizing local discontinuities. Experimental results on the UESD space target dataset demonstrate superior performance, with ODS, OIS, and AP scores of 0.659, 0.715, and 0.596, respectively. On the BSDS500, STED-KD achieves ODS, OIS, and AP scores of 0.818, 0.829, and 0.850, respectively, demonstrating strong competitiveness and further confirming its stability.