2021 IEEE International Conference on Acoustics, Speech and Signal Processing

6-11 June 2021 • Toronto, Ontario, Canada

Extracting Knowledge from Information

2021 IEEE International Conference on Acoustics, Speech and Signal Processing

6-11 June 2021 • Toronto, Ontario, Canada

Extracting Knowledge from Information

Technical Program

Paper Detail

Paper IDBIO-6.4
Paper Title SEMI-SUPERVISED SKIN LESION SEGMENTATION WITH LEARNING MODEL CONFIDENCE
Authors Zhiqiang Xie, Enmei Tu, Hao Zheng, Yun Gu, Jie Yang, Institute of Image Processing and Pattern Recognition, Shanghai Jiao Tong University, China
SessionBIO-6: Medical Image Segmentation
LocationGather.Town
Session Time:Wednesday, 09 June, 13:00 - 13:45
Presentation Time:Wednesday, 09 June, 13:00 - 13:45
Presentation Poster
Topic Biomedical Imaging and Signal Processing: [BIO-MIA] Medical image analysis
IEEE Xplore Open Preview  Click here to view in IEEE Xplore
Virtual Presentation  Click here to watch in the Virtual Conference
Abstract Segmentation of skin lesions is important for disease diagnoses and treatment planning. Over the years, semi-supervised methods using pseudo labels have boosted the segmentation performance with limited labeled data and abundant unlabeled data. However, the unreliable targets in pseudo labels might lead to meaningless guidance for unlabeled data. In this paper, to solve this issue, we propose a novel confidence aware semi-supervised learning method based on a mean teacher scheme. Concretely, we design a confidence module to predict the model confidence guided by the True Class Probability. Then in the mean teacher framework, the student model gradually learns trustworthy targets from teacher model. To further improve the segmentation quality, we fine-tune the student model with reliable content in pseudo labels. We conduct extensive experiments on 2018 ISIC skin lesion segmentation dataset and our method outperforms other state-of-the-art semi-supervised approaches.