Volume 15 • Issue 1 • PP: 01-16 • 2025
Brain Tumor Detection: Integrating Machine Learning and Deep Learning for Robust Brain Tumor Classification
View PDF Abstract
Accurate detection and classification of brain tumors are essential for timely diagnosis and effective treatment planning. This study presents an integrated framework leveraging both machine learning (ML) and deep learning (DL) models for brain tumor detection and classification using MRI images. Two publicly available datasets are utilized: one for binary classification (tumor vs. no tumor) and another for multiclass classification (glioma, meningioma, and pituitary tumors). Comprehensive preprocessing steps, including resizing, feature extraction using the Gray Level Co-occurrence Matrix (GLCM), and feature selection via Chi-square testing, were employed to optimize the dataset for modeling. Machine learning models such as Decision Trees, K-Nearest Neighbors (KNN), Support Vector Machines (SVM), and AdaBoost were compared with deep learning architectures like Convolutional Neural Networks (CNNs) and the pre-trained VGG16 model. Hyperparameter optimization techniques, including grid search and the Adam optimizer, were used to enhance model performance. The models were evaluated using metrics such as accuracy, precision, recall, F1-score, Mean Squared Error (MSE), and Mean Absolute Error (MAE). Results indicate that the VGG16 model consistently outperformed other approaches, achieving high validation accuracy. This study highlights the potential of integrating ML and DL techniques for accurate and efficient brain tumor detection and classification, offering valuable tools for medical diagnostics.
Keywords
References
[1] G. C¸ ınarer and B. G. Emiro˘glu, “Classificatin of brain tumors by machine learning algorithms,” in 2019 3rd international symposium on multidisciplinary studies and innovative technologies (ISMSIT), pp. 1–4, IEEE, 2019.
[2] D. N. George, H. B. Jehlol, A. S. A. Oleiwi, et al., “Brain tumor detection using shape features and machine learning algorithms,” International Journal of Advanced Research in Computer Science and Software Engineering, vol. 5, no. 10, pp. 454–459, 2015.
[3] K. Sharma, A. Kaur, and S. Gujral, “Brain tumor detection based on machine learning algorithms,” International Journal of Computer Applications, vol. 103, no. 1, 2014.
[4] F. Polly, S. Shil, M. A. Hossain, A. Ayman, and Y. M. Jang, “Detection and classification of hgg and lgg brain tumor using machine learning,” in 2018 International Conference on Information Networking (ICOIN), pp. 813–817, IEEE, 2018.
[5] 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,” IEEE Access, vol. 7, pp. 12–19, 2018.
[6] A. Vidyarthi and N. Mittal, “Brain tumor segmentation approaches: Review, analysis and anticipated solutions in machine learning,” in 2015 39th National Systems Conference (NSC), pp. 1–6, IEEE, 2015.
[7] R. Gurusamy and V. Subramaniam, “A machine learning approach for mri brain tumor classification,” Computers, Materials and Continua, vol. 53, no. 2, pp. 91–109, 2017.
[8] J. Amin, M. Sharif, A. Haldorai, M. Yasmin, and R. S. Nayak, “Brain tumor detection and classification using machine learning: a comprehensive survey,” Complex & intelligent systems, vol. 8, no. 4, pp. 3161– 3183, 2022.
[9] E. Carver, C. Liu, W. Zong, Z. Dai, J. M. Snyder, J. Lee, and N. Wen, “Automatic brain tumor segmentation and overall survival prediction using machine learning algorithms,” in Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries: 4th International Workshop, BrainLes 2018, Held in Conjunction with MICCAI 2018, Granada, Spain, September 16, 2018, Revised Selected Papers, Part II 4, pp. 406–418, Springer, 2019.
[10] S. M. Vijithananda, M. L. Jayatilake, B. Hewavithana, T. Gonc¸alves, L. M. Rato, B. S. Weerakoon, T. D. Kalupahana, A. D. Silva, and K. D. Dissanayake, “Feature extraction from mri adc images for brain tumor classification using machine learning techniques,” Biomedical engineering online, vol. 21, no. 1, p. 52, 2022.
[11] B. Balakumar, P. Raviraj, and E. D. Devi, “Brain tumor classification using machine learning algorithms,” Elysium Journal of Engineering Research and Management, vol. 4, no. 2, pp. 30–41, 2017.
[12] S. Rinesh, K. Maheswari, B. Arthi, P. Sherubha, A. Vijay, S. Sridhar, T. Rajendran, Y. A.Waji, et al., “Investigations on brain tumor classification using hybrid machine learning algorithms,” Journal of Healthcare Engineering, vol. 2022, 2022.
[13] S. Anantharajan, S. Gunasekaran, T. Subramanian, and R. Venkatesh, “Mri brain tumor detection using deep learning and machine learning approaches,” Measurement: Sensors, vol. 31, p. 101026, 2024.
[14] L. Chato and S. Latifi, “Machine learning and deep learning techniques to predict overall survival of brain tumor patients using mri images,” in 2017 IEEE 17th international conference on bioinformatics and bioengineering (BIBE), pp. 9–14, IEEE, 2017.
[15] P. R. Kshirsagar, A. N. Rakhonde, and P. Chippalkatti, “Mri image based brain tumor detection using machine learning,” Test Engineering and Management, vol. 81, pp. 3672–3680, 2020.
Cite This Article
Choose your preferred format