Brain tumor classification using deep convolutional autoencoder-based neural network: multi-task approach

Diagnosis, detection and classification of tumors, in the brain MRI images, are important because misdiagnosis can lead to death. This paper proposes a method that can diagnose brain tumors in the MRI images and classify them into 5 categories using a Convolutional Neural Network (CNN). The proposed network uses a Convolutional Auto-Encoder Neural Network (CANN) to extract and learn deep features of input images. Extracted deep features from each level are combined to make desirable features and improve results. To classify brain tumor into three categories (Meningioma, Glioma, and Pituitary) the proposed method was applied on Cheng dataset and has reached a considerable performance accuracy of 99.3%. To diagnosis and grading Glioma tumors, the proposed method was applied on IXI and BraTS 2017 datasets, and to classify brain images into six classes including Meningioma, Pituitary, Astrocytoma, High-Grade Glioma, Low-Grade Glioma and Normal images (No tumor), the all datasets including IXI, BraTS2017, Cheng and Hazrat-e-Rassol, was used by the proposed network, and it has reached desirable performance accuracy of 99.1% and 98.5%, respectively.

[1]  Bram van Ginneken,et al.  A survey on deep learning in medical image analysis , 2017, Medical Image Anal..

[2]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[3]  Tanveer F. Syeda-Mahmood,et al.  Building medical image classifiers with very limited data using segmentation networks , 2018, Medical Image Anal..

[4]  Guoyan Zheng,et al.  Periacetabular osteotomy through the pararectus approach: technical feasibility and control of fragment mobility by a validated surgical navigation system in a cadaver experiment , 2016, International Orthopaedics.

[5]  Walid Al-Atabany,et al.  Multi-Classification of Brain Tumor Images Using Deep Neural Network , 2019, IEEE Access.

[6]  Alessandro Sorichetta,et al.  High resolution global gridded data for use in population studies , 2017, Scientific Data.

[7]  Lalit M. Patnaik,et al.  Classification of magnetic resonance brain images using wavelets as input to support vector machine and neural network , 2006, Biomed. Signal Process. Control..

[8]  Antonio Criminisi.  Machine learning for medical images analysis , 2016, Medical Image Anal..

[9]  Sergio Guadarrama,et al.  Speed/Accuracy Trade-Offs for Modern Convolutional Object Detectors , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

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

[11]  Caterina Giannini,et al.  2016 Updates to the WHO Brain Tumor Classification System: What the Radiologist Needs to Know. , 2017, Radiographics : a review publication of the Radiological Society of North America, Inc.

[12]  Sebastian Thrun,et al.  Dermatologist-level classification of skin cancer with deep neural networks , 2017, Nature.

[13]  Jürgen Schmidhuber,et al.  Deep learning in neural networks: An overview , 2014, Neural Networks.

[14]  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.

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

[16]  Hassan Khotanlou,et al.  Glioma Brain Tumors Diagnosis and Classification in MR Images based on Convolutional Neural Networks , 2019, 2019 9th International Conference on Computer and Knowledge Engineering (ICCKE).

[17]  Subhashis Banerjee,et al.  Automated 3D segmentation of brain tumor using visual saliency , 2018, Inf. Sci..

[18]  Wei Shen,et al.  Multi-crop Convolutional Neural Networks for lung nodule malignancy suspiciousness classification , 2017, Pattern Recognit..

[19]  Chen Sun,et al.  Rethinking Spatiotemporal Feature Learning: Speed-Accuracy Trade-offs in Video Classification , 2017, ECCV.

[20]  Daniel L. Rubin,et al.  Automated Grading of Gliomas using Deep Learning in Digital Pathology Images: A modular approach with ensemble of convolutional neural networks , 2015, AMIA.

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

[22]  Stephen M Smith,et al.  Fast robust automated brain extraction , 2002, Human brain mapping.

[23]  David Dagan Feng,et al.  An Ensemble of Fine-Tuned Convolutional Neural Networks for Medical Image Classification , 2017, IEEE Journal of Biomedical and Health Informatics.

[24]  Konstantinos N. Plataniotis,et al.  Capsule Networks for Brain Tumor Classification Based on MRI Images and Coarse Tumor Boundaries , 2018, ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[25]  Heung-Il Suk,et al.  Deep Learning in Medical Image Analysis. , 2017, Annual review of biomedical engineering.

[26]  Nima Tajbakhsh,et al.  Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning? , 2016, IEEE Transactions on Medical Imaging.

[27]  Roberto Cipolla,et al.  SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[28]  Jae Y. Shin,et al.  Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning? , 2016, IEEE transactions on medical imaging.

[29]  K. Doi,et al.  Computer-aided diagnosis and artificial intelligence in clinical imaging. , 2011, Seminars in nuclear medicine.

[30]  Zhen Li,et al.  Towards Better Analysis of Deep Convolutional Neural Networks , 2016, IEEE Transactions on Visualization and Computer Graphics.

[31]  Sven Koitka,et al.  Traditional Feature Engineering and Deep Learning Approaches at Medical Classification Task of ImageCLEF 2016 , 2016, CLEF.

[32]  Yan Xu,et al.  Deep learning of feature representation with multiple instance learning for medical image analysis , 2014, 2014 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[33]  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.

[34]  G. Reifenberger,et al.  The 2016 World Health Organization Classification of Tumors of the Central Nervous System: a summary , 2016, Acta Neuropathologica.

[35]  Abdel-Badeeh M. Salem,et al.  Hybrid intelligent techniques for MRI brain images classification , 2010, Digit. Signal Process..

[36]  Robin Grant,et al.  Medical Management of Adult Glioma , 2019, Management of Adult Glioma in Nursing Practice.

[37]  Saed Khawaldeh,et al.  Noninvasive Grading of Glioma Tumor Using Magnetic Resonance Imaging with Convolutional Neural Networks , 2017 .

[38]  Qianjin Feng,et al.  Retrieval of Brain Tumors by Adaptive Spatial Pooling and Fisher Vector Representation , 2016, PloS one.

[39]  Mohsen Guizani,et al.  Deep Feature Learning for Medical Image Analysis with Convolutional Autoencoder Neural Network , 2017, IEEE Transactions on Big Data.

[40]  P. Mohamed Shakeel,et al.  Developing brain abnormality recognize system using multi-objective pattern producing neural network , 2018, Journal of Ambient Intelligence and Humanized Computing.

[41]  Hongfei Lin,et al.  Deep Transfer Learning for Modality Classification of Medical Images , 2017, Inf..

[42]  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.

[43]  Thomas Wahl,et al.  An improved database of coastal flooding in the United Kingdom from 1915 to 2016 , 2017, Scientific Data.

[44]  Hyunjin Park,et al.  Classification of low-grade and high-grade glioma using multi-modal image radiomics features , 2017, 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).