Improved Road Extraction Models through Semi-Supervised Learning with ACCT

Abstract

Improving the performance and reducing the training cost of road extraction models in the absence of samples is important for updating road maps. Despite the success of recent road extraction models on standard datasets, they often fail to perform when applied to new datasets or real-world scenarios where labeled samples are not available. In this paper, our focus diverges from the typical quest to pinpoint the optimal road extraction model or evaluate generalization prowess across models. Instead, we propose a method called Asymmetric Consistent Co-Training (ACCT) to train existing road extraction models faster and make them perform better in new scenarios lacking samples. ACCT uses two models with different structures and a supervision module to enhance accuracy through mutual learning. Labeled and unlabeled images are processed by both models to generate road maps from different perspectives. The supervision module ensures consistency between predictions by computing losses based on labeling status. ACCT iteratively adjusts parameters using unlabeled data, improving generalization. Empirical evaluations show that ACCT improves IoU by 2.79% to 10.26% using only 1/8 of the labeled data compared to fully supervised methods. It also reduces parameters by over 49% compared to state-of-the-art semi-supervised methods while maintaining similar accuracy. These results highlight the potential of leveraging large amounts of unlabeled data to enhance road extraction models as data acquisition technology advances.

Keywords

• road extraction
• semi-supervised learning
• feature learning
• training cost