AdaptMatch: Adaptive Matching for Semisupervised Binary Segmentation of Remote Sensing Images

Wei Huang, Yilei Shi, Zhitong Xiong, Xiao Xiang Zhu

Research output: Contribution to journalArticlepeer-review

12 Scopus citations

Abstract

There are various binary semantic segmentation tasks in remote sensing (RS) that aim to extract the foreground areas of interest, such as buildings and roads, from the background in satellite images. In particular, semisupervised learning (SSL), which can use limited labeled data to guide a large amount of unlabeled data for model training, can significantly promote the fast applications of these tasks in practice. However, due to the predominance of the background in RS images, the foreground only accounts for a small proportion of the pixels. It poses a challenge: models are biased toward the majority class of the background, leading to poor performance on the minority class of the foreground. To address this issue, this article proposes a novel and effective SSL framework, adaptive matching (AdaptMatch), for RS binary segmentation. AdaptMatch calculates individual and adaptive thresholds of the foreground and background based on their convergence difficulty in an online manner at the training stage; the adaptive thresholds are then used to select the high-confidence pseudo-labeled data of the two classes for model self-training in turn. Extensive experiments are conducted on two widely studied RS binary segmentation tasks, building footprint extraction and road extraction, to demonstrate the effectiveness and generalizability of the proposed method. The results show that the proposed AdaptMatch achieves superior performance compared with some state-of-the-art semisupervised methods in RS binary segmentation tasks. The codes will be publicly available at https://github.com/zhu-xlab/AdaptMatch.

Original languageEnglish
Article number5625416
Pages (from-to)1-16
Number of pages16
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume61
DOIs
StatePublished - 2023

Keywords

  • Adaptive threshold
  • binary segmentation
  • building footprint extraction
  • remote sensing (RS)
  • road extraction
  • semisupervised learning (SSL)

Fingerprint

Dive into the research topics of 'AdaptMatch: Adaptive Matching for Semisupervised Binary Segmentation of Remote Sensing Images'. Together they form a unique fingerprint.

Cite this