Multi-Class classification of brain tumor images using data augmentation with deep neural network

Abstract Brain is central to the nervous system (CNS) of the human body, it is a group of white masses of cells. These cells grow abnormally by multiplying rapidly with each other is called a brain tumor, which can lead to the death of an individual if not detected. The most common types of tumors which are occurred in the brain are meningioma, glioma, and pituitary tumors. Manually diagnosing and classifying brain tumors is a big challenge for physicians and clinical professionals, accuracy depends solely on their experience. Therefore, computer-assisted technology must be used to overcome these limitations. With the daily improvement of modern medical standards, artificial intelligence plays an important role in identifying and classifying brain tissues and this is of great interest to researchers. It is currently being developed with various machine learning and DL structures. In the past, several ML algorithms such as SVM, KNN, and CNN have been implemented for tumor identification and tumor binary and multi classification using MR images. But these algorithms are not giving the best results. Therefore, we have proposed a DNN based VGG-16 network to improve the accuracy of multi-classification of brain tumor MR images. Finally, our proposed model gives the best results with 98% accuracy after 20 epochs of model training.

[1]  Mihaela Lascu,et al.  Tumor Detection and Classification of MRI Brain Image using Different Wavelet Transforms and Support Vector Machines , 2019, 2019 42nd International Conference on Telecommunications and Signal Processing (TSP).

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

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

[4]  Srikanth Tammina,et al.  Transfer learning using VGG-16 with Deep Convolutional Neural Network for Classifying Images , 2019, International Journal of Scientific and Research Publications (IJSRP).

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

[6]  Adria Casamitjana,et al.  Cascaded V-Net using ROI masks for brain tumor segmentation , 2018 .

[7]  Zijun Zhang,et al.  Improved Adam Optimizer for Deep Neural Networks , 2018, 2018 IEEE/ACM 26th International Symposium on Quality of Service (IWQoS).

[8]  Dharun V.S,et al.  Extraction of Texture Features using GLCM and Shape Features using Connected Regions , 2016 .

[9]  Mark D. McDonnell,et al.  Understanding Data Augmentation for Classification: When to Warp? , 2016, 2016 International Conference on Digital Image Computing: Techniques and Applications (DICTA).

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