Polyp segmentation with consistency training and continuous update of pseudo-label

Sci Rep. 2022 Aug 26;12(1):14626. doi: 10.1038/s41598-022-17843-3.

Abstract

Polyp segmentation has accomplished massive triumph over the years in the field of supervised learning. However, obtaining a vast number of labeled datasets is commonly challenging in the medical domain. To solve this problem, we employ semi-supervised methods and suitably take advantage of unlabeled data to improve the performance of polyp image segmentation. First, we propose an encoder-decoder-based method well suited for the polyp with varying shape, size, and scales. Second, we utilize the teacher-student concept of training the model, where the teacher model is the student model's exponential average. Third, to leverage the unlabeled dataset, we enforce a consistency technique and force the teacher model to generate a similar output on the different perturbed versions of the given input. Finally, we propose a method that upgrades the traditional pseudo-label method by learning the model with continuous update of pseudo-label. We show the efficacy of our proposed method on different polyp datasets, and hence attaining better results in semi-supervised settings. Extensive experiments demonstrate that our proposed method can propagate the unlabeled dataset's essential information to improve performance.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Datasets as Topic / standards
  • Datasets as Topic / trends
  • Humans
  • Image Processing, Computer-Assisted
  • Polyps / diagnostic imaging
  • Polyps / pathology*
  • Supervised Machine Learning*