Exploring statistical parameters of machine learning techniques for detection and classification of brain tumor

Abstract

A computerized system can improve the disease identifying abilities of doctor and also reduce the time needed for the identification and decision-making in healthcare. Gliomas are the brain tumors that can be labeled as Benign (non-cancerous) or Malignant (cancerous) tumor. Hence, the different stages of the tumor are extremely important for identification of appropriate medication. In this paper, a system has been proposed to detect brain tumor of different stages by MR images. The proposed system uses Fuzzy C-Mean (FCM) as a clustering technique for better outcome. The main focus in this paper is to refine the required features in two steps with the help of Discrete Wavelet Transform (DWT) and Independent Component Analysis (ICA) using three machine learning techniques i.e. Random Forest (RF), Artificial Neural Network (ANN) and Support Vector Machine (SVM). The final outcome of our experiment indicated that the proposed computerized system identifies the brain tumor using RF, ANN and SVM with 100%, 91.6% and 95.8%, accuracy respectively. We have also calculated Sensitivity, Specificity, Matthews’s Correlation Coefficient and AUC-ROC curve. Random forest shows the highest accuracy as compared to Support Vector Machine and Artificial Neural Networks.