Deep Learning Based Lightweight Model for Brain Tumor Classification and Segmentation

Abstract

This paper presents two lightweight deep learning models for efficient detection and segmentation of brain tumors from MRI scans. A custom-made Convolutional Neural Network (CNN) is designed for identification of four different classes of brain tumors viz. Meningioma, Glioma, Pituitary brain tumor and normal (no tumor). Furthermore, another tailor-made lightweight model is presented for the segmentation of the tumor from the Magnetic Resonance Imaging (MRI) scans. The output of the segmentation model is the ‘mask’ depicting the tumor region. The overall performance in terms of detection accuracy, and segmentation accuracy, for the two models is found to be approximately 95% for both the cases individually. The proposed models are worthy additions to the existing literature on brain tumor classification and segmentation models due to their low-parameter count which make the models amenable for deployment on resource-constrained edge hardware.