Segmentation and Classification of Brain Tumor using 3D-UNet Deep Neural Networks

Pranjal Agrawal,Nitish Katal,Nishtha Hooda
DOI: https://doi.org/10.1016/j.ijcce.2022.11.001
2022-11-12
International Journal of Cognitive Computing in Engineering
Abstract:Highlights • The proposed work considers an image registration model, a 3D U-Net model, for the volumetric segmentation of the brain tumor. • The next step is the classification of the brain tumors into meningioma, glioma, and pituitary tumors. • Work proposes a convolutional neural network (CNN) based automated brain tumour segmentation/classification system. CNN is made up of a convolutional network that performs automated segmentation and feature extraction, supplemented by a classical neural network that performs classification. • The work has also been compared with the other state-of-the-art approaches reported in the literature. Early detection and diagnosis of a brain tumour enhance the medical options and the patient's chance of recovery. Magnetic resonance imaging (MRI) is used to detect and diagnose brain tumours. However, the manual identification of brain tumours from a large number of MRI images in clinical practice solely depends on the time and experience of medical professionals. Presently, computer aided expert systems are booming to facilitate medical diagnosis and treatment recommendations. Numerous machine learning and deep learning based frameworks are employed for brain tumour detection. This paper aims to design an efficient framework for brain tumour segmentation and classification using deep learning techniques. The study employs the 3D-UNet model for the volumetric segmentation of the MRI images, followed by the classification of the tumour using CNNs. The loss and precision diagrams are presented to establish the validity of the models. The performance of proposed models is measured, and the results are compared with those of other approaches reported in the literature. It is found that the proposed work is more efficacious than the state-of-the-art techniques.
What problem does this paper attempt to address?