Automated diagnosis of multi-class brain abnormalities using MRI images: A deep convolutional neural network based method

Abstract Automated detection of multi-class brain abnormalities through magnetic resonance imaging (MRI) has received much attention due to its clinical significance and therefore has become an active area of research over the years. The earlier automated schemes often followed traditional machine learning paradigms, in which the proper choice of features and classifiers has remained a major concern. Therefore, deep learning algorithms have been profoundly applied in various medical imaging applications. In this paper, a deep convolutional neural network (CNN) based automated approach is designed for the diagnosis of multi-class brain abnormalities. The proposed CNN model comprises five layers with learnable parameters: four convolutional layers and one fully-connected layer. The objective of designing such a custom deep network is to achieve greater classification performance with reduced number of parameters. The proposed model is evaluated on two benchmark multi-class brain MRI datasets namely, MD-1 and MD-2. The model achieved a classification accuracy of 100.00% and 97.50% on MD-1 and MD-2 datasets respectively. Moreover, four pre-trained CNN models based on the transfer learning approach have been tested over the same datasets. The comparative analysis with existing schemes indicates the superiority of the proposed method.

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

[2]  Mahrokh G. Shayesteh,et al.  Robust algorithm for brain magnetic resonance image (MRI) classification based on GARCH variances series , 2013, Biomed. Signal Process. Control..

[3]  Yu-Dong Zhang,et al.  Chinese Sign Language Fingerspelling via Six-Layer Convolutional Neural Network with Leaky Rectified Linear Units for Therapy and Rehabilitation , 2019, J. Medical Imaging Health Informatics.

[4]  Banshidhar Majhi,et al.  Discrete ripplet-II transform and modified PSO based improved evolutionary extreme learning machine for pathological brain detection , 2017, Neurocomputing.

[5]  Jean Ponce,et al.  A Theoretical Analysis of Feature Pooling in Visual Recognition , 2010, ICML.

[6]  Yudong Zhang,et al.  Polarimetric synthetic aperture radar image segmentation by convolutional neural network using graphical processing units , 2017, Journal of Real-Time Image Processing.

[7]  Yudong Zhang,et al.  PATHOLOGICAL BRAIN DETECTION BY ARTIFICIAL INTELLIGENCE IN MAGNETIC RESONANCE IMAGING SCANNING (INVITED REVIEW) , 2016 .

[8]  Banshidhar Majhi,et al.  Deep extreme learning machine with leaky rectified linear unit for multiclass classification of pathological brain images , 2019, Multimedia Tools and Applications.

[9]  Anjan Gudigar,et al.  Application of multiresolution analysis for automated detection of brain abnormality using MR images: A comparative study , 2019, Future Gener. Comput. Syst..

[10]  Yudong Zhang,et al.  A hybrid method for MRI brain image classification , 2011, Expert Syst. Appl..

[11]  Anjan Gudigar,et al.  Brain pathology identification using computer aided diagnostic tool: A systematic review , 2019, Comput. Methods Programs Biomed..

[12]  Banshidhar Majhi,et al.  Brain MR image classification using two-dimensional discrete wavelet transform and AdaBoost with random forests , 2016, Neurocomputing.

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

[14]  Dipankar Das,et al.  Enhanced SenticNet with Affective Labels for Concept-Based Opinion Mining , 2013, IEEE Intelligent Systems.

[15]  Yoshua Bengio,et al.  Extracting and composing robust features with denoising autoencoders , 2008, ICML '08.

[16]  Sidan Du,et al.  Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation , 2019, Multimedia Tools and Applications.

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

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

[19]  Yudong Zhang,et al.  Effect of spider-web-plot in MR brain image classification , 2015, Pattern Recognit. Lett..

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

[21]  Chenxi Huang,et al.  Multiple Sclerosis Identification by 14-Layer Convolutional Neural Network With Batch Normalization, Dropout, and Stochastic Pooling , 2018, Front. Neurosci..

[22]  Anjan Gudigar,et al.  Automated Categorization of Multi-Class Brain Abnormalities Using Decomposition Techniques With MRI Images: A Comparative Study , 2019, IEEE Access.

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

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

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

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

[27]  Paul K. Joseph,et al.  Classification of MRI brain images using combined wavelet entropy based spider web plots and probabilistic neural network , 2013, Pattern Recognit. Lett..

[28]  Kenneth Revett,et al.  Computer-aided diagnosis of human brain tumor through MRI: A survey and a new algorithm , 2014, Expert Syst. Appl..

[29]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[30]  Xiaojun Chang,et al.  Automated Diagnosis of Pathological Brain Using Fast Curvelet Entropy Features , 2020, IEEE Transactions on Sustainable Computing.

[31]  Geoffrey E. Hinton,et al.  Reducing the Dimensionality of Data with Neural Networks , 2006, Science.

[32]  Li Fei-Fei,et al.  ImageNet: A large-scale hierarchical image database , 2009, CVPR.

[33]  Yudong Zhang,et al.  Smart pathological brain detection by synthetic minority oversampling technique, extreme learning machine, and Jaya algorithm , 2017, Multimedia Tools and Applications.

[34]  Khan Muhammad,et al.  Five-category classification of pathological brain images based on deep stacked sparse autoencoder , 2017, Multimedia Tools and Applications.

[35]  Sidan Du,et al.  Application of stationary wavelet entropy in pathological brain detection , 2018, Multimedia Tools and Applications.