Multiclass Light Weight Brain Tumor Classification and Detection using Machine Learning Model Yolo 5
Abstract
Abstract Views: 187
Early brain tumor identification is a critical challenge for neurologists and radiologists. Manually identifying brain tumors through magnetic resonance imaging (MRI) is difficult and prone to mistakes. The diagnosis of tumor is a complex job when performed in a traditional manner. Brain abnormalities can be fatal, lowering a patient's quality of life and adversely harming their overall health. Brain tumors vary in nature based on where they are situated and how rapidly they develop inside the skull. Tumors are a proliferation of abnormal nerve cells that form a mass. Some brain tumors begin in the cells that support the brain's nerve cells. This paper proposes a machine learning algorithm known as YOLO v5 SSD (single shot detection) to detect and classify such tumors namely meningioma, glioma, and pituitary gland with 88% accuracy. For this purpose, data augmentation was applied to the publically available dataset from Kaggle. MRI of different classes including 396 glioma images, 397 meningioma, 380 no tumor, and 399 images of pituitary tumors were employed. The current study presents false negative, true positive false positive, and true negative, which were used to test the YOLO v5 (You Only Look Once) classifier performance. It was determined that the YOLO v5 model is giving 88% accuracy.
Downloads
References
Johns Hopkins Medicine. “Brain tumor types.” Johns Hopkins Medicine. https://www.hopkinsmedicine.org/health/conditions-and-diseases/brain-tumor/brain-tumor-types
National Cancer Institute. “Diffuse midline gliomas diagnosis and treatment.” NCI. https://www.cancer.gov/rare-brain-spine-tumor/tumors/diffuse-midline-gliomas
Z. Krawczyk and J. Starzynski, “Bones detection in the pelvic area on the basis of YOLO neural network,” in 19th Int. Conf. Comput. Prob. Elect. Eng., Banska Stiavnica, Sep. 2018, pp. 1–4. doi: https://doi.org/10.1109/CPEE.2018.8506970
T. Saba, A. S. Mohamed, M. El-Affendi, J. Amin, and M. Sharif, "Brain tumor detection using fusion of hand crafted and deep learning features," Cog. Syst. Res., vol. 59, pp. 221–230, Jan. 2020, doi: https://doi.org/10.1016/j.cogsys.2019.09.007
A. Hu and N. Razmjooy, “Brain tumor diagnosis based on metaheuristics and deep learning,” Int. J. Imaging. Syst. Technol., vol. 31, no. 2, pp. 657–669, Jun. 2021, doi: https://doi.org.10.1002/ima.22495
N. Noreen, S. Palaniappan, A. Qayyum, I. Ahmad, M. Imran, and M. Shoaib, “A Deep learning model based on concatenation approach for the diagnosis of brain tumor,” IEEE Access, vol. 8, pp. 55135–55144, 2020, doi: https://doi.org/10.1109/ACCESS.2020.2978629
P. Kshirsagar, A. N. Rakhonde, and P. Chippalkatti, “MRI image based brain tumor detection using machine learning,” vol. 81, pp. 4431–4434, Dec. 2019.
R. Hashemzehi, S. J. S. Mahdavi, M. Kheirabadi, and S. R. Kamel, "Detection of brain tumors from MRI images base on deep learning using hybrid model CNN and NADE," Biocybernet. Biomed. Engi., vol. 40, no.3, pp. 1225–1232, Sep. 2020, doi: https://doi.org/10.1016/j.bbe.2020.06.001
M. A. Khan, et al., "Brain tumor detection and classification: A framework of marker‐based watershed algorithm and multilevel priority features selection," Microscop. Res. Tech., vol. 82, no. 6, pp. 909–922, Feb. 2019, doi: https://doi.org/10.1002/jemt.23238
U. Agrawal, E. N. Brown, and L. D. Lewis, “Model-based physiological noise removal in fast fMRI,” NeuroImage, vol. 205, Art. no. 116231, Jan. 2020, doi: https://doi.org/10.1016/j.neuroimage.2019.116231
Y. K. Dubey and M. M. Mushrif, “FCM clustering algorithms for segmentation of brain MR images,” vol. 2016, Art. no. 3406406, doi: https://doi.org/10.1155/2016/3406406
S. Paul, D. T. Ahad, and M. Hasan, “Brain cancer segmentation using YOLOv5 deep neural network,” arxiv, doi:
https://doi.org/10.48550/arXiv.2212.13599
N. M. Dipu, S. A. Shohan, and K. M. A. Salam, “Brain tumor detection using various deep learning algorithms,” in 2021 Int. Conf. Sci. Contem. Technol., Dhaka, Bangladesh, Aug. 2021, pp. 1–6. doi: https://doi.org.10.1109/ICSCT53883.2021.9642649
S. Arunachalam and G. Sethumathavan, “An effective tumor detection in MR brain images based on deep CNN approach: i- YOLOV5,” Appl. Artif. Intell., vol. 36, no. 1, Art. no. 2151180, Dec. 2022, doi: https://doi.org/10.1080/08839514.2022.2151180
A.-A. Nayan, et al., “A deep learning approach for brain tumor detection using magnetic resonance imaging,” Int. J. Elec. Comput. Eng., vol. 13, no. 1, pp. 1039–1047, Feb. 2023, doi: https://doi.org/10.11591/ijece.v13i1
A. M. G. Allah, A. M. Sarhan, and N. M. Elshennawy, “Classification of brain MRI tumor images based on deep learning PGGAN augmentation,” Diagnostics (Basel), vol. 11, no. 12, Art. no. 2343, Dec. 2021, doi: https://doi.org/10.3390/diagnostics11122343
S. Ghosh, A. Chaki, and K. Santosh, "Improved U-Net architecture with VGG-16 for brain tumor segmentation," Phys. Eng. Sci. Med., vol. 44, pp. 703–712 May 2021, doi: https://doi.org/10.1007/s13246-021-01019-w
N. M. Dipu, S. A. Shohan, and K. M. A. Salam, “Deep learning based brain tumor detection and classification,” in 2021 Int. Conf. Intell. Technol., Hubli, India, Jun. 2021, pp. 1–6. doi: https://doi.org/10.1109/CONIT51480.2021.9498384
M. B. Naceur, R. Saouli, M. Akil, and R. Kachouri, “Fully automatic brain tumorsegmentation using end-to-end incremental deep neural networks in MRI images,” Comput. Meth. Prog. Biomed., vol. 166, pp. 39–49, Nov. 2018, doi: https://doi.org/10.1016/j.cmpb.2018.09.007
P. R. Lorenzo et al., “Segmenting brain tumors from FLAIR MRI using fully convolutional neural networks,” Comput. Meth. Prog. Biomed., vol. 176, pp. 135–148, Jul. 2019, doi: https://doi.org/10.1016/j.cmpb.2019.05.006
N. F. Alhussainan, B. B. Youssef, and M. M. Ben Ismail, “A deep learning approach for brain tumor firmness detection using YOLOv4,” in 2022 45th Int. Conf. Telecommun. Signal Proc., Prague, Czech Republic, Jul. 2022, pp. 342–348. doi: https://doi.org/10.1109/TSP55681.2022.9851237
A. Myronenko, “3D MRI brain tumor segmentation using autoencoder regularization," in Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries. BrainLes 2018. Lecture Notes in Computer Science, A. Crimi, S. Bakas, H. Kuijf, F. Keyvan, M. T. Reyes, and T. van Walsum, Eds., Cham, Springer, 2019, doi: https://doi.org/10.1007/978-3-030-11726-9_28
S. Pereira, A. Pinto, J. Amorim, A. Ribeiro, V. Alves, and C. A. Silva, “Adaptive feature recombination and recalibration for semantic segmentation with fully convolutional networks,” IEEE Trans. Med. Imaging, vol. 38, no. 12, pp. 2914–2925, Dec. 2019, doi: https://doi.org/10.1109/TMI.2019.2918096
M. Hammami, D. Friboulet, and R. Kechichian, “Cycle GAN-Based data augmentation for multi-organ detection in CT Images Via Yolo,” in 2020 IEEE Int. Conf. Image Proc., Abu Dhabi, United Arab Emirates, Oct. 2020, pp. 390–393. doi: https://doi.org/10.1109/ICIP40778.2020.9191127
K. Muhammad, S. Khan, J. D. Ser, and V. H. C. de Albuquerque, “Deep learning for multigrade brain tumor classification in smart healthcare systems: A prospective survey,” IEEE Trans. Neural Netw. Learning Syst., vol. 32, no. 2, pp. 507–522, Feb. 2021, doi: https://doi.org/10.1109/TNNLS.2020.2995800
T. Shelatkar, Urvashi, M. Shorfuzzaman, A. Alsufyani, and K. Lakshmanna, “Diagnosis of brain tumor using light weight deep learning model with fine-tuning approach,” Comput. Mathemat. Meth. Med., vol. 2022, pp. 1–9, Jul. 2022, doi: https://doi.org/10.1155/2022/2858845
Copyright (c) 2022 Asif Raza Raza, Usman Amjad, Muhammad Abubakr, Dr.Asad Abbasi, Humera Azam, Asher Ali
This work is licensed under a Creative Commons Attribution 4.0 International License.
UMT-AIR follow an open-access publishing policy and full text of all published articles is available free, immediately upon publication of an issue. The journal’s contents are published and distributed under the terms of the Creative Commons Attribution 4.0 International (CC-BY 4.0) license. Thus, the work submitted to the journal implies that it is original, unpublished work of the authors (neither published previously nor accepted/under consideration for publication elsewhere). On acceptance of a manuscript for publication, a corresponding author on the behalf of all co-authors of the manuscript will sign and submit a completed the Copyright and Author Consent Form.
