An Image Processing-based and Deep Learning Model to Classify Brain Cancer
Received: 12 May 2024 | Revised: 26 May 2024 and 30 May 2024 | Accepted: 31 May 2024 | Online: 8 June 2024
Corresponding author: Amal Al-Shahrani
Abstract
In recent years, the prevalence of cancer has increased significantly around the world. Cancer is considered one of the most dangerous diseases in humans. Cancer screening devices, such as Magnetic Resonance Imaging (MRI), X-ray imaging, ultrasound imaging, and others, play an important role in its early detection. This study aims to facilitate cancer tumor detection on mobile phones with high accuracy in a short period of time using deep learning techniques. A brain tumor dataset was used, consisting of 4,489 images and 14 classified types, and experiments were carried out using ResNet 12, DenseNet, YOLOv8, and MobileNet to evaluate them in terms of accuracy, speed, and model size. ResNet12, DenseNet, YOLOv8, and MobileNet results indicated satisfactory accuracy ranging from 88% to 97.3%. YOLOv8 was the most suitable model, as its fastest inference time of 1.8 ms, preprocessing time of 0.1 ms, highest accuracy of 97.3%, and compact model size make it ideal for real-time mobile applications.
Keywords:
cancer detection, You-Only-Look-Once (YOLO), MRI, image processing, diagnosis, brain cancer, deep learningDownloads
References
"Brain Tumors and Brain Cancer." https://www.hopkinsmedicine.org/health/conditions-and-diseases/brain-tumor.
P. K. Mall and P. K. Singh, "BoostNet: a method to enhance the performance of deep learning model on musculoskeletal radiographs X-ray images," International Journal of System Assurance Engineering and Management, vol. 13, no. 1, pp. 658–672, Jan. 2022.
K. C. Kamal, Z. Yin, M. Wu, and Z. Wu, "Evaluation of deep learning-based approaches for COVID-19 classification based on chest X-ray images," Signal, Image and Video Processing, vol. 15, no. 5, pp. 959–966, Jan. 2021.
"Types of Deep Learning & Their Uses in Healthcare." https://healthitanalytics.com/features/types-of-deep-learning-their-uses-in-healthcare.
"Transfer learning." https://en.wikipedia.org/w/index.php?title=Transfer_learning.
J. Ding and X. Li, "An Approach for Validating Quality of Datasets for Machine Learning," in 2018 IEEE International Conference on Big Data (Big Data), Seattle, WA, USA, Dec. 2018, pp. 2795–2803.
J. Cho, K. Lee, E. Shin, G. Choy, and S. Do, "How much data is needed to train a medical image deep learning system to achieve necessary high accuracy?" arXiv, Jan. 2016.
"Brain Tumor MRI Images 44 Classes." https://www.kaggle.com/datasets/fernando2rad/brain-tumor-mri-images-44c.
M. Havaei et al., "Brain tumor segmentation with Deep Neural Networks," Medical Image Analysis, vol. 35, pp. 18–31, Jan. 2017.
"What is the purpose of image preprocessing in deep learning?" https://www.isahit.com/blog/what-is-the-purpose-of-image preprocessing-in-deep-learning.
H. Singh, N. Helian, R. Adams, and Y. Sun, "Sentiment Analysis using BLSTM-ResNet on Textual Images," in 2022 International Joint Conference on Neural Networks (IJCNN), Padua, Italy, Jul. 2022, pp. 1–8.
H. Choi, S. Ryu, and H. Kim, "Short-Term Load Forecasting based on ResNet and LSTM," in 2018 IEEE International Conference on Communications, Control, and Computing Technologies for Smart Grids (SmartGridComm), Aalborg, Denmark, Oct. 2018, pp. 1–6.
S. Anwar, S. R. Soomro, S. K. Baloch, A. A. Patoli, and A. R. Kolachi, "Performance Analysis of Deep Transfer Learning Models for the Automated Detection of Cotton Plant Diseases," Engineering, Technology & Applied Science Research, vol. 13, no. 5, pp. 11561–11567, Oct. 2023.
E. Elfatimi, R. Eryigit, and L. Elfatimi, "Beans Leaf Diseases Classification Using MobileNet Models," IEEE Access, vol. 10, pp. 9471–9482, 2022.
D. Sinha and M. El-Sharkawy, "Thin MobileNet: An Enhanced MobileNet Architecture," in 2019 IEEE 10th Annual Ubiquitous Computing, Electronics & Mobile Communication Conference (UEMCON), New York City, NY, USA, Oct. 2019, pp. 0280–0285.
A. Alzahrani, "Digital Image Forensics: An Improved DenseNet Architecture for Forged Image Detection," Engineering, Technology & Applied Science Research, vol. 14, no. 2, pp. 13671–13680, Apr. 2024.
V. S. Kumar, Dr. S. A. Pearline, and S. R. Bose, "Real-Time Plant Species Recognition Using Non-Averaged Densenet-169 Deep Learning Paradigm," May 2022.
A. Alayed, R. Alidrisi, E. Feras, S. Aboukozzana, and A. Alomayri, "Real-Time Inspection of Fire Safety Equipment using Computer Vision and Deep Learning," Engineering, Technology & Applied Science Research, vol. 14, no. 2, pp. 13290–13298, Apr. 2024.
J. Du, "Understanding of Object Detection Based on CNN Family and YOLO," Journal of Physics: Conference Series, vol. 1004, no. 1, Dec. 2018, Art. no. 012029.
"What is YOLOv8? The Ultimate Guide." https://blog.roboflow.com/whats-new-in-yolov8/.
"How to Activate GPU Computing in Google Colab." https://saturncloud.io/blog/how-to-activate-gpu-computing-in-google-colab/.
B. Selcuk and T. Serif, "A Comparison of YOLOv5 and YOLOv8 in the Context of Mobile UI Detection," in Mobile Web and Intelligent Information Systems, Aug. 2023, pp. 161–174.
A. Jabbar, S. Naseem, T. Mahmood, T. Saba, F. S. Alamri, and A. Rehman, "Brain Tumor Detection and Multi-Grade Segmentation Through Hybrid Caps-VGGNet Model," IEEE Access, vol. 11, pp. 72518–72536, Jun. 2023.
H. Mohsen, E. S. A. El-Dahshan, E. S. M. El-Horbaty, and A. B. M. Salem, "Classification using deep learning neural networks for brain tumors," Future Computing and Informatics Journal, vol. 3, no. 1, pp. 68–71, Jun. 2018.
A. Ari and D. Hanbay, "Deep learning based brain tumor classification and detection system," Turkish Journal of Electrical Engineering and Computer Sciences, vol. 26, no. 5, pp. 2275–2286, Jan. 2018.
G. Manogaran, P. M. Shakeel, A. S. Hassanein, P. M. Kumar, and G. C. Babu, "Machine Learning Approach-Based Gamma Distribution for Brain Tumor Detection and Data Sample Imbalance Analysis," in IEEE Access, vol. 7, pp. 12-19, Jan. 2019.
Downloads
How to Cite
License
Copyright (c) 2024 Amal Al-Shahrani, Wusaylah Al-Amoudi, Raghad Bazaraah, Atheer Al-Sharief, Hanan Farouquee
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.
