Streamlining Brain Tumor Classification with Custom Transfer Learning in MRI Images (2310.13108v1)
Abstract: Brain tumors are increasingly prevalent, characterized by the uncontrolled spread of aberrant tissues in the brain, with almost 700,000 new cases diagnosed globally each year. Magnetic Resonance Imaging (MRI) is commonly used for the diagnosis of brain tumors and accurate classification is a critical clinical procedure. In this study, we propose an efficient solution for classifying brain tumors from MRI images using custom transfer learning networks. While several researchers have employed various pre-trained architectures such as RESNET-50, ALEXNET, VGG-16, and VGG-19, these methods often suffer from high computational complexity. To address this issue, we present a custom and lightweight model using a Convolutional Neural Network-based pre-trained architecture with reduced complexity. Specifically, we employ the VGG-19 architecture with additional hidden layers, which reduces the complexity of the base architecture but improves computational efficiency. The objective is to achieve high classification accuracy using a novel approach. Finally, the result demonstrates a classification accuracy of 96.42%.
- A. Rehman, M. A. Khan, T. Saba, Z. Mehmood, U. Tariq, and N. Ayesha, “Microscopic brain tumor detection and classification using 3D CNN and feature selection architecture,” Microsc. Res. Tech., vol. 84, no. 1, pp. 133–149, 2021.
- R. Chelghoum, A. Ikhlef, A. Hameurlaine, and S. Jacquir, “Transfer learning using convolutional neural network architectures for brain tumor classification from MRI images,” in IFIP Advances in Information and Communication Technology, vol. 583, Cham: Springer International Publishing, 2020, pp. 189–200.
- S. Ahuja, B. K. Panigrahi, and T. Gandhi, “Transfer learning based brain tumor detection and segmentation using superpixel technique,” in 2020 International Conference on Contemporary Computing and Applications (IC3A), 2020.
- “Brain tumor - statistics,” Cancer.net, 25-Jun-2012. [Online]. Available: https://www.cancer.net/cancer-types/brain-tumor/statistics. [Accessed: 07-Feb-2023].
- M. T. H. K. Tusar and R. K. Anik, “Automated detection of acute lymphoblastic leukemia subtypes from microscopic blood smear images using Deep Neural Networks,” arXiv [eess.IV], 2022.
- K. Simonyan and A. Zisserman, “Very deep convolutional networks for large-scale image recognition,” arXiv [cs.CV], 2014.
- M. K. Islam, M. S. Ali, M. S. Miah, M. M. Rahman, M. S. Alam, and M. A. Hossain, “Brain tumor detection in MR image using superpixels, principal component analysis and template based K-means clustering algorithm,” Mach. Learn. Appl., vol. 5, no. 100044, p. 100044, 2021.
- M. T. Islam, S. R. Rafa, and M. G. Kibria, “Early prediction of heart disease using PCA and hybrid genetic algorithm with k-means,” in 2020 23rd International Conference on Computer and Information Technology (ICCIT), 2020, pp. 1–6.
- M. T. Haque Khan Tusar, M. T. Islam, and F. I. Raju, “Detecting chronic kidney disease(CKD) at the initial stage: A novel hybrid feature-selection method and robust data preparation pipeline for different ML techniques,” in 2022 5th International Conference on Computing and Informatics (ICCI), 2022, pp. 400–407.
- A. Younis, L. Qiang, C. O. Nyatega, M. J. Adamu, and H. B. Kawuwa, “Brain tumor analysis using deep learning and VGG-16 ensembling learning approaches,” Appl. Sci. (Basel), vol. 12, no. 14, p. 7282, 2022.
- E. Irmak, “Multi-classification of brain tumor MRI images using deep convolutional neural network with fully optimized framework,” Iran. J. Sci. Technol. Trans. Electr. Eng., vol. 45, no. 3, pp. 1015–1036, 2021.
- Javed Hossain (1 paper)
- Md. Touhidul Islam (62 papers)
- Md. Taufiqul Haque Khan Tusar (5 papers)