Dynamic prototypical feature representation learning framework for semi-supervised skin lesion segmentation

Zhenxi Zhang, Chunna Tian, Xinbo Gao, Cui Wang, Xue Feng, Harrison X. Bai, Zhicheng Jiao

Research output: Contribution to journalArticlepeer-review

8 Citations (Scopus)

Abstract

Automated skin lesion segmentation is an essential yet challenging task for computer-aided skin disease diagnosis. One major challenge for learning-based segmentation method is the limited manually annotated dermoscopy images. Many semi-supervised methods are proposed to exploit unlabeled data by self-training with pseudo labels. However, the plain pseudo labels are less accurate and the pixel-wise features of unlabeled data are always not well formulated due to the large variations among different lesions. Aiming at producing a good segmentation embedding space in a semi-supervised manner, in this paper, we propose a novel dynamic prototypical feature representation learning framework to address these problems. Specifically, we propose a novel denoised pseudo label generation method, which effectively filters out the unreliable components in plaint pseudo labels and provides the guidance for the subsequent feature representation learning. Then, we propose a memory relation learning method to enhance the intermediate feature representation globally. Additionally, we propose a prototype-based confidence-aware contrastive learning method to learn a better local feature structure in semi-supervised training, strengthening intra-class compactness and inter-class separability. Extensive experiments on two skin lesion segmentation datasets demonstrate that our method outperforms other popular semi-supervised segmentation methods.

Original languageEnglish
Pages (from-to)369-382
Number of pages14
JournalNeurocomputing
Volume507
DOIs
Publication statusPublished - 1 Oct 2022
Externally publishedYes

Keywords

  • Contrastive learning
  • Memory relational learning
  • Semi-supervised learning
  • Skin lesion segmentation

Fingerprint

Dive into the research topics of 'Dynamic prototypical feature representation learning framework for semi-supervised skin lesion segmentation'. Together they form a unique fingerprint.

Cite this