An improved framework for brain tumor analysis using MRI based on YOLOv2 and convolutional neural network

Abstract

Brain tumor is a group of anomalous cells. The brain is enclosed in a more rigid skull. The abnormal cell grows and initiates a tumor. Detection of tumor is a complicated task due to irregular tumor shape. The proposed technique contains four phases, which are lesion enhancement, feature extraction and selection for classification, localization, and segmentation. The magnetic resonance imaging (MRI) images are noisy due to certain factors, such as image acquisition, and fluctuation in magnetic field coil. Therefore, a homomorphic wavelet filer is used for noise reduction. Later, extracted features from inceptionv3 pre-trained model and informative features are selected using a non-dominated sorted genetic algorithm (NSGA). The optimized features are forwarded for classification after which tumor slices are passed to YOLOv2-inceptionv3 model designed for the localization of tumor region such that features are extracted from depth-concatenation (mixed-4) layer of inceptionv3 model and supplied to YOLOv2. The localized images are passed toMcCulloch'sKapur entropy method to segment actual tumor region. Finally, the proposed technique is validated on three benchmark databases BRATS 2018, BRATS 2019, and BRATS 2020 for tumor detection. The proposed method achieved greater than 0.90 prediction scores in localization, segmentation and classification of brain lesions. Moreover, classification and segmentation outcomes are superior as compared to existing methods.