| Literature DB >> 35009911 |
Muhannad Faleh Alanazi1, Muhammad Umair Ali2, Shaik Javeed Hussain3, Amad Zafar4, Mohammed Mohatram3, Muhammad Irfan5, Raed AlRuwaili1, Mubarak Alruwaili1, Naif H Ali6, Anas Mohammad Albarrak7.
Abstract
With the advancement in technology, machine learning can be applied to diagnose the mass/tumor in the brain using magnetic resonance imaging (MRI). This work proposes a novel developed transfer deep-learning model for the early diagnosis of brain tumors into their subclasses, such as pituitary, meningioma, and glioma. First, various layers of isolated convolutional-neural-network (CNN) models are built from scratch to check their performances for brain MRI images. Then, the 22-layer, binary-classification (tumor or no tumor) isolated-CNN model is re-utilized to re-adjust the neurons' weights for classifying brain MRI images into tumor subclasses using the transfer-learning concept. As a result, the developed transfer-learned model has a high accuracy of 95.75% for the MRI images of the same MRI machine. Furthermore, the developed transfer-learned model has also been tested using the brain MRI images of another machine to validate its adaptability, general capability, and reliability for real-time application in the future. The results showed that the proposed model has a high accuracy of 96.89% for an unseen brain MRI dataset. Thus, the proposed deep-learning framework can help doctors and radiologists diagnose brain tumors early.Entities:
Keywords: brain MRI images; brain mass; brain tumor; deep-learning model; tumor classification
Mesh:
Year: 2022 PMID: 35009911 PMCID: PMC8749789 DOI: 10.3390/s22010372
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
Classification of brain MRI images.
| No Tumor | Glioma Tumor | Meningioma Tumor | Pituitary Tumor | |
|---|---|---|---|---|
| Brain MRI Images |
|
|
|
|
Information related to the 22-layer CNN developed from scratch.
| Layer No. | Layer Type | Properties | Learnable |
|---|---|---|---|
| 1 | Image Input | 227 × 227 × 3 images with ‘zerocenter’ normalization | - |
| 2 | Convolutional | 128 6 × 6 convolutions with stride [4 4] and padding [0 0 0 0] | Weights: 6 × 6 × 3 × 128 |
| 3 | ReLU | ReLU | - |
| 4 | Cross Channel Normalization | cross channel normalization with 5 channels per element | - |
| 5 | Max | 2 × 2 max pooling with stride [2 2] and padding [0 0 0 0] | - |
| 6 | Convolutional | 96 6 × 6 convolutions with stride [1 1] and padding [2 2 2 2] | Weights: 6 × 6 × 3 × 128 × 96 |
| 7 | ReLU | ReLU | - |
| 8 | Max | 2 × 2 max pooling with stride [2 2] and padding [0 0 0 0] | - |
| 9 | Convolutional | 96 2 × 2 convolutions with stride [1 1] and padding [2 2 2 2] | Weights: 2 × 2 × 96 × 96 |
| 10 | ReLU | ReLU | - |
| 11 | Max | 2 × 2 max pooling with stride [2 2] and padding [0 0 0 0] | - |
| 12 | Convolutional | 24 6 × 6 convolutions with stride [1 1] and padding [2 2 2 2] | Weights: 6 × 6 × 96 × 24 |
| 13 | ReLU | ReLU | - |
| 14 | Max | 2 × 2 max pooling with stride [2 2] and padding [0 0 0 0] | - |
| 15 | Convolutional | 24 6 × 6 convolutions with stride [1 1] and padding [2 2 2 2] | Weights: 2 × 2 × 24 × 24 |
| 16 | ReLU | ReLU | - |
| 17 | Batch Normalization | Batch normalization | Offset: 1 × 1 × 24 |
| 18 | Fully | 512 fully connected layer | Weights: 512 × 96 |
| 19 | Dropout | 30% dropout | - |
| 20 | Fully | 2 fully connected layer | Weights: 2 × 512 |
| 21 | Softmax | - | - |
| 22 | Classification Output | - | - |
Figure 1The architecture of the isolated-CNN model was built from scratch.
Figure 2Process of transfer learning for brain-image classification.
Figure 3The weights update/learning process of convolutional neural network.
Figure 4The framework of the proposed approach.
Comparison of isolated-CNN models for binary-class classification (tumor and no tumor) using dataset I.
| Network | Training Accuracy (%) | Training Loss | Training Time | Validation Accuracy (%) | Validation Loss |
|---|---|---|---|---|---|
| 19-layers | 100 | 2.8016 × 10−5 | 15 min 58 s | 98.50 | 0.0850 |
| 22-layers | 100 | 4.8811 × 10−6 | 16 min | 99.33 | 0.0534 |
| 25-layers | 100 | 4.8243 × 10−7 | 15 min 43 s | 98.33 | 0.1412 |
Comparison of isolated-CNN models for four-class classification using dataset II.
| Network | Training Accuracy (%) | Training Loss | Training Time | Validation Accuracy (%) | Validation Loss |
|---|---|---|---|---|---|
| 19-layers | 100 | 1.9454 × 10−4 | 14 min 57 s | 91.27 | 0.4637 |
| 22-layers | 100 | 2.7508 × 10−5 | 14 min 35 s | 92.67 | 0.3208 |
| 25-layers | 100 | 1.6904 × 10−6 | 14 min 22 s | 91.62 | 0.7276 |
Figure 5Comparison of different isolated-CNN models for dataset-I; (a) Training-accuracy curves; (b) Training-loss curves; (c) Validation-accuracy curves.
Figure 6Comparison of different isolated-CNN models for dataset-II; (a) Training-accuracy curves; (b) Training-loss curves; (c) Validation-accuracy curves.
Figure 7Results of testing of developed transfer-learned model for dataset-III.
Detection of type of brain tumor using developed transfer-learned network.
| Class | Classified as | TPR | FNR | PPV | FDR | Training Time | Validation | ||
|---|---|---|---|---|---|---|---|---|---|
| Glioma | Meningioma | Pituitary | |||||||
|
| 157 | 6 | 0 | 96.32 | 3.68 | 95.15 | 4.85 | 13 min 8 s | 95.75% |
|
| 8 | 151 | 0 | 94.97 | 5.03 | 92.07 | 7.93 | ||
|
| 0 | 7 | 165 | 95.93 | 4.07 | 100 | 0 | ||
Figure 8Receiver operation characteristic (ROC) curves for dataset-III; (a) glioma; (b) meningioma; (c) pituitary.
Performance comparison of the proposed model with literature.
| Study | Type of Dataset | Model | Accuracy (%) | Training Time |
|---|---|---|---|---|
| Abiwinanda et al. [ | Dataset-III | 13-layer CNN | 84.19 | - |
| Irmak. [ | Dataset-II | 25-layer CNN | 92.66 | - |
| Kang et al. [ | Dataset-III | Pre-trained CNN models with machine-learning classifiers | 93.72 | - |
| Rehman et al. [ | Dataset-III | AlexNet | 95.86 | 43 min |
| Proposed | Dataset-II | Developed transfer-learned CNN | 95.75 | 13 min |