Enhanced performance of Dark-Nets for brain tumor classification and segmentation using colormap-based superpixel techniques

Abstract

The brain tumor is the deadliest disease in adults as it arises due to an abnormal mass of cells that grows rapidly and it alters the proper functioning of the organs. In clinical practice, radiographic images of different modalities are used to diagnose types of brain tumors, their size, and location. The proposed work aims to automatically classify, localize, and segment brain tumors from T1W-CE Magnetic Resonance Image (MRI) datasets. The T1W-CE MRI dataset is divided into 8:1:1, i.e., 80% training set, 10% of each validation, and testing set. To address the overfitting issues, the training data set is augmented using 2-levels wavelet decomposition and geometrical operations (scaling, rotation, translation). Performance of pre-trained DarkNet model (DarkNet-19 and DarkNet-53) is evaluated for the multi-class classification and localization of brain tumors. The best performing pre-trained DarkNet model achieved 99.60% of training accuracy and 98.81% of validation accuracy. The performance evaluation parameters confirm the superiority of the proposed methodology in comparison to the state-of-the-art on the T1W-CE MRI dataset. On 1070 T1W-CE testing images, the best-performing pre-trained DarkNet-53 model obtained a testing accuracy of 98.54% and Area Under Curve (AUC) of 0.99. The tumor is segmented using a 2-D superpixel segmentation technique with an average dice index of 0.94 ± 2.6% on the 793 brain tumor testing data. To prove the superiority of the proposed technique, it is implemented on MRI images from the BraTS2018 dataset. The comparative analysis of performance evaluation parameters of the proposed methodology with the state-of-the-art technique proves its robustness and clinical significance.

Keywords

• Brain tumor
• MRI
• Deep learning
• Superpixel segmentation