Machine learning for brain tumor classification: evaluating feature extraction and algorithm efficiency

被引：0
作者：
Kumar, Krishan [1 ,2 ]
Jyoti, Kiran [1 ]
Kumar, Krishan [3 ]
机构：
[1] Department of Computer Science, Guru Nanak Dev Engineering College, Punjab, Ludhiana
[2] IKG Punjab Technical University, Kapurthala
[3] Hindu College, Punjab, Amritsar
来源：
Discover Artificial Intelligence | 2024年 / 4卷 / 01期
关键词：
Brain tumors; Comparative analysis; Feature extraction; Machine learning; Magnetic resonance imaging;
D O I：
10.1007/s44163-024-00214-4
中图分类号：
学科分类号：
摘要：
Uncontrolled fast cell growth causes brain tumors, posing a significant threat to global health and leading to millions of deaths annually. Early cancer detection is crucial to save lives. The purpose of this study is to investigates the capability of machine learning algorithms and feature extraction methods to detection and classification of brain tumors. We implemented six machine learning algorithms and three features extraction methods, including Image Loading, HOG, and LBP. The objective of this study is to identify best combination of machine learning and features extraction method for brain tumor detection and classification. This study utilized two Brain Tumor MRI Datasets downloaded from Kaggle. Our analysis revealed that Random Forest emerged as the most effective classifier by achieving an accuracy of 99% with image loading feature extraction method based on different metrics, closely followed by SVM and Logistic Regression. However, performance varied with KNN, Naive Bayes, and Decision Tree, highlighting the importance of tailored approaches for optimal classification accuracy. Further optimization and experimentation are crucial for improving algorithm performance in real-world applications of brain tumor classification. A case study with interpretable machine learning is also presented in the paper. © The Author(s) 2024.
引用
收藏
相关论文
共 85 条
  • [1] Ullah M.S., Khan M.A., Albarakati H.M., Damasevicius R., Alsenan S., Multimodal brain tumor segmentation and classification from MRI scans based on optimized DeepLabV3+ and interpreted networks information fusion empowered with explainable AI, Comput Biol Med, 1, 182, (2024)
  • [2] Brain Tumor – Statistics, (2023)
  • [3] Xuanzhi L., Hakeem A., Mohaisen L., Umer M., Khan M.A., Alsenan S., Alsubai S., Innab N., BrainNet: an automated approach for brain stress prediction utilizing electrodermal activity signal with XLNet model, Front Comput Neurosci, (2024)
  • [4] Nodirov J., Abdusalomov A.B., Whangbo T.K., Attention 3D U-Net with multiple skip connections for segmentation of brain tumor images, Sensors, 22, (2022)
  • [5] Jlassi A., ElBedoui K., Barhoumi W., Brain tumor segmentation of lower-grade glioma across MRI images using hybrid Convolutional neural networks, Proceedings of the 15th International Conference on Agents and Artificial Intelligence, pp. 454-465, (2023)
  • [6] Ahuja S., Panigrahi B.K., Gandhi T.K., Enhanced performance of Dark-Nets for brain tumor classification and segmentation using colormap-based superpixel techniques, Mach Learn Appl, 7, (2022)
  • [7] Pereira S., Meier R., Alves V., Reyes M., Silva C.A., Automatic Brain Tumor Grading from MRI Data Using Convolutional Neural Networks and Quality Assessment. In Lecture Notes in Computer Science (Including Subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
  • [8] Springer, Cham, Switzerland, 11038, pp. 106-114, (2018)
  • [9] Tandel G.S., Tiwari A., Kakde O.G., Performance optimisation of deep learning models using majority voting algorithm for brain tumour classification, Comput Biol Med, 135, (2021)
  • [10] Aldape K., Et al., Challenges to curing primary brain tumours, Nat Rev Clin Oncol, 16, 8, pp. 509-520, (2019)
123456789