Transfer Learning Based Brain Tumor Detection and Segmentation using Superpixel Technique

Manual detection and segmentation of glioma is difficult because of its asymmetrical shape, flexibility in location and uneven boundaries. The proposed work presents a transfer learning-based brain tumor detection and segmentation is done using superpixel technique. In the initial phase, brain images are classified into three categories namely, normal, Low-Grade Glioma (LGG) and High-Grade Glioma (HGG) based on the availability of tumor. The proposed methodology is examined on Brain Tumor Segmentation (BraTS) 2019 challenge database. The tumor detection task is performed on VGG-19 transfer learning model. On the VGG-19 transfer learning model at epoch 6, training data yield 99.82% training accuracy, 96.32% validation accuracy and 99.30% testing accuracy. The obtained specificity is 100% and sensitivity is 97.81%, and Area Under Curve (AUC) is 0.99. In the second phase, segment the tumor within LGG and HGG images using the superpixel segmentation technique. Superpixel segmentation approach leads to an average detection dice index of 0.932 against the ground truth data. The experimental results prove the superiority of the proposed methodology in comparison to existing methods.

[1]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[2]  Christos Davatzikos,et al.  Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features , 2017, Scientific Data.

[3]  Victor Alves,et al.  Brain Tumor Segmentation Using Convolutional Neural Networks in MRI Images , 2016, IEEE Transactions on Medical Imaging.

[4]  Moosa Ayati,et al.  Magnetic resonance imaging-based brain tumor grades classification and grading via convolutional neural networks and genetic algorithms , 2019, Biocybernetics and Biomedical Engineering.

[5]  Changming Sun,et al.  Supervised Brain Tumor Segmentation Based on Gradient and Context-Sensitive Features , 2019, Front. Neurosci..

[6]  Hesham F. A. Hamed,et al.  Two-phase multi-model automatic brain tumour diagnosis system from magnetic resonance images using convolutional neural networks , 2018, EURASIP Journal on Image and Video Processing.

[7]  P. M. Ameer,et al.  Brain tumor classification using deep CNN features via transfer learning , 2019, Comput. Biol. Medicine.

[8]  G. G. Lakshmi Priya,et al.  Brain tumour segmentation from MRI using superpixels based spectral clustering , 2018, J. King Saud Univ. Comput. Inf. Sci..

[9]  Fang Zhang,et al.  Deep convolutional activation features for large scale Brain Tumor histopathology image classification and segmentation , 2015, 2015 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[10]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[11]  Brian B. Avants,et al.  The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS) , 2015, IEEE Transactions on Medical Imaging.

[12]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[13]  Amit Verma,et al.  Deep learning based enhanced tumor segmentation approach for MR brain images , 2019, Appl. Soft Comput..

[14]  Christopher Joseph Pal,et al.  Brain tumor segmentation with Deep Neural Networks , 2015, Medical Image Anal..

[16]  et al.,et al.  Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challenge , 2018, ArXiv.

[17]  Pascal Fua,et al.  SLIC Superpixels Compared to State-of-the-Art Superpixel Methods , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[18]  Zaka Ur Rehman,et al.  Fully automated multi-parametric brain tumour segmentation using superpixel based classification , 2019, Expert Syst. Appl..

[19]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[20]  Ronald M. Summers,et al.  Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning , 2016, IEEE Transactions on Medical Imaging.