Segmenting the Lesion Area of Brain Tumor using Convolutional Neural Networks and Fuzzy K-Means Clustering

Document Type : Original Article

Authors

Department of Electrical and Computer Engineering, University of Birjand, Birjand, Iran

Abstract

Brain tumor Segmentation is one of the most crucial methods of medical image processing. Non-automatic segmentations are broadly used in clinical diagnosis and medication. However, this kind of segmentation does not have accuracy in medical images, especially in terms of brain tumors, and it provides a low level of reliability. The primary objective of this paper is to develop a methodology for brain tumor segmentation. In this paper, a combination of Convolutional  Neural Network and Fuzzy K-means algorithm has been presented to segment the lesion area of brain tumor. It contains three phases, Image preprocessing to reduce computational complexity, Attribute extraction and selection and Segmentation. At first, the database images are pre-processed using adaptive filters and wavelet transform in order to recover the image from the noise state and reduce the computational complexity. Then feature extraction is performed by the proposed deep neural network. Finally, it is processed through the Fuzzy K-Means algorithm to segment the tumor region separately. The innovation of this article is related to the implementation of deep neural network with optimal parameters, identification of related features and removal of unrelated and repetitive features with the aim of observing a subset of features that describe the problem well and with minimal reduction in efficiency. This results in reduced feature sets, storage of data collection resources during operation, and overall data reduction to limit storage requirements. This proposed segmentation approach has been verified on BRATS dataset and produces the accuracy of 98.64%, sensitivity of 100% specificity of 99%.

Keywords

Main Subjects


  1. Yogananda, C.G.B., Shah, B.R., Vejdani-Jahromi, M., Nalawade, S.S., Murugesan, G.K., Yu, F.F., Pinho, M.C., Wagner, B.C., Emblem, K.E. and Bjørnerud, A., "A fully automated deep learning network for brain tumor segmentation", Tomography, Vol. 6, No. 2, (2020), 186-193. doi: 10.18383/j.tom.2019.00026.
  2. Soomro, T.A., Zheng, L., Afifi, A.J., Ali, A., Soomro, S., Yin, M. and Gao, J., "Image segmentation for mr brain tumor detection using machine learning: A review", IEEE Reviews in Biomedical Engineering, (2022). doi: 10.1109/RBME.2022.3185292.
  3. Liu, Z., Tong, L., Chen, L., Jiang, Z., Zhou, F., Zhang, Q., Zhang, X., Jin, Y. and Zhou, H., "Deep learning based brain tumor segmentation: A survey", Complex & Intelligent Systems, Vol. 9, No. 1, (2023), 1001-1026. doi: 10.3390/app122311980.
  4. Magadza, T. and Viriri, S., "Deep learning for brain tumor segmentation: A survey of state-of-the-art", Journal of Imaging, Vol. 7, No. 2, (2021), 19. doi: 10.3390/jimaging7020019.
  5. Somasundaram, S. and Gobinath, R., "Current trends on deep learning models for brain tumor segmentation and detection–a review", in 2019 International conference on machine learning, big data, cloud and parallel computing (COMITCon), IEEE. (2019), 217-221.
  6. Biratu, E.S., Schwenker, F., Ayano, Y.M. and Debelee, T.G., "A survey of brain tumor segmentation and classification algorithms", Journal of Imaging, Vol. 7, No. 9, (2021), 179. doi: 10.3390/jimaging7090179.
  7. Devunooru, S., Alsadoon, A., Chandana, P. and Beg, A., "Deep learning neural networks for medical image segmentation of brain tumours for diagnosis: A recent review and taxonomy", Journal of Ambient Intelligence and Humanized Computing, Vol. 12, (2021), 455-483. doi: 10.1007/s12652-020-01998-w.
  8. Zhang, W., Wu, Y., Yang, B., Hu, S., Wu, L. and Dhelim, S., "Overview of multi-modal brain tumor mr image segmentation", in Healthcare, MDPI. Vol. 9, (2021), 1051.
  9. Karimi, D. and Salcudean, S.E., "Reducing the hausdorff distance in medical image segmentation with convolutional neural networks", IEEE Transactions on Medical Imaging, Vol. 39, No. 2, (2019), 499-513. doi: 10.1109/TMI.2019.2930068.
  10. Arabi, H., Dowling, J.A., Burgos, N., Han, X., Greer, P.B., Koutsouvelis, N. and Zaidi, H., "Comparative study of algorithms for synthetic ct generation from mri: Consequences for mri‐guided radiation planning in the pelvic region", Medical Physics, Vol. 45, No. 11, (2018), 5218-5233. doi: 10.1002/mp.13187.
  11. Arabi, H., Zeng, G., Zheng, G. and Zaidi, H., "Novel adversarial semantic structure deep learning for mri-guided attenuation correction in brain pet/mri", European Journal of Nuclear Medicine and Molecular Imaging, Vol. 46, (2019), 2746-2759. doi: 10.1007/s00259-019-04380.
  12. Bahrami, A., Karimian, A., Fatemizadeh, E., Arabi, H. and Zaidi, H., "A new deep convolutional neural network design with efficient learning capability: Application to ct image synthesis from mri", Medical Physics, Vol. 47, No. 10, (2020), 5158-5171. doi: 10.1002/mp.14418.
  13. Angulakshmi, M. and Deepa, M., "A review on deep learning architecture and methods for mri brain tumour segmentation", Current Medical Imaging, Vol. 17, No. 6, (2021), 695-706. doi: 10.2174/1573405616666210108122048.
  14. Isensee, F., Kickingereder, P., Wick, W., Bendszus, M. and Maier-Hein, K.H., "Brain tumor segmentation and radiomics survival prediction: Contribution to the brats 2017 challenge", in Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries: Third International Workshop, BrainLes 2017, Held in Conjunction with MICCAI 2017, Quebec City, QC, Canada, September 14, 2017, Revised Selected Papers 3, Springer. (2018), 287-297.
  15. Toğaçar, M., Cömert, Z. and Ergen, B., "Classification of brain mri using hyper column technique with convolutional neural network and feature selection method", Expert Systems with Applications, Vol. 149, (2020), 113274. doi: 10.1016/j.eswa.2020.113274Get rights and content.
  16. Amin, J., Sharif, M., Gul, N., Raza, M., Anjum, M.A., Nisar, M.W. and Bukhari, S.A.C., "Brain tumor detection by using stacked autoencoders in deep learning", Journal of Medical Systems, Vol. 44, (2020), 1-12. doi: 10.1007/s10916-019-1483.
  17. Islam, R., Imran, S., Ashikuzzaman, M. and Khan, M.M.A., "Detection and classification of brain tumor based on multilevel segmentation with convolutional neural network", Journal of Biomedical Science and Engineering, Vol. 13, No. 4, (2020), 45-53. doi: 10.4236/jbise.2020.134004.
  18. Zhang, W., Yang, G., Huang, H., Yang, W., Xu, X., Liu, Y. and Lai, X., "Me‐net: Multi‐encoder net framework for brain tumor segmentation", International Journal of Imaging Systems and Technology, Vol. 31, No. 4, (2021), 1834-1848. doi: 10.1002/ima.22571.
  19. Hasan, S.K. and Linte, C.A., "A modified u-net convolutional network featuring a nearest-neighbor re-sampling-based elastic-transformation for brain tissue characterization and segmentation", in 2018 IEEE Western New York Image and Signal Processing Workshop (WNYISPW), IEEE. (2018), 1-5.
  20. Rajan, P. and Sundar, C., "Brain tumor detection and segmentation by intensity adjustment", Journal of Medical Systems, Vol. 43, (2019), 1-13. doi: 10.1007/s10916-019-1368-4&.
  21. Begum, S.S. and Lakshmi, D.R., "Combining optimal wavelet statistical texture and recurrent neural network for tumour detection and classification over mri", Multimedia Tools and Applications, Vol. 79, (2020), 14009-14030. doi: 10.1007/s11042-020-08643.
  22. Thaha, M.M., Kumar, K.P.M., Murugan, B., Dhanasekeran, S., Vijayakarthick, P. and Selvi, A.S., "Brain tumor segmentation using convolutional neural networks in mri images", Journal of Medical Systems, Vol. 43, (2019), 1-10. doi: 10.1007/s10916-019-1416-0.
  23. Gao, X. and Qian, Y., "Segmentation of brain lesions from ct images based on deep learning techniques", in Medical Imaging 2018: Biomedical Applications in Molecular, Structural, and Functional Imaging, SPIE. Vol. 10578, (2018), 610-615.
  24. Emadi, M., Jafarian Dehkordi, Z. and Iranpour Mobarakeh, M., "Improving the accuracy of brain tumor identification in magnetic resonanceaging using super-pixel and fast primal dual algorithm", International Journal of Engineering, Transactions C: Aspects, Vol. 36, No. 3, (2023), 505-512. doi: 10.5829/IJE.2023.36.03C.10.
  25. Azimi, B., Rashno, A. and Fadaei, S., "Fully convolutional networks for fluid segmentation in retina images", in 2020 International Conference on Machine Vision and Image Processing (MVIP), IEEE. (2020), 1-7.
  26. Khan, A.R., Khan, S., Harouni, M., Abbasi, R., Iqbal, S. and Mehmood, Z., "Brain tumor segmentation using k‐means clustering and deep learning with synthetic data augmentation for classification", Microscopy Research and Technique, Vol. 84, No. 7, (2021), 1389-1399. doi: 10.1002/jemt.23694.
  27. Rai, H.M., Chatterjee, K. and Dashkevich, S., "Automatic and accurate abnormality detection from brain mr images using a novel hybrid unetresnext-50 deep cnn model", Biomedical Signal Processing and Control, Vol. 66, (2021), 102477. doi: 10.1016/j.bspc.2021.102477.
  28. Engineering, J.O.H., "Retracted: Brain tumor detection and classification by mri using biologically inspired orthogonal wavelet transform and deep learning techniques", Journal of Healthcare Engineering, Vol. 2023, (2023), 9845732. doi: 10.1155/2023/9845732.
  29. Obeidavi, M.R. and Maghooli, K., "Tumor detection in brain mri using residual convolutional neural networks", in 2022 International Conference on Machine Vision and Image Processing (MVIP), IEEE. (2022), 1-5.
  30. Dubey, S.R., Singh, S.K. and Chaudhuri, B.B., "Activation functions in deep learning: A comprehensive survey and benchmark", Neurocomputing, (2022). doi: 10.1016/j.neucom.2022.06.111.
  31. Clough, J.R., Byrne, N., Oksuz, I., Zimmer, V.A., Schnabel, J.A. and King, A.P., "A topological loss function for deep-learning based image segmentation using persistent homology", IEEE Transactions on Pattern Analysis and Machine Intelligence, Vol. 44, No. 12, (2020), 8766-8778. doi: 10.1109/TPAMI.2020.3013679.
  32. Balasubramanian, K. and Ananthamoorthy, N., "Correlation-based feature selection using bio-inspired algorithms and optimized kelm classifier for glaucoma diagnosis", Applied Soft Computing, Vol. 128, (2022), 109432. doi: 10.1016/j.asoc.2022.109432.
  33. Nawaz, M., Mehmood, Z., Nazir, T., Naqvi, R.A., Rehman, A., Iqbal, M. and Saba, T., "Skin cancer detection from dermoscopic images using deep learning and fuzzy k‐means clustering", Microscopy Research and Technique, Vol. 85, No. 1, (2022), 339-351. doi: 10.1002/jemt.23908.
  34. Musallam, A.S., Sherif, A.S. and Hussein, M.K., "A new convolutional neural network architecture for automatic detection of brain tumors in magnetic resonance imaging images", IEEE Access, Vol. 10, (2022), 2775-2782. doi: 10.1109/ACCESS.2022.3140289.