Enhancing Brain Tumor Classification Performance Through Feature Selection In Machine Learning Models
Accurate classification of brain tumor images is crucial for accurate prognosis and effective treatment planning. This study involved an analysis of the application of machine learning in the classification of brain tumors. We employed segmentation using mU-Net, as well as feature extraction and feature selection techniques to identify the most prominent features. We identified 14 significant features using the Information Gain algorithm to classify brain tumors from 33 available features. The K-NN, SVM, Decision Tree, and Random Forest classification models were employed, and the SVM model yielded the highest performance with an accuracy of 0.929.
Authors
- Pradini Puspitaningayu (ORCID: https://orcid.org/0000-0002-2053-2907)
- Ari Kusumaningsih (ORCID: https://orcid.org/0000-0002-8716-9890)
- Lusia Rakhmawati (ORCID: https://orcid.org/0000-0002-2325-1465)
- Hapsari Peni Agustin Tjahyaningtijas
Institutions
- State University of Semarang (ID)
- Universitas Trunodjoyo Madura (ID)
Publication Details
- Published
- 2024-11-18
- DOI
- https://doi.org/10.1109/healthcom60970.2024.10880757
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00