Noisy Supervised Deep Learning for Remote Sensing Image Segmentation Using Electronic Maps

Qing Zhang,Shulin Pan,Fan Min,Yinghe Wu,Zilin Zhang,Haoran Luo
DOI: https://doi.org/10.1109/lgrs.2023.3285600
IF: 5.343
2023-06-30
IEEE Geoscience and Remote Sensing Letters
Abstract:Deep learning has made substantial progress in remote sensing image segmentation tasks. It usually requires a large number of high-quality annotation maps (i.e., clean labels), which are labor-intensive. In this letter, we propose to use abundant electronic maps (i.e., noisy labels) to supplement a small number of clean labels to solve the problem of massive label production. In addition, we propose a multistage noise supervised framework to prevent noise from deteriorating the performance of deep model. Noise-supervised deep learning framework (NSDI) consists of clean training, weight initialization, and hybrid training stages. In the clean training stage, we train a segmentation model using a small amount of clean labels and remote sensing images to compute the confusion probability matrix. In the weight initialization stage, we use the confusion probability matrix and the prediction probability to calculate the label reliability of the electronic map. In the hybrid training stage, an adaptive weighted loss function based on cross-entropy is used to dynamically update the label reliability. Then we train the model further using electronic maps with the support of the adaptive weighted loss function and label reliability. Experiments were undertaken on 2656 images of 512 × 512 pixels. Ablation studies show that NSDI improves the model robustness and the segmentation quality.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?