Enhanced Brain Tumor Classification using VGG19 and Data Augmentation Techniques
Keywords:
Image classification, MRI, Transfer Learning, Deep Learning, Brain Tumor DetectionAbstract
Timely and successful therapy relies on the correct diagnosis and categorization of brain tumors. In this study, we present a deep learning-based method for the automated identification of brain tumors in MR images. We used a data collection that was available to the public and included both MRI images of brain tumors and normal brain. We down sampled the images, added more data, and normalized the pixel values as preliminary processing. We utilized the cleaned data set to construct a deep learning model based on VGG19. Our model showed a 96.3% accuracy, 96.3% AUC ROC score, 96% precision, 96% recall, and 96% F1 score. To evaluate our model's performance, we used various metrics such as accuracy, precision, recall, F1 score, confusion matrix, AUC ROC score, and ROC curve. Furthermore, we compared our model's results with the best performing model from a related study that utilized machine learning algorithms such as Random Forest, SVM, Logistic regression, Gradient boosting, and the achieved accuracy is 92.4%, precision is 85.0%, recall is 94.4%, F1 score of 89.5%, and AUC ROC score touching 97.2%.
References
[1] D. Y. Lee, ''Roles of mTOR signaling in brain development,'' Experim. Neurobiol., vol. 24, no. 3, pp. 177-185, Sep. 2015. https://doi.org/10.5607/en.2015.24.3.177
[2] Subhamoy Mandal, Aaron B. Greenblatt and Jingzhi An, -Imaging Intelligence: AI Is Transforming Medical Imaging Across the Imaging Spectrum‖, IEEE Pulse, vol.9, Issue.5, pp. 16 - 24, 2018. https://doi.org/10.1109/MPUL.2018.2857226
[3] Ankit Ghosh, Purbita Kole and Alok Kole, -Automatic Identification of Covid-19 from Chest X-ray Images using Enhanced Machine Learning Techniques‖, International Research Journal of Engineering and Technology (IRJET),vol.8, issue.9, no.115, pp.765-772, 2021.
[4] Miles N. Wernick, Yongyi Yang, Jovan G. Brankov, Grigori Yourganov and Stephen C. Strother, -Machine Learning in Medical Imaging‖, IEEE Signal Processing Magazine, vol.27, issue.4, pp.25-38, 2010. https://doi.org/10.1109/MSP.2010.936730
[5] Chetanpal Singh, -Medical Imaging using Deep Learning Models‖, European Journal of Engineering and Technology Research, vol.6, issue.5, pp.156-167, 2021. https://doi.org/10.24018/ejeng.2021.6.5.2491
[6] Raut, Gajendra, et al. "Deep learning approach for brain tumor detection and segmentation." 2020 International Conference on Convergence to Digital World-Quo Vadis (ICCDW). IEEE, 2020. https://doi.org/10.1109/ICCDW45521.2020.9318681
[7] Hossain, Tonmoy, et al. "Brain tumor detection using convolutional neural network." 2019 1st international conference on advances in science, engineering and robotics technology (ICASERT). IEEE, 2019. https://doi.org/10.1109/ICASERT.2019.8934561
[8] Pashaei, Ali, Hedieh Sajedi, and Niloofar Jazayeri. "Brain tumor classification via convolutional neural network and extreme learning machines." 2018 8th International conference on computer and knowledge engineering (ICCKE). IEEE, 2018. https://doi.org/10.1109/ICCKE.2018.8566571
[9] Srinivas, B., and G. Sasibhusana Rao. "Unsupervised learning algorithms for MRI brain tumor segmentation." 2018 Conference on Signal Processing And Communication Engineering Systems (SPACES). IEEE, 2018. https://doi.org/10.1109/SPACES.2018.8316341
[10] Jiang, Jun, et al. "3D brain tumor segmentation in multimodal MR images based on learning population-and patient-specific feature sets." Computerized Medical Imaging and Graphics 37.7-8 (2013): 512-521. https://doi.org/10.1016/j.compmedimag.2013.05.007
[11] Cheng, Jun, et al. "Enhanced performance of brain tumor classification via tumor region augmentation and partition. https://doi.org/10.1371/journal.pone.0140381
[12] Shin, Hoo-Chang, et al. "Deep convolutional neural networks for computer-aided detection: CNN architectures, dataset characteristics and transfer learning." IEEE transactions on medical imaging 35.5 (2016): 1285-1298. https://doi.org/10.1109/TMI.2016.2528162
[13] Howard, Andrew G. "Some improvements on deep convolutional neural network based image classification." arXiv preprint arXiv:1312.5402 (2013).
[14] Hawkins D. The Problem of Overfitting. J Chem Inf Comput Sci. 2004;44(1):1-12. https://doi.org/10.1021/ci0342472
[15] Perez, Luis, and Jason Wang. "The effectiveness of data augmentation in image classification using deep learning." arXiv preprint arXiv:1712.04621 (2017).
[16] Ghosh, Ankit, and ALOK KOLE. "A Comparative Study of Enhanced Machine Learning Algorithms for Brain Tumor Detection and Classification." (2021). https://doi.org/10.36227/techrxiv.16863136.v1
Downloads
Published
Issue
Section
License
Copyright (c) 2023 Kaishav Saini , Pankaj Richhariya
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Re-users must give appropriate credit, provide a link to the license, and indicate if changes were made. You may do so in any reasonable manner, but not in any way that suggests the licensor endorses you or your use. This license allows for redistribution, commercial and non-commercial, as long as the original work is properly credited.
7
