Detection of abnormal brain in MRI via improved AlexNet and ELM optimized by chaotic bat algorithm

Computer-aided diagnosis system is becoming a more and more important tool in clinical treatment, which can provide a verification of the doctors’ decisions. In this paper, we proposed a novel abnormal brain detection method for magnetic resonance image. Firstly, a pre-trained AlexNet was modified with batch normalization layers and trained on our brain images. Then, the last several layers were replaced with an extreme learning machine. A searching method was proposed to find the best number of layers to be replaced. Finally, the extreme learning machine was optimized by chaotic bat algorithm to obtain better classification performance. Experiment results based on 5 × hold-out validation revealed that our method achieved state-of-the-art performance.

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

[2]  Robert P. W. Duin,et al.  Feedforward neural networks with random weights , 1992, Proceedings., 11th IAPR International Conference on Pattern Recognition. Vol.II. Conference B: Pattern Recognition Methodology and Systems.

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

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

[5]  Nilesh Bhaskarrao Bahadure,et al.  Comparative Approach of MRI-Based Brain Tumor Segmentation and Classification Using Genetic Algorithm , 2018, Journal of Digital Imaging.

[6]  Zhiping Lin,et al.  Extreme Learning Machine for Joint Embedding and Clustering , 2018, Neurocomputing.

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

[8]  Ankit Chaudhary,et al.  Bit mask-oriented genetic algorithm for grammatical inference and premature convergence , 2018, International Journal of Bio-Inspired Computation (IJBIC).

[9]  Xue-cun Yang,et al.  Pressure Prediction of Coal Slurry Transportation Pipeline Based on Particle Swarm Optimization Kernel Function Extreme Learning Machine , 2015 .

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

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

[12]  Feilong Cao,et al.  A study on effectiveness of extreme learning machine , 2011, Neurocomputing.

[13]  Dejan J. Sobajic,et al.  Learning and generalization characteristics of the random vector Functional-link net , 1994, Neurocomputing.

[14]  Yan Liu,et al.  Deep object recognition across domains based on adaptive extreme learning machine , 2017, Neurocomputing.

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

[16]  Sung Wook Baik,et al.  Multi-grade brain tumor classification using deep CNN with extensive data augmentation , 2019, J. Comput. Sci..

[17]  Ankit Chaudhary,et al.  Grammar induction using bit masking oriented genetic algorithm and comparative analysis , 2016, Appl. Soft Comput..

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

[19]  Zhihai Lu,et al.  Pathological brain detection based on AlexNet and transfer learning , 2019, J. Comput. Sci..

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

[21]  Akugbe Martins Arasomwan,et al.  An Investigation into the Performance of Particle Swarm Optimization with Various Chaotic Maps , 2014 .

[22]  Li Cheng,et al.  A New Metaheuristic Bat-Inspired Algorithm , 2010 .

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

[24]  Giancarlo Mauri,et al.  Combining Noise-to-Image and Image-to-Image GANs: Brain MR Image Augmentation for Tumor Detection , 2019, IEEE Access.

[25]  Chee Kheong Siew,et al.  Extreme learning machine: Theory and applications , 2006, Neurocomputing.

[26]  Hongming Zhou,et al.  Optimization method based extreme learning machine for classification , 2010, Neurocomputing.

[27]  Xin-She Yang,et al.  A New Metaheuristic Bat-Inspired Algorithm , 2010, NICSO.

[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]  U Rajendra Acharya,et al.  Automated Detection of Alzheimer’s Disease Using Brain MRI Images– A Study with Various Feature Extraction Techniques , 2019, Journal of Medical Systems.

[30]  Graham Kendall,et al.  Maintaining regularity and generalization in data using the minimum description length principle and genetic algorithm: Case of grammatical inference , 2016, Swarm Evol. Comput..

[31]  Ankit Chaudhary,et al.  A comparative review of approaches to prevent premature convergence in GA , 2014, Appl. Soft Comput..

[32]  Guang-Bin Huang,et al.  Extreme Learning Machine for Multilayer Perceptron , 2016, IEEE Transactions on Neural Networks and Learning Systems.

[33]  David Windridge,et al.  A Comprehensive Classification of Deep Learning Libraries , 2018, Advances in Intelligent Systems and Computing.

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

[35]  Guanghui Sun,et al.  New Developments in Sliding Mode Control and Its Applications , 2014 .

[36]  Yudong Zhang,et al.  A Pathological Brain Detection System Based on Radial Basis Function Neural Network , 2016 .