Skin lesion segmentation with attention-based SC-Conv U-Net and feature map distortion

Yu Wang,Shengsheng Wang
DOI: https://doi.org/10.1007/s11760-021-02100-3
2022-01-18
Abstract:Automatic skin lesion segmentation remains a challenging task due to particular characteristics of the patient and various artifacts. In this paper, a novel network named attention-based SC-Conv U-net with stepwise multi-scale input was proposed, which effectively solves the rough boundary problem under interference and the inaccuracy segmentation of abnormally sized skin lesions. It is a symmetrical encoder-and-decoder structure. In different layers of encoder, original image information of different resolution was gradually added to the network. Subsequently, successive self-calibrated convolutions module (namely SC-Conv) captures context with broader respective field, which ensures the maximum effective information to be extracted and reduces the interference of irrelevant information. In the decoder, we employ the attention-based skip connection to fuse inter-level features, which can extract multiple delicate local representations and retain more precise spatial information. It is key to guide the network in adapting to the segmentation of lesions in different scales. Meanwhile, the feature map distortion mechanism is adopted to further improve network performance with better generalization ability. As shown by the result, our approach outperforms the state-of-the-art, especially for JA (0.7828), SE (0.8961) and DI (0.8864) scores which are significantly increased by 5.3%, 11.3% and 6.2%, respectively, compared with the U-net on ISIC2017.
What problem does this paper attempt to address?