A Multi-Stage Framework With Context Information Fusion Structure For Skin Lesion Segmentation

Yujiao Tang,Feng Yang,Shaofeng Yuan,Chang'an Zhan
DOI: https://doi.org/10.1109/isbi.2019.8759535
2019-04-01
Abstract:The computer-aided diagnosis (CAD) systems can highly improve the reliability and efficiency of melanoma recognition. As a crucial step of CAD, skin lesion segmentation has the unsatisfactory accuracy in existing methods due to large variability in lesion appearance and artifacts. In this work, we propose a framework employing multi-stage UNets (MS-UNet) in the auto-context scheme to segment skin lesion accurately end-to-end. We apply two approaches to boost the performance of MS-UNet. First, UNet is coupled with a context information fusion structure (CIFS) to integrate the low-level and context information in the multi-scale feature space. Second, to alleviate the gradient vanishing problem, we use deep supervision mechanism through supervising MS-UNet by minimizing a weighted Jaccard distance loss function. Three out of five commonly used performance metrics, including Jaccard index and Dice coefficient, show that our approach outperforms the state-of the-art deep learning based methods on the ISBI 2016 Skin Lesion Challenge dataset.
What problem does this paper attempt to address?