Skin lesion segmentation using an improved framework of encoder-decoder based convolutional neural network
Corresponding Author
Ranpreet Kaur
School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, Auckland, New Zealand
Correspondence
Ranpreet Kaur, School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, 55 Wellesley Street, Auckland 1010, New Zealand.
Email: [email protected]
Search for more papers by this authorHamid GholamHosseini
School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, Auckland, New Zealand
Search for more papers by this authorRoopak Sinha
School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, Auckland, New Zealand
Search for more papers by this authorCorresponding Author
Ranpreet Kaur
School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, Auckland, New Zealand
Correspondence
Ranpreet Kaur, School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, 55 Wellesley Street, Auckland 1010, New Zealand.
Email: [email protected]
Search for more papers by this authorHamid GholamHosseini
School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, Auckland, New Zealand
Search for more papers by this authorRoopak Sinha
School of Engineering, Computer & Mathematical Sciences, Auckland University of Technology, Auckland, New Zealand
Search for more papers by this authorAbstract
Automatic lesion segmentation is a key phase of skin lesion analysis that significantly increases the performance of subsequent classification steps. Segmentation is a highly complex task due to the varying nature of lesions, such as unique shapes, different colors, and structures. In this study, a two-step system is proposed comprising preprocessing algorithm and lesion segmentation network. The hairlines removal algorithm is designed using morphological operators to eliminate noise artifacts. The resulting output images are fed to the convolutional neural network (CNN) to perform lesion segmentation. The proposed CNN network is a new framework designed from scratch based on encoder-decoder architecture. The layers are stacked in a unique sequence to perform downsampling and upsampling, generating a high-resolution segmentation map. Additionally, a Tversky loss function is implemented to reduce the error rate between predicted and target output. In this study, we focus on the challenges related to the extraction of accurate lesion regions and the presence of hairlines in lesion images that can occlude important information resulting in poor segmentation. The proposed model is evaluated on four publicly available datasets, namely ISIC 2016–2018. The mean intersection over union (IoU) obtained for ISIC 2016–2018 and is 87.1%, 77.8%, 85.2%, and 88.8%, respectively. The proposed model demonstrated higher performance as compared to other state-of-the-art methods. The hair removal step aids in the improvement of the overall performance.
CONFLICT OF INTEREST
The authors declare no conflicts of interest.
Open Research
DATA AVAILABILITY STATEMENT
The publicly available skin cancer dataset is used in the proposed work. There is no conflicts in the use of data.
REFERENCES
- 1 Skin Cancer Foundation. Skin cancer facts and statistics. 2020. https://www.skincancer.org/skin-cancer-information/skin-cancer-facs/.
- 2 Cancer Australia. Melanoma of the skin. 2020. https://www.canceraustralia.gov.au/cancer-types/melanoma/statistics/.
- 3Macià F, Pumarega J, Gallén M, Porta M. Time from (clinical or certainty) diagnosis to treatment onset in cancer patients: the choice of diagnostic date strongly influences differences in therapeutic delay by tumor site and stage. J Clin Epidemiol. 2013; 66(8): 928-939.
- 4Schadendorf D, van Akkooi AC, Berking C, et al. Melanoma. The Lancet. 2018; 392(10151): 971-984.
- 5 Massey University of New Zealand. Environmental health indicators New Zealand. https://www.ehinz.ac.nz/indicators/uv-exposure/melanoma/
- 6Singh D, Gautam D, Ahmed M. Detection techniques for melanoma diagnosis: A performance evaluation. 2014 International Conference on Signal Propagation and Computer Technology (ICSPCT 2014). IEEE; 2014; 567–572.
- 7Jaglan P, Dass R, Duhan M. A comparative analysis of various image segmentation techniques. Proceedings of 2nd International Conference on Communication, Computing and Networking. Springer. 2019; 359–374.
- 8Kayalibay B, Jensen G, van der Smagt P. CNN-based segmentation of medical imaging data. arXiv preprint arXiv:1701.03056, 2017.
- 9Souza LFDF, Silva ICL, Marques AG, et al. Internet of medical things: an effective and fully automatic IoT approach using deep learning and fine-tuning to lung CT segmentation. Sensors. 2020; 20(23): 6711.
- 10Yang Z, Tan B, Pei H, Jiang W. Segmentation and multi-scale convolutional neural network-based classification of airborne laser scanner data. Sensors. 2018; 18(10): 3347.
- 11d'Acremont A, Fablet R, Baussard A, Quin G. CNN-based target recognition and identification for infrared imaging in defense systems. Sensors. 2019; 19(9): 2040.
- 12Sobhaninia Z, Rezaei S, Noroozi A, Ahmadi M, Zarrabi H, Karimi N, Emami A, Samavi S. Brain tumor segmentation using deep learning by type specific sorting of images. arXiv preprint arXiv:1809.07786. 2018.
- 13Hssayeni MD, Croock MS, Salman AD, Al-khafaji HF, Yahya ZA, Ghoraani B. Intracranial hemorrhage segmentation using a deep convolutional model. Data. 2020; 5(1): 14.
- 14Redmon J. Darknet: Open source neural networks in C. 2013–2016. http://pjreddie.com/darknet/.
- 15Salehi SSM, Erdogmus D, Gholipour A. Tversky loss function for image segmentation using 3D fully convolutional deep networks. International Workshop on Machine Learning in Medical Imaging. Springer; 2017: 379-387.
10.1007/978-3-319-67389-9_44 Google Scholar
- 16Celebi ME, Wen Q, Iyatomi H, Shimizu K, Zhou H, Schaefer G. A state-of-the-art survey on lesion border detection in dermoscopy images. Dermoscopy Image Analysis. Vol 10. CRC Press; 2015: 97-129.
10.1201/b19107-5 Google Scholar
- 17Celebi ME, Zornberg A. Automated quantification of clinically significant colors in dermoscopy images and its application to skin lesion classification. IEEE Syst J. 2014; 8(3): 980-984.
- 18Salido JA, Ruiz C JR. Hair artifact removal and skin lesion segmentation of dermoscopy images. Asian J Pharm Clin Res. 2018; 11(3): 36-39.
10.22159/ajpcr.2018.v11s3.30025 Google Scholar
- 19Lee T, Ng V, Gallagher R, Coldman A, McLean D. Dullrazor: a software approach to hair removal from images. Comput Biol Med. 1997; 27(6): 533-543.
- 20Khalid S, Jamil U, Saleem K, et al. Segmentation of skin lesion using cohen - daubechies - feauveau biorthogonal wavelet. SpringerPlus. 2016; 5(1): 1-17.
- 21Xie F-Y, Qin S-Y, Jiang Z-G, Meng R-S. PDE-based unsupervised repair of hair-occluded information in dermoscopy images of melanoma. Comput Med Imaging Graph. 2009; 33(4): 275-282.
- 22Zaqout I. An efficient block-based algorithm for hair removal in dermoscopic images. Comp Opt. 2017; 41: 521-527.
- 23Sridevi M, Mala C. A survey on monochrome image segmentation methods. Procedia Technol. 2012; 6: 548-555.
10.1016/j.protcy.2012.10.066 Google Scholar
- 24Celebi ME, Iyatomi H, Schaefer G, Stoecker WV. Lesion border detection in dermoscopy images. Comput Med Imaging Graph. 2009; 33(2): 148-153.
- 25Oliveira RB, Mercedes Filho E, Ma Z, Papa JP, Pereira AS, Tavares JMR. Computational methods for the image segmentation of pigmented skin lesions: a review. Comput Methods Programs Biomed. 2016; 131: 127-141.
- 26Moghaddam MJ, Soltanian-Zadeh H. Medical image segmentation using artificial neural networks. Artificial Neural Networks-Methodological Advances and Biomedical Applications; IntechOpen; 2011: 121-138.
- 27Guo Y, Liu Y, Georgiou T, Lew MS. A review of semantic segmentation using deep neural networks. Int J Multimed Inf Retr. 2018; 7(2): 87-93.
- 28Long J, Shelhamer E, Darrell T. Fully convolutional networks for semantic segmentation. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2015; 3431–3440.
- 29Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556. 2014.
- 30Ronneberger O, Fischer P, and Brox T. U-net: convolutional networks for biomedical image segmentation. International Conference on Medical Image Computing and Computer-Assisted Intervention. Springer; 2015; 234–241.
- 31Al-Masni MA, Al-Antari MA, Choi M-T, Han S-M, Kim T-S. Skin lesion segmentation in dermoscopy images via deep full resolution convolutional networks. Comput Methods Programs Biomed. 2018; 162: 221-231.
- 32Bi L, Kim J, Ahn E, Kumar A, Fulham M, Feng D. Dermoscopic image segmentation via multistage fully convolutional networks. IEEE Trans Biomed Eng. 2017; 64(9): 2065-2074.
- 33Liu L, Tsui YY, Mandal M. Skin lesion segmentation using deep learning with auxiliary task. J Imag. 2021; 7(4): 67.
- 34Hasan MK, Dahal L, Samarakoon PN, Tushar FI, Martí R. DSNet: Automatic dermoscopic skin lesion segmentation. Comput Biol Med. 2020; 120:103738.
- 35Tang Y, Yang F, Yuan S et al. A multi-stage framework with context information fusion structure for skin lesion segmentation. 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019). IEEE. 2019; 1407–1410.
- 36Bi L, Feng D, Fulham M, Kim J. Improving skin lesion segmentation via stacked adversarial learning. IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019). IEEE; 2019; 1100–1103.
- 37Haque IRI, Neubert J. Deep learning approaches to biomedical image segmentation. Informatics in Medicine Unlocked. Vol 18. Elsevier; 2020:100297.
- 38Badrinarayanan V, Kendall A, Cipolla R. SegNet: a deep convolutional encoder-decoder architecture for image segmentation. IEEE Trans Pattern Anal Mach Intell. 2017; 39(12): 2481-2495.
- 39Gutman D, Codella NCF, Celebi E, Helba B, Marchetti M, Mishra N, Halpern A. Skin lesion analysis toward melanoma detection: A challenge. International Symposium on Biomedical Imaging (ISBI) 2016, hosted by the International Skin Imaging Collaboration (ISIC). 2016.
- 40Codella NC, Gutman D, Celebi ME, et al. Skin lesion analysis toward melanoma detection: A challenge at the 2017 international symposium on biomedical imaging (ISBI), hosted by the international skin imaging collaboration (ISIC). IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018). IEEE; 2018; 168–172.
- 41Tschandl P, Rosendahl C, Kittler H. The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions. Sci Data. 2018; 5(1): 1-9.
- 42Codella N, Rotemberg V, Tschandl P, et al. Skin lesion analysis toward melanoma detection 2018: A challenge hosted by the international skin imaging collaboration (ISIC). arXiv preprint arXiv:1902.03368. 2019.
- 43Mendonça T, Ferreira PM, Marques JS, Marcal AR, Rozeira J.
-A dermoscopic image database for research and benchmarking. 35th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC). IEEE; 2013; 5437–5440.
- 44Brodersen KH, Ong CS, Stephan KE, Buhmann JM. The balanced accuracy and its posterior distribution. 20th International Conference on Pattern Recognition; IEEE; 2010; 3121–3124.
- 45Dice LR. Measures of the amount of ecologic association between species. Ecology. 1945; 26(3): 297-302.
- 46Jaccard P. The distribution of the flora in the alpine zone. 1. New Phytol. 1912; 11(2): 37-50.
10.1111/j.1469-8137.1912.tb05611.x Google Scholar
- 47Sanchez, U. ISBI 2016 challenge results. 2016. https://challenge.kitware.com/submission/56fe2b60cad3a55ecee8cf74
- 48Yuan Y, Lo Y-C. Improving dermoscopic image segmentation with enhanced convolutional-deconvolutional networks. IEEE J Biomed Health Inform. 2019; 23(2): 519-526. doi:10.1109/JBHI.2017.2787487
- 49Yu L, Chen H, Dou Q, Qin J, Heng P-A. Automated melanoma recognition in dermoscopy images via very deep residual networks. IEEE Trans Med Imaging. 2016; 36(4): 994-1004.
- 50Berseth M. ISIC 2017 - skin lesion analysis towards melanoma detection. ArXiv, vol. abs/1703.00523; 2017.
- 51Rahman M. ISBI 2016 challenge results. 2016. https://challenge.kitware.com/submission/56fbfa1bcad3a54f8bb809bf
- 52Bi L, Kim J, Ahn E, Feng D. Automatic skin lesion analysis using large-scale dermoscopy images and deep residual networks. arXiv preprint arXiv:1703.04197. 2017.
- 53Huang L, Zhao Y-G, Yang T-J. Skin lesion segmentation using object scale-oriented fully convolutional neural networks. Signal, Image Video Process. 2019; 13(3): 431-438.
- 54Menegola A, Tavares J, Fornaciali M, Li LT, Avila S, Valle E. Recod titans at isic challenge 2017. arXiv preprint arXiv:1703.04819. 2017.
- 55Xie F, Yang J, Liu J, Jiang Z, Zheng Y, Wang Y. Skin lesion segmentation using high-resolution convolutional neural network. Comput Methods Programs Biomed. 2020; 186:105241.
- 56Zafar K, Gilani SO, Waris A, et al. Skin lesion segmentation from dermoscopic images using convolutional neural network. Sensors. 2020; 20(6): 1601.
- 57Hasan MK, Elahi MTE, Alam MA, Jawad MT. Dermoexpert: Skin lesion classification using a hybrid convolutional neural network through segmentation, transfer learning, and augmentation. medRxiv. 2021.
- 58Lei B, Xia Z, Jiang F, et al. Skin lesion segmentation via generative adversarial networks with dual discriminators. Med Image Anal. 2020; 64:101716.
- 59Al-Masni MA, Kim D-H, Kim T-S. Multiple skin lesions diagnostics via integrated deep convolutional networks for segmentation and classification. Comput Methods Programs Biomed. 2020; 190:105351.
- 60Abhishek K, Hamarneh G, Drew MS. Illumination-based transformations improve skin lesion segmentation in dermoscopic images. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops; 2020; 728–729.
- 61Qian C, Liu T, Jiang H, Wang Z, Wang P, Guan M, Sun B. A detection and segmentation architecture for skin lesion segmentation on dermoscopy images. arXiv preprint arXiv:1809.03917. 2018.
- 62Goyal M, Oakley A, Bansal P, Dancey D, Yap MH. Skin lesion segmentation in dermoscopic images with ensemble deep learning methods. IEEE Access. 2019; 8: 4171-4181.
- 63Hao D, Seok JY, Ngiam DN, Yuan K, Feng M. Team holidayburned at ISIC challenge 2018. 2018. https://challenge.isic-archive.com/leaderboards/2018
- 64Ji Y, Li X, Zhang G, Lin D, Chen H. Automatic skin lesion segmentation by feature aggregation convolutional neural network. Technical Report. 2018.
- 65Bi L, Kim J, Ahn E, Kumar A, Feng D, Fulham M. Step-wise integration of deep class-specific learning for dermoscopic image segmentation. Pattern Recogn. 2019; 85: 78-89.
- 66Xue Y, Gong L, Peng W, Huang X, Zheng Y. “ Automatic skin lesion analysis with deep networks. 2018. https://challenge.isic-archive.com/leaderboards/2018
- 67Ali R, Hardie RC, De Silva MS, Kebede TM. Skin lesion segmentation and classification for ISIC 2018 by combining deep CNN and handcrafted features. arXiv preprint arXiv:1908.05730. 2019.