State-of-the-Art CNN Optimizer for Brain Tumor Segmentation in Magnetic Resonance Images

Brain tumors have become a leading cause of death around the globe. The main reason for this epidemic is the difficulty conducting a timely diagnosis of the tumor. Fortunately, magnetic resonance images (MRI) are utilized to diagnose tumors in most cases. The performance of a Convolutional Neural Network (CNN) depends on many factors (i.e., weight initialization, optimization, batches and epochs, learning rate, activation function, loss function, and network topology), data quality, and specific combinations of these model attributes. When we deal with a segmentation or classification problem, utilizing a single optimizer is considered weak testing or validity unless the decision of the selection of an optimizer is backed up by a strong argument. Therefore, optimizer selection processes are considered important to validate the usage of a single optimizer in order to attain these decision problems. In this paper, we provides a comprehensive comparative analysis of popular optimizers of CNN to benchmark the segmentation for improvement. In detail, we perform a comparative analysis of 10 different state-of-the-art gradient descent-based optimizers, namely Adaptive Gradient (Adagrad), Adaptive Delta (AdaDelta), Stochastic Gradient Descent (SGD), Adaptive Momentum (Adam), Cyclic Learning Rate (CLR), Adaptive Max Pooling (Adamax), Root Mean Square Propagation (RMS Prop), Nesterov Adaptive Momentum (Nadam), and Nesterov accelerated gradient (NAG) for CNN. The experiments were performed on the BraTS2015 data set. The Adam optimizer had the best accuracy of 99.2% in enhancing the CNN ability in classification and segmentation.

[1]  Kestrilia R. Prilianti,et al.  Performance comparison of the convolutional neural network optimizer for photosynthetic pigments prediction on plant digital image , 2019 .

[2]  Tao Yang,et al.  Semantic segmentation via highly fused convolutional network with multiple soft cost functions , 2019, Cognitive Systems Research.

[3]  J. Barnholtz-Sloan,et al.  Malignant primary brain and other central nervous system tumors diagnosed in Canada from 2009 to 2013 , 2019, Neuro-oncology.

[4]  Jongtae Rhee,et al.  A Personalized Healthcare Monitoring System for Diabetic Patients by Utilizing BLE-Based Sensors and Real-Time Data Processing , 2018, Sensors.

[5]  Brian B. Avants,et al.  The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS) , 2015, IEEE Transactions on Medical Imaging.

[6]  Han Zhang,et al.  Research on a learning rate with energy index in deep learning , 2019, Neural Networks.

[7]  Ling Huang,et al.  Optimization of deep convolutional neural network for large scale image retrieval , 2018, Neurocomputing.

[8]  Buket D. Barkana,et al.  Performance analysis of descriptive statistical features in retinal vessel segmentation via fuzzy logic, ANN, SVM, and classifier fusion , 2017, Knowl. Based Syst..

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

[10]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[11]  Kunihiko Fukushima,et al.  Neocognitron: A hierarchical neural network capable of visual pattern recognition , 1988, Neural Networks.

[12]  Muazzam Maqsood,et al.  A Deep Siamese Convolution Neural Network for Multi-Class Classification of Alzheimer Disease , 2020, Brain sciences.

[13]  Jianfeng Lu,et al.  Content-Based Brain Tumor Retrieval for MR Images Using Transfer Learning , 2019, IEEE Access.

[14]  Edison Pignaton de Freitas,et al.  NoSQL real-time database performance comparison , 2018, Int. J. Parallel Emergent Distributed Syst..

[15]  B. Chandra,et al.  Deep learning with adaptive learning rate using laplacian score , 2016, Expert Syst. Appl..

[16]  Rolf Findeisen,et al.  A fast gradient method for embedded linear predictive control , 2011 .

[17]  Mohammad Yusri Hassan,et al.  A review on applications of ANN and SVM for building electrical energy consumption forecasting , 2014 .

[18]  Amjad Rehman,et al.  An evolution based hybrid approach for heart diseases classification and associated risk factors identification , 2017 .

[19]  Geoffrey E. Hinton,et al.  Deep Learning , 2015, Nature.

[20]  Shengqi Yang,et al.  Type 2 diabetes mellitus prediction model based on data mining , 2018 .

[21]  Alexander Knaack,et al.  APPLICATIONS OF DEEP LEARNING TO BRAIN SEGMENTATION AND LABELING OF MRI BRAIN STRUCTURES , 2020 .

[22]  Ali Wali,et al.  Deep Multi-Scale 3D Convolutional Neural Network (CNN) for MRI Gliomas Brain Tumor Classification , 2020, Journal of Digital Imaging.

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

[24]  John M. Pecarina,et al.  Improving optimization of convolutional neural networks through parameter fine-tuning , 2017, Neural Computing and Applications.

[25]  Richard K. G. Do,et al.  Convolutional neural networks: an overview and application in radiology , 2018, Insights into Imaging.

[26]  Jun-Ho Huh,et al.  Big Data Analysis for Personalized Health Activities: Machine Learning Processing for Automatic Keyword Extraction Approach , 2018, Symmetry.

[27]  Amanda M Seed,et al.  Cognitive adaptations of social bonding in birds , 2007, Philosophical Transactions of the Royal Society B: Biological Sciences.

[28]  Christopher Joseph Pal,et al.  Brain tumor segmentation with Deep Neural Networks , 2015, Medical Image Anal..

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

[30]  Lei Zhang,et al.  Canny edge detection enhancement by scale multiplication , 2005, IEEE Transactions on Pattern Analysis and Machine Intelligence.