journal article·2024Open access

Weakly Supervised Network with Scribble-Supervised and Edge-Mask for Road Extraction from High-Resolution Remote Sensing Images

Supeng Yu, Fen Huang, Chengcheng Fan

Computers, Materials & Continua· Volume 79· Issue 1

Abstract

Significant advancements have been achieved in road surface extraction based on high-resolution remote sensing image processing. Most current methods rely on fully supervised learning, which necessitates enormous human effort to label the image. Within this field, other research endeavors utilize weakly supervised methods. These approaches aim to reduce the expenses associated with annotation by leveraging sparsely annotated data, such as scribbles. This paper presents a novel technique called a weakly supervised network using scribble-supervised and edge-mask (WSSE-net). This network is a three-branch network architecture, whereby each branch is equipped with a distinct decoder module dedicated to road extraction tasks. One of the branches is dedicated to generating edge masks using edge detection algorithms and optimizing road edge details. The other two branches supervise the model’s training by employing scribble labels and spreading scribble information throughout the image. To address the historical flaw that created pseudo-labels that are not updated with network training, we use mixup to blend prediction results dynamically and continually update new pseudo-labels to steer network training. Our solution demonstrates efficient operation by simultaneously considering both edge-mask aid and dynamic pseudo-label support. The studies are conducted on three separate road datasets, which consist primarily of high-resolution remote-sensing satellite photos and drone images. The experimental findings suggest that our methodology performs better than advanced scribble-supervised approaches and specific traditional fully supervised methods.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Automated Road and Building Extraction · Remote Sensing and LiDAR Applications · Video Surveillance and Tracking Methods

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

References (28)

Show source-provided references
1. CDANet: Contextual detail-aware network for high-spatial-resolution remote-sensing imagery shadow detection10.1109/TGRS.2022.3230829 ↗
2. DSMSA-Net: Deep spatial and multi-scale attention network for road extraction in high spatial resolution satellite images10.1007/s13369-022-07082-z ↗
3. A. Bearman, O. Russakovsky, V. Ferrari, and L. Fei-Fei, “What’s the point: Semantic segmentation with point supervision,” arXiv preprint arXiv:1506.02106, 2015.10.1007/978-3-319-46478-7_34 ↗
4. ScribbleSup: Scribble-supervised convolutional networks for semantic segmentation
5. A. Khoreva, R. Benenson, J. Hosang, M. Hein, and B. Schiele, “Simple does it: Weakly supervised instance and semantic segmentation,” arXiv preprint arXiv:1603.07485, 2016.10.1109/CVPR.2017.181 ↗
6. J. Ahn and S. Kwak, “Learning pixel-level semantic affinity with image-level supervision for weakly supervised semantic segmentation,” arXiv preprint arXiv:1803.10464, 2018.10.1109/CVPR.2018.00523 ↗
7. GrabCut: Interactive foreground extraction using iterated graph cuts
8. Fully convolutional networks for semantic segmentation
9. ATSO: Asynchronous teacher-student optimization for semi-supervised image segmentation
10. K. Sohn et al., “FixMatch: Simplifying semi-supervised learning with consistency and confidence,” arXiv preprint arXiv:2001.07685, 2001.
11. H. Zhang, M. Cisse, Y. N. Dauphin, and D. Lopez-Paz, “mixup: Beyond empirical risk minimization,” arXiv preprint arXiv:1710.09412, 2017.
12. Scribble-based weakly supervised deep learning for road surface extraction from remote sensing images
13. Holistically-nested edge detection
14. Contour detection and hierarchical image segmentation10.1109/TPAMI.2010.161 ↗
15. Dropout: A simple way to prevent neural networks from overfitting
16. O. Ronneberger, P. Fischer, and T. Brox, “U-Net: Convolutional networks for biomedical image segmentation,” arXiv preprint arXiv:1505.04597, 2015.10.1007/978-3-319-24574-4_28 ↗
17. L. C. Chen, Y. Zhu, G. Papandreou, F. Schroff, and H. Adam, “Encoder-decoder with atrous separable convolution for semantic image segmentation,” arXiv preprint arXiv:1802.02611, 2018.10.1007/978-3-030-01234-2_49 ↗
18. Building extraction from high spatial resolution remote sensing images via multiscale-aware and segmentation-prior conditional random fields10.3390/rs12233983 ↗
19. A global context-aware and batch-independent network for road extraction from VHR satellite imagery10.1016/j.isprsjprs.2021.03.016 ↗
20. Learning aerial image segmentation from online maps10.1109/TGRS.2017.2719738 ↗
21. J. Dai, K. He, and J. Sun, “BoxSup: Exploiting bounding boxes to supervise convolutional networks for semantic segmentation,” arXiv preprint arXiv:1503.01640, 2015.10.1109/ICCV.2015.191 ↗
22. Boundary perception guidance: A scribble-supervised semantic segmentation approach
23. J. Zhang, X. Yu, A. Li, P. Song, B. Liu, and Y. Dai, “Weakly-supervised salient object detection via scribble annotations,” arXiv preprint arXiv:2003.07685, 2020.10.1109/CVPR42600.2020.01256 ↗
24. Road extraction from very high resolution images using weakly labeled openstreetmap centerline10.3390/ijgi8110478 ↗
25. SegNet: A deep convolutional encoder-decoder architecture for image segmentation10.1109/TPAMI.2016.2644615 ↗
26. D-LinkNet: LinkNet with pretrained encoder and dilated convolution for high resolution satellite imagery road extraction
27. A computational approach to edge detection10.1109/TPAMI.1986.4767851 ↗
28. G. Bertasius, J. Shi, and L. Torresani, “DeepEdge: A multi-scale bifurcated deep network for top-down contour detection,” arXiv preprint arXiv:1412.1123, 2014.10.1109/CVPR.2015.7299067 ↗