An efficient and optimized Convolution Neural Network for Brain Tumour Detection

Abstract

Brain tumour is a life threatning disease and can affect children and adults. This study focuses on classifying MRI scan images of brain into one of 4 classes namely: glioma tumour, meningioma tumour, pituitary tumour and normal brain. Person affected with brain tumours will need treatments such as surgery, radiation therapy or chemotherapy. Pretrained Convolution Neural Networks such as VGG19, MobileNet, and AlexNet which have been widely used for image classification using transfer learning. However due to huge storage space requirements these are not effectively deployed on edge devices for creation of robotic devices. Hence a compressed version of these models have been created using Genetic Algorithm algorithm which occupies nearly 30-40% of space and also a reduced inference time which is less by around 50% of original model. The accuracy provided by VGG19, AlexNet, MobileNet and Proposed CNN before compression was 92.18%, 89.45%, 93.75% and 96.85% respectively. Similarly the accuracy after compression for VGG19, AlexNet, MobileNet and Proposed CNN was 91.34%, 88.92%, 94.40% and 95.29%.