Automated Classification of Brain Tumor Images Employing DWT and Deep Neural Networks

Automated Classification of Brain Tumor Images Employing DWT and Deep Neural Networks

Ms. Khushboo Nagar, Varun Jaiswal,

Mr. Atish Mishra, Department CSE & Malwa Institute of Technology, Indore

Mr. Pratyush Sharma,HOD CSE Department Malwa Institute of Technology ,Indore

---------------------------------------------------------------------***---------------------------------------------------------------------

Abstract - Brain tumour classification using conventional techniques involving human intervention is prone to errors. Errors in feature extraction and/or classification may turn out to be fatal. Hence focus has shifted on automated techniques for the feature extraction and classification of brain tumour images. In the present work, a mechanism for the same has been proposed involving Artificial Neural Networks (ANN). Three categories have been incorporated in the present work, viz. normal brain, brain with benign tumour and brain with malignant tumour. In the present case, MRI images of the brain have been used to train an Artificial Neural Network which subsequently predicts the category of some new MRI data. The key challenge in designing such a system is attaining high accuracy of classification. This can be achieved by accurate feature extraction mechanism and then designing an appropriate Neural Network for training and testing. At the outset, segmentation has been employed to separate the region of the tumour (if any). Later the Discrete Wavelet Transform (DWT) is used as a preprocessing tool. The DWT helps in removing the non linearities of the image signal and smoothen it out. Since images exhibit abrupt changes in pixel values, conventional Fourier methods like the Fourier Transform prove to be unfit for image analysis. Subsequently, features like contrast, correlation, energy, homogeneity, mean, standard deviation, entropy, RMS value, variance, smoothness, kurtosis and skewness have been evaluated. Prior to training the Neural Network using the feature values, Principal Component Analysis (PCA) has been employed to find trends in the feature values. A Probabilistic Neural Network (PNN) has been designed and it has been trained using the feature values. I has been found that brain tumour MRI image are classified into normal, benign and malignant category using PNN classifier with an accuracy of 97% for the used data set which can be attributed to the efficacy of the proposed method.

Key Words:  Brain Tumor Classification, Machine Learning, Feature Extraction, Neural Networks, Confusion Matrix, Classification Accuracy.