SegNet and Salp Water Optimization-driven Deep Belief Network for Segmentation and Classification of Brain Tumor

Abstract

Classification of brain tumor in Magnetic Resonance Imaging (MRI) images is highly popular in treatment planning, early diagnosis, and outcome evaluation. It is very difficult for classifying and diagnosing tumors from several images. Thus, an automatic prediction strategy is essential in classifying brain tumors as malignant, core, edema, or benign. In this research, a novel approach using Salp Water Optimization-based Deep Belief network (SWO-based DBN) is introduced to classify brain tumor. At the initial stage, the input image is pre-processed to eradicate the artifacts present in input image. Following pre-processing, the segmentation is executed by SegNet, where the SegNet is trained using the proposed SWO. Moreover, the Convolutional Neural Network (CNN) features are employed to mine the features for future processing. At last, the introduced SWO-based DBN technique efficiently categorizes the brain tumor with respect to the extracted features. Thereafter, the produced output of the introduced SegNet + SWO-based DBN is made use of in brain tumor segmentation and classification. The developed technique produced better results with highest values of accuracy at 0.933, specificity at 0.880, and sensitivity at 0.938 using BRATS, 2018 datasets and accuracy at 0.921, specificity at 0.853, and sensitivity at 0.928 for BRATS, 2020 dataset.