Paper ID | BIO-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 | ||
Session | BIO-6: Medical Image Segmentation | ||
Location | Gather.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 | ||
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. |