Skip to main navigation menu Skip to main content Skip to site footer

Identification of Brain Tumors Using CNN and ML with Diverse Feature Selection Techniques

Abstract

Early diagnosis and treatment is very essential in monitoring Brain tumor using MRI images. Convolutional Neural Networks (CNN) and Machine Learning (ML) classifiers have been widely used but there is not much work on how feature selection techniques would affect the performance of the CNN. Secondly, there is a need for investigation concerning small dataset adaptability and ML-CNN comparisons. To improve the classification accuracy, we integrate Univariate, Recursive Feature Elimination (RFE), Recursive Feature Elimination with Cross Validation (RFECV) with CNN in this study. Preprocessing, feature extraction & selection was carried out on the dataset consisting of 253 MRI images and they are classified using CNN and ML models (Logistic Regression, Decision Tree, Random Forest, Naïve Bayes). With the results 96%, CNN with Univariate Feature Selection performed better than ML classifiers, and other selection techniques. The results demonstrate that feature selection is necessary to get the best performance out of CNN models operating on small datasets. Future studies should be based on different deep learning architectures to improve classification and application in other datasets.

Keywords

brain tumor detection, CNN, feature selection, machine learning, medical imaging, MRI classification

PDF

References

  1. E. S. Biratu, F. Schwenker, Y. M. Ayano, and T. G. Debelee, “A Survey of Brain Tumor Segmentation and Classification Algorithms,” J Imaging, vol. 7, no. 9, p. 179, Sep. 2021, doi: 10.3390/jimaging7090179.
  2. M. C. Clark, L. O. Hall, D. B. Goldgof, R. Velthuizen, F. R. Murtagh, and M. S. Silbiger, “Automatic tumor segmentation using knowledge-based techniques,” IEEE Trans Med Imaging, vol. 17, no. 2, pp. 187–201, Apr. 1998, doi: 10.1109/42.700731.
  3. E.-S. A. El-Dahshan, H. M. Mohsen, K. Revett, and A.-B. M. Salem, “Computer-aided diagnosis of human brain tumor through MRI: A survey and a new algorithm,” Expert Syst Appl, vol. 41, no. 11, pp. 5526–5545, Sep. 2014, doi: 10.1016/j.eswa.2014.01.021.
  4. C. Shorten and T. M. Khoshgoftaar, “A survey on Image Data Augmentation for Deep Learning,” J Big Data, vol. 6, no. 1, p. 60, Dec. 2019, doi: 10.1186/s40537-019-0197-0.
  5. E. Av?ar and K. Salçin, “Detection and classification of brain tumours from MRI images using faster R-CNN,” Tehni?ki glasnik, vol. 13, no. 4, pp. 337–342, Dec. 2019, doi: 10.31803/tg-20190712095507.
  6. S. Sarkar, A. Kumar, S. Chakraborty, S. Aich, J.-S. Sim, and H.-C. Kim, “A CNN based Approach for the Detection of Brain Tumor Using MRI Scans,” Test Engineering and Management, vol. 83, pp. 16580–16586, 2020.
  7. R. Ranjbarzadeh, A. Bagherian Kasgari, S. Jafarzadeh Ghoushchi, S. Anari, M. Naseri, and M. Bendechache, “Brain tumor segmentation based on deep learning and an attention mechanism using MRI multi-modalities brain images,” Sci Rep, vol. 11, no. 1, p. 10930, May 2021, doi: 10.1038/s41598-021-90428-8.
  8. B. Kokila, M. S. Devadharshini, A. Anitha, and S. Abisheak Sankar, “Brain Tumor Detection and Classification Using Deep Learning Techniques based on MRI Images,” J Phys Conf Ser, vol. 1916, no. 1, p. 012226, May 2021, doi: 10.1088/1742-6596/1916/1/012226.
  9. A. Gumaei, M. M. Hassan, M. R. Hassan, A. Alelaiwi, and G. Fortino, “A Hybrid Feature Extraction Method With Regularized Extreme Learning Machine for Brain Tumor Classification,” IEEE Access, vol. 7, pp. 36266–36273, 2019, doi: 10.1109/ACCESS.2019.2904145.
  10. S. M. Vijithananda et al., “Feature extraction from MRI ADC images for brain tumor classification using machine learning techniques,” Biomed Eng Online, vol. 21, no. 1, p. 52, Dec. 2022, doi: 10.1186/s12938-022-01022-6.
  11. T. P, B. Bhat, and K. Prakash, “Detection and Classification of Tumour in Brain MRI,” International Journal of Engineering and Manufacturing, vol. 9, no. 1, pp. 11–20, Jan. 2019, doi: 10.5815/ijem.2019.01.02.
  12. M. W. Nadeem et al., “Brain Tumor Analysis Empowered with Deep Learning: A Review, Taxonomy, and Future Challenges,” Brain Sci, vol. 10, no. 2, p. 118, Feb. 2020, doi: 10.3390/brainsci10020118.
  13. A. S. Chandrabhatla, T. M. Horgan, C. C. Cotton, N. K. Ambati, and Y. E. Shildkrot, “Clinical Applications of Machine Learning in the Management of Intraocular Cancers: A Narrative Review,” Investigative Opthalmology & Visual Science, vol. 64, no. 10, p. 29, Jul. 2023, doi: 10.1167/iovs.64.10.29.
  14. S. Dalal et al., “An Efficient Brain Tumor Segmentation Method Based on Adaptive Moving Self-Organizing Map and Fuzzy K-Mean Clustering,” Sensors, vol. 23, no. 18, p. 7816, Sep. 2023, doi: 10.3390/s23187816.
  15. A. A. Akinyelu, F. Zaccagna, J. T. Grist, M. Castelli, and L. Rundo, “Brain Tumor Diagnosis Using Machine Learning, Convolutional Neural Networks, Capsule Neural Networks and Vision Transformers, Applied to MRI: A Survey,” J Imaging, vol. 8, no. 8, p. 205, Jul. 2022, doi: 10.3390/jimaging8080205.
  16. Chakrabarty Navoneel, “rain MRI Images for Brain Tumor Detection,” www.kaggle.com.
  17. I. Tuncer et al., “Swin-textural: A novel textural features-based image classification model for COVID-19 detection on chest computed tomography,” Inform Med Unlocked, vol. 36, p. 101158, 2023, doi: 10.1016/j.imu.2022.101158.
  18. A. K. Aggarwal, “Learning Texture Features from GLCM for Classification of Brain Tumor MRI Images using Random Forest Classifier,” WSEAS TRANSACTIONS ON SIGNAL PROCESSING, vol. 18, pp. 60–63, Apr. 2022, doi: 10.37394/232014.2022.18.8.
  19. Lohar Mamata and Chorage Suvarna, “A Robust Autism Brain MRI Classification with GLCM Features and Machine Learning,” Frontiers in Health Informatics, 2024, doi: 10.52783/fhi.9.
  20. P. K. Mall, P. K. Singh, and D. Yadav, “GLCM Based Feature Extraction and Medical X-RAY Image Classification using Machine Learning Techniques,” in 2019 IEEE Conference on Information and Communication Technology, IEEE, Dec. 2019, pp. 1–6. doi: 10.1109/CICT48419.2019.9066263.
  21. Priyanka and D. Kumar, “Feature Extraction and Selection of kidney Ultrasound Images Using GLCM and PCA,” Procedia Comput Sci, vol. 167, pp. 1722–1731, 2020, doi: 10.1016/j.procs.2020.03.382.
  22. Z. L. Chuan et al., “A Comparative of Two-Dimensional Statistical Moment Invariants Features in Formulating an Automated Probabilistic Machine Learning Identification Algorithm for Forensic Application,” Malaysian Journal of Fundamental and Applied Sciences, vol. 19, no. 4, pp. 525–538, Aug. 2023, doi: 10.11113/mjfas.v19n4.2917.
  23. M. Heydarian, T. E. Doyle, and R. Samavi, “MLCM: Multi-Label Confusion Matrix,” IEEE Access, vol. 10, pp. 19083–19095, 2022, doi: 10.1109/ACCESS.2022.3151048.
  24. J. Miao and W. Zhu, “Precision–recall curve (PRC) classification trees,” Evol Intell, vol. 15, no. 3, pp. 1545–1569, Sep. 2022, doi: 10.1007/s12065-021-00565-2.
  25. Z. Sun, B. Jiang, X. Li, J. Li, and K. Xiao, “A Data-Driven Approach for Lithology Identification Based on Parameter-Optimized Ensemble Learning,” Energies (Basel), vol. 13, no. 15, p. 3903, Jul. 2020, doi: 10.3390/en13153903.
  26. R. Yacouby and D. Axman, “Probabilistic Extension of Precision, Recall, and F1 Score for More Thorough Evaluation of Classification Models,” in Proceedings of the First Workshop on Evaluation and Comparison of NLP Systems, Stroudsburg, PA, USA: Association for Computational Linguistics, 2020, pp. 79–91. doi: 10.18653/v1/2020.eval4nlp-1.9.
  27. M.T.R Shawon,. G.M.S Shibli, F Ahmed, “Explainable cost-sensitive deep neural networks for brain tumor detection from brain MRI images considering data imbalance” in Multimed Tools Appl 2025. https://doi.org/10.1007/s11042-025-20842-x.
  28. A. M. Nancy, R. Maheswari, “Brain tumor segmentation and classification using transfer learning based CNN model with model agnostic concept interpretation” in Multimed Tools Appl,vol. 84, pp. 2509–2538, 2025. https://doi.org/10.1007/s11042-024-20353-1

Downloads

Download data is not yet available.

Similar Articles

1-10 of 32

You may also start an advanced similarity search for this article.