A Region of Interest Focused Triple UNet Architecture for Skin Lesion Segmentation

Abstract

Skin lesion segmentation is of great significance for skin lesion analysis and subsequent treatment. It is still a challenging task due to the irregular and fuzzy lesion borders, and diversity of skin lesions. In this paper, we propose Triple-UNet to automatically segment skin lesions. It is an organic combination of three UNet architectures with suitable modules. In order to concatenate the first and second sub-networks more effectively, we design a region of interest enhancement module (ROIE). The ROIE enhances the target object region of the image by using the predicted score map of the first UNet. The features learned by the first UNet and the enhanced image help the second UNet obtain a better score map. Finally, the results are fine-tuned by the third UNet. We evaluate our algorithm on a publicly available dataset of skin lesion segmentation. Experiments show that Triple-UNet outperforms the state-of-the-art on skin lesion segmentation.

15 pages, 5 figures

References in corpus (7)

• Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)
• A survey, review, and future trends of skin lesion segmentation and classification
• The Effects of Skin Lesion Segmentation on the Performance of Dermatoscopic Image Classification
• Fast, nonlocal and neural: a lightweight high quality solution to image denoising
• Multi-level Context Gating of Embedded Collective Knowledge for Medical Image Segmentation
• DoubleU-Net: A Deep Convolutional Neural Network for Medical Image Segmentation
• Joint Demosaicking and Denoising Benefits from a Two-stage Training Strategy