Comparative Analysis of Different Deep Convolutional Neural Network Architectures for Classification of Brain Tumor on Magnetic Resonance Images

被引:3
作者
Sachdeva, Jainy [1 ]
Sharma, Deepanshu [2 ]
Ahuja, Chirag Kamal [2 ]
机构
[1] TIET, Dept Elect & Instrumentat Engn, Patiala, India
[2] PGIMER, Dept Radiodiag & Imaging, Chandigarh, India
关键词
COMPUTER-AIDED DIAGNOSIS; FEATURE-EXTRACTION; MRI IMAGES; SEGMENTATION; OPTIMIZATION; REDUCTION; MACHINE; SVM;
D O I
10.1007/s11831-023-10041-y
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
In the current study, the capability of pre-trained Deep Convolutional Neural Network (DCNN) by ImageNet features is proposed for categorization of brain tumors by utilizing MR images. The pre-trained models like ResNet50, InceptionV3, Xception, DenseNet121, MobileNetV3Large, EffcientNetB0, EfficientNetV2L, EfficientNetV2B0 have been exploited for classification purpose. The selection criteria is based upon the diverse proficiency each model depicts. For e.g., EfficientNetB0, ResNet50, MobileNet, Xception employ lower number of training parameters that makes them time efficient whereas, VGG16 though has higher number of training parameters, while increasing the training time without compromising with accuracy. The AlexNet on the other hand also has a reduced number of parameters in comparison to Google's Inception module which is more memory efficient. AlexNet takes more memory for training. The EfficientNet's are based on inverted residual blocks of MobileNetV2 with addition to Squeeze and Excitation blocks which makes them highly efficient for feature extraction. Therefore, in this research study, the above stated pre-trained models are used in by retaining the ReLU as an activation function due to its unbounded nature which also helps the model from Vanishing Gradient problem. By hyper tuning the top layers of different pre-trained models by adding Global Average Pooling layer and Dropout layer along with Fully Connected layer with classifier as SoftMax increases the overall efficiency and also reduces overfitting. The proposed comparative study shows the working of different pre-trained DCNN models for classifying brain tumors. The proposed experimental study is performed on three different databases i.e., Kaggle, BraTS 2018, and Real time dataset acquired from PGIMER and comparison analysis. The comparison analysis with existing methods as well as statistical analysis is also performed. It is observed from the results that EfficientNetB0 has outperformed all the existing methods. The pre-trained EfficientNetB0 architecture by hyper tuning the parameters, the testing accuracy has increased by 2.14% on Kaggle and increased the classification accuracy by 3.98% on BraTS dataset. On Real time dataset comprising of Glioblastoma Multiforme and Oligodendroglioma, highest accuracy of 97.32% is achieved among all other models.
引用
收藏
页码:1959 / 1978
页数:20
相关论文
共 62 条
  • [1] Abd-Ellah MK, 2016, INT C MICROELECTRON, P73, DOI 10.1109/ICM.2016.7847911
  • [2] Afshar P, 2019, INT CONF ACOUST SPEE, P1368, DOI 10.1109/ICASSP.2019.8683759
  • [3] Brain Tumor/Mass Classification Framework Using Magnetic-Resonance-Imaging-Based Isolated and Developed Transfer Deep-Learning Model
    Alanazi, Muhannad Faleh
    Ali, Muhammad Umair
    Hussain, Shaik Javeed
    Zafar, Amad
    Mohatram, Mohammed
    Irfan, Muhammad
    AlRuwaili, Raed
    Alruwaili, Mubarak
    Ali, Naif H.
    Albarrak, Anas Mohammad
    [J]. SENSORS, 2022, 22 (01)
  • [4] Alfonse A., 2016, EGYPTIAN COMPUTER SC, V4, P11
  • [5] Development of computer-aided approach for brain tumor detection using random forest classifier
    Anitha, R.
    Raja, D. Siva Sundhara
    [J]. INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2018, 28 (01) : 48 - 53
  • [6] [Anonymous], 2018, Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challenge, DOI DOI 10.17863/CAM.38755
  • [7] Classification of Brain Tumors from MRI Images Using a Convolutional Neural Network
    Badza, Milica M.
    Barjaktarovic, Marko C.
    [J]. APPLIED SCIENCES-BASEL, 2020, 10 (06):
  • [8] Comparative Approach of MRI-Based Brain Tumor Segmentation and Classification Using Genetic Algorithm
    Bahadure, Nilesh Bhaskarrao
    Ray, Arun Kumar
    Thethi, Har Pal
    [J]. JOURNAL OF DIGITAL IMAGING, 2018, 31 (04) : 477 - 489
  • [9] Data Descriptor: Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features
    Bakas, Spyridon
    Akbari, Hamed
    Sotiras, Aristeidis
    Bilello, Michel
    Rozycki, Martin
    Kirby, Justin S.
    Freymann, John B.
    Farahani, Keyvan
    Davatzikos, Christos
    [J]. SCIENTIFIC DATA, 2017, 4
  • [10] Bhuvaji S., 2020, Brain Tumor Classification (MRI) Data Set, P2020, DOI [10 . 34740 / kaggle / dsv / 1183165, DOI 10.34740/KAGGLE/DSV/1183165, 10.34740/KAGGLE/DSV/1183165]