A Novel Method for Medical Image Segmentation based on Convolutional Neural Networks with SGD Optimization

Taheri, M.; Rastgarpour, M.; Koochari, A.

doi:10.22061/jecei.2020.7404.390

Document Type : Original Research Paper

Authors

¹ Department of Computer Engineering, Faculty of Engineering, Islamic Azad University, Science and Research Branch, Tehran, Iran.

² Department of Computer Engineering, Faculty of Engineering, Islamic Azad University, Saveh Branch, Saveh, Iran.

https://doi.org/10.22061/jecei.2020.7404.390

Abstract

Background and Objectives: medical image Segmentation is a challenging task due to low contrast between Region of Interest and other textures, hair artifacts in dermoscopic medical images, illumination variations in images like Chest-Xray and various imaging acquisition conditions.
Methods: In this paper, we have utilized a novel method based on Convolutional Neural Networks (CNN) for medical image Segmentation and finally, compared our results with two famous architectures, include U-net and FCN neural networks. For loss functions, we have utilized both Jaccard distance and Binary-crossentropy and the optimization algorithm that has used in this method is SGD+Nestrov algorithm. In this method, we have used two preprocessing include resizing image’s dimensions for increasing the speed of our process and Image augmentation for improving the results of our network. Finally, we have implemented threshold technique as postprocessing on the outputs of neural network to improve the contrast of images. We have implemented our model on the famous publicly, PH2 Database, toward Melanoma lesion segmentation and chest Xray images because as we have mentioned, these two types of medical images contain hair artifacts and illumination variations and we are going to show the robustness of our method for segmenting these images and compare it with the other methods.
Results: Experimental results showed that this method could outperformed two other famous architectures, include Unet and FCN convolutional neural networks. Additionally, we could improve the performance metrics that have used in dermoscopic and Chest-Xray segmentation which used before.
Conclusion: In this work, we have proposed an encoder-decoder framework based on deep convolutional neural networks for medical image segmentation on dermoscopic and Chest-Xray medical images. Two techniques of image augmentation, image rotation and horizontal flipping on the training dataset are performed before feeding it to the network for training. The predictions produced from the model on test images were postprocessed using the threshold technique to remove the blurry boundaries around the predicted lesions.

Keywords

20.1001.1.23223952.2021.9.1.4.0

Main Subjects

Computer Vision

Open Access

This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit: http://creativecommons.org/licenses/by/4.0/

Publisher’s Note

JECEI Publisher remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Publisher

Shahid Rajaee Teacher Training University

References

[1] J.U. Kim, H.G. Kim, Y.M. Ro, “Iterative deep convolutional encoder-decoder network for medical image segmentation,” presented at the 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Seogwipo, South Korea, 2017.

[2] Y. Chang et al., “Automatic segmentation and cardiopathy classification in cardiac MRI images based on deep neural networks,” presented at the 2018 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Calgary, AB, Canada, 2018.

[3] A. Işın, C. Direkoğlu, M. Şah, “Review of MRI-based brain tumor image segmentation using deep learning methods,” Procedia Comput. Sci., 102: 317-324, 2016.

[4] P. Moeskops et al., “Deep learning for multi-task medical image segmentation in multiple modalities,” in Proc. International Conference on Medical Image Computing and Computer-Assisted Intervention,: 468-486, 2016.

[5] D. Abdelhafiz et al., “Convolutional neural network for automated mass segmentation in mammography,” presented at the 2018 IEEE 8th International Conference on Computational Advances in Bio and Medical Sciences (ICCABS), Las Vegas, NV, USA, 2018.

[6] H. Fu et al., “Retinal vessel segmentation via deep learning network and fully-connected conditional random fields,” Presented at the 2016 IEEE 13th international symposium on biomedical imaging (ISBI), Prague, Czech Republic, 2016.

[7] Z. Guo et al., “Deep learning-based image segmentation on multimodal medical imaging,” IEEE Trans. Radiat. Plasma Med. Sci., 3(2): 162-169, 2019.

[8] M. Emre Celebi, Q. Wen, S. Hwang, H. Iyatomi, G. Schaefer, “Lesion border detection in dermoscopy images usingensembles of thresholding methods,” Skin Res. Technol., 19(1): e252– e258, 2013.

[9] H. Zhou, X. Li, G. Schaefer, M.E. Celebi, P. Miller, “Meanshift based gradient vector flow for image segmentation,” Comput.Vis. Image Understand., 117(9): 1004–1016, 2013.

[10] F. Xie, A.C. Bovik, “Automatic segmentation of dermoscopy images using self-generating neural networks seeded by genetic algorithm,” Pattern Recognit., 46(3): 1012–1019, 2013.

[11] T. Mendonça et al., “PH 2-A dermoscopic image database for research and benchmarking,” presented at the 2013 35th annual international conference of the IEEE engineering in medicine and biology society (EMBC), Osaka, Japan 2013.

[12] S. Pereira, A. Pinto, V. Alves, C.A. Silva, “Brain tumor segmentation using convolutional neural networks in MRI images,” IEEE Trans. Med. Imag., 35(5): 1240–1251, 2016.

[13] Y. LeCun, L. Bottou, Y. Bengio, P. Haffner, “Gradientbased learning applied to document recognition,” in Proc. the IEEE, 6(11): 2278–2324, 1998.

[14] S. Ioffe, C. Szegedy, "Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift,” in Proc. the 32nd International Conference on Machine Learning, PMLR, 37:448-456, 2015.

[15] T. Zhang “Solving large scale linear prediction problems using stochastic gradient descent algorithms” presented at the 21th International Conference on Machine Learning, Banff, Canada, 2004.

[16] Y. Yuan, M. Chao, Y.C. Lo, "Automatic skin lesion segmentation using deep fully convolutional networks with jaccard distance," IEEE Trans. Med. Imaging, 36(9): 1876-2017.

[17] M.H. Jafari, E. Nasr-Esfahani, N. Karimi, S. Soroushmehr, S. Samavi, K. Najarian, “Extraction of skin lesions from nondermoscopic images using deep learning,” arXiv preprint arXiv: 1609.02374, 2016.

[18] M. Avendi, A. Kheradvar, H. Jafarkhani, “A combined deeplearning and deformable-model approach to fully automatic segmentation of the left ventricle in cardiac MRI,” Med. Image Anal., 30: 108–119, 2016.

[19] https://www.kaggle.com/search?q=Chest+xray+segmentation

[20] J. Carvalho Souza, J.O.B. Diniz, J.L. Ferreira, G.L. Franc¸a da Silva, A. Corr´ea Silva, A. Cardoso de Paiva, “An automatic method for lung segmentation and reconstruction in chest X-Ray using deep neural networks,” Comput. Methods Programs Biomed., 177: 285-296, 2019.

[21] D. Gutman, N.C.F. Codella, E. Celebi, B. Helba, M. Marchetti, N. Mishra, A. Halpern, "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)," presented at the 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI), Washington, DC, USA, 2018.

[22] D. Gutman, N.C.F. Codella, E. Celebi, B. Helba, M. Marchetti, N. Mishra, A. Halpern, "Skin lesion analysis toward melanoma detection: A challenge at the International Symposium on Biomedical Imaging (ISBI) 2016, hosted by the International Skin Imaging Collaboration (ISIC)," May 2016.

LETTERS TO EDITOR

Journal of Electrical and Computer Engineering Innovations (JECEI) welcomes letters to the editor for the post-publication discussions and corrections which allows debate post publication on its site, through the Letters to Editor. Letters pertaining to manuscript published in JECEI should be sent to the editorial office of JECEI within three months of either online publication or before printed publication, except for critiques of original research. Following points are to be considering before sending the letters (comments) to the editor.

[1] Letters that include statements of statistics, facts, research, or theories should include appropriate references, although more than three are discouraged.

[2] Letters that are personal attacks on an author rather than thoughtful criticism of the author’s ideas will not be considered for publication.

[3] Letters can be no more than 300 words in length.

[4] Letter writers should include a statement at the beginning of the letter stating that it is being submitted either for publication or not.

[5] Anonymous letters will not be considered.

[6] Letter writers must include their city and state of residence or work.

[7] Letters will be edited for clarity and length.

Name *

Email Address *

Affiliation *

Comments *

Security Code *

Journal of Electrical and Computer Engineering Innovations (JECEI)

A Novel Method for Medical Image Segmentation based on Convolutional Neural Networks with SGD Optimization

References

References

Send comment about this article

Volume 9, Issue 1
January 2021
Pages 37-46

A Novel Method for Medical Image Segmentation based on Convolutional Neural Networks with SGD Optimization

References

References

Send comment about this article

Volume 9, Issue 1January 2021Pages 37-46

Volume 9, Issue 1
January 2021
Pages 37-46