Enhanced Skin Lesion Segmentation: DeepLabV3and U-Net with Spatial Attention Mechanisms
Accurate segmentation of skin lesions is crucial for early skin cancer detection and treatment. This study introduces a novel approach that enhances DeepLabv3 and U-N et architectures with spatial attention mechanisms to improve skin lesion segmentation. Using the ISIC dataset, the study shows that integrating attention mechanisms into these models significantly enhances their performance. DeepLabV3captures multi-scale contextual information with its atrous spatial pyramid pooling (ASPP), while U-Net excels at capturing fine-grained details through its encoder-decoder structure. The spatial attention mechanisms enable the models to focus dynamically on the most relevant image regions, improving accuracy and robustness. The models are trained on the diverse ISIC dataset, using data augmentation techniques like rotation, scaling, and color jittering to handle variability in lesion appearance and limited data size. A composite loss function balancing pixel-wise accuracy and boundary precision guides the training process. Experimental results demonstrate that the attention-enhanced DeepLabV3and U-Net models outperform their baseline versions achieving higher accuracy and Intersection over Union (IoU) scores. This study highlights the potential of attention-enhanced DeepLabV3 and U-Net models for skin lesion segmentation, suggesting they could be valuable tools for dermatologists in early skin cancer detection and treatment. The proposed models not only improve segmentation accuracy but also offer scalable solutions for clinical applications.
Authors
- Su Myat Thwin (ORCID: https://orcid.org/0009-0007-5121-8092)
- Hyun-Seok Park (ORCID: https://orcid.org/0000-0002-1237-8831)
Institutions
- Ewha Womans University (KR)
Publication Details
- Published
- 2024-10-16
- DOI
- https://doi.org/10.1109/ictc62082.2024.10827768
- Citations
- 1
- Primary Topic
- Cutaneous Melanoma Detection and Management
- Type
- article
- Field-Weighted Citation Impact
- 0.28