Brain Tumor Classification Using ResNet-101 Based Squeeze and Excitation Deep Neural Network

The brain tumor is one of the leading and most alarming cause of death with a high socio-economic impact in Occidental as well as eastern countries. Differential diagnosis and classification of tumor types (Gliomas, Meningioma, and Pituitary tumor) from MRI data are required to assist radiologists as well as to avoid the dangerous histological biopsies. In the meantime, improving the accuracy and stability of diagnosis is also one challenging task. Many methods have been proposed for this purpose till now. In this work, an automatic tool for classification of brain tumor from MRI data is presented where the image slice samples are passed into a Squeeze and Excitation ResNet model based on Convolutional Neural Network (CNN). The use of zero-centering and normalization of intensity for smooth variation of the intensity over the tissues was also investigated as a preprocessing step which together with data augmentation proved to be very effective. A relative study had been done to prove the efficacy of the proposed CNN model in free tumor database. Experimental evaluation shows that the proposed CNN archives an overall accuracy rate of 89.93% without data augmentation. Addition of data augmentation has further improved the accuracies up to 98.67%, 91.81% and 91.03% for Glioma, Meningioma and Pituitary tumor respectively with an overall accuracy of 93.83%. Promising improvement with reference to sensitivity and specificity compared with some of the state-of-the-art methods was also observed.

[1]  T. Gupta,et al.  Indian data on central nervous tumors: A summary of published work , 2016, South Asian Journal of Cancer.

[2]  Wojciech Czarnecki,et al.  On Loss Functions for Deep Neural Networks in Classification , 2017, ArXiv.

[3]  Vibha Vyas,et al.  A novel training algorithm for convolutional neural network , 2016, Complex & Intelligent Systems.

[4]  Pascal Vincent,et al.  Representation Learning: A Review and New Perspectives , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

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

[6]  Josef Kittler,et al.  Elastic transformation of the image pixel grid for similarity based face identification , 2002, Object recognition supported by user interaction for service robots.

[7]  Y. Skaik Understanding and using sensitivity, specificity and predictive values , 2008, Indian journal of ophthalmology.

[8]  Luis Perez,et al.  The Effectiveness of Data Augmentation in Image Classification using Deep Learning , 2017, ArXiv.

[9]  Gang Sun,et al.  Squeeze-and-Excitation Networks , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[10]  Matthew J. Schneider,et al.  Mean centering helps alleviate “micro” but not “macro” multicollinearity , 2016, Behavior research methods.

[11]  Christos Davatzikos,et al.  Classification of brain tumor type and grade using MRI texture and shape in a machine learning scheme , 2009, Magnetic resonance in medicine.

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

[13]  El-Sayed M. El-Horbaty,et al.  Classification using deep learning neural networks for brain tumors , 2017, Future Computing and Informatics Journal.

[14]  Yike Guo,et al.  TensorLayer: A Versatile Library for Efficient Deep Learning Development , 2017, ACM Multimedia.

[15]  Daniel Fabbri,et al.  Deep learning for brain tumor classification , 2017, Medical Imaging.

[16]  Jun Cheng,et al.  brain tumor dataset , 2016 .

[17]  J. Seetha,et al.  Brain Tumor Classification Using Convolutional Neural Networks , 2018, Biomedical and Pharmacology Journal.

[18]  Qianjin Feng,et al.  Enhanced Performance of Brain Tumor Classification via Tumor Region Augmentation and Partition , 2015, PloS one.

[19]  A. Munshi Central nervous system tumors: Spotlight on India , 2016, South Asian journal of cancer.

[20]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.