Progressive Weighted Self-training Ensemble for Multi-Type Skin Lesion Semantic Segmentation

• Published in: IEEE access Vol. 10; p. 1
• Main Authors: Lee, Cheolwon, Yoo, Sangwook, Kim, Semin, Lee, Jongha
• Format: Journal Article
• Language: English
• Published: Piscataway IEEE 2022; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Artificial intelligence; Data models; Deep learning; Ensemble; Lesions; Mathematical models; Multi-type Skin Lesion; Semantic Segmentation; Semantics; Semi-supervised learning; Skin; Training; Uncertainty
• ISSN: 2169-3536; 2169-3536
• DOI: 10.1109/ACCESS.2022.3222788

In this study, we propose the Progressive Weighted Self-training Ensemble (PWStE) method that reinforces efficiency of labeled data for multi-type skin lesion semantic segmentation. The generation of multi-type skin lesion labeled data is extremely expensive as it should only be performed by dermatologists due to the small pixel variations and irregularly shaped lesion characteristics. For the reason, the reality is that labeled data for skin lesion segmentation model training is absolutely insufficient. The core idea of the proposed PWStE method is to minimize the transfer of uncertainty in the training phase of general SSL by progressively using the pseudo-labeled data referenced in training. The PWStE uses procedures such as Progressive Selector, Ensemble, and Pseudo Labeler designed using conventional Semi-Supervised Learning (SSL) concepts to more accurately generate detailed features of skin lesions from unlabeled data to pseudo-labeled data. We performed ensembles using a combination of models (U-Net, FPN, LinkerNet, PSPNet) and backbones (ResNet50, EfficientNet-b3, InceptionV3, DenseNet121, SE-ResNet101, SE-ResNeXt101). Validation was performed on our Multi-Type Skin Lesion Label Database (MSLD) dataset compared to conventional SSL methods. The experiments have shown that the model trained with PWStE shows similar results to the model of trained the entire label data using the Supervised Learning (SL) method, even with 30% less label data. These results show that our proposed PWStE can increase the efficiency of the given labeled data even in the multi-type skin lesion field.