| Literature DB >> 36236674 |
Naeem Ullah1, Mohammad Sohail Khan2, Javed Ali Khan3, Ahyoung Choi4, Muhammad Shahid Anwar4.
Abstract
Detection of a brain tumor in the early stages is critical for clinical practice and survival rate. Brain tumors arise in multiple shapes, sizes, and features with various treatment options. Tumor detection manually is challenging, time-consuming, and prone to error. Magnetic resonance imaging (MRI) scans are mostly used for tumor detection due to their non-invasive properties and also avoid painful biopsy. MRI scanning of one patient's brain generates many 3D images from multiple directions, making the manual detection of tumors very difficult, error-prone, and time-consuming. Therefore, there is a considerable need for autonomous diagnostics tools to detect brain tumors accurately. In this research, we have presented a novel TumorResnet deep learning (DL) model for brain detection, i.e., binary classification. The TumorResNet model employs 20 convolution layers with a leaky ReLU (LReLU) activation function for feature map activation to compute the most distinctive deep features. Finally, three fully connected classification layers are used to classify brain tumors MRI into normal and tumorous. The performance of the proposed TumorResNet architecture is evaluated on a standard Kaggle brain tumor MRI dataset for brain tumor detection (BTD), which contains brain tumor and normal MR images. The proposed model achieved a good accuracy of 99.33% for BTD. These experimental results, including the cross-dataset setting, validate the superiority of the TumorResNet model over the contemporary frameworks. This study offers an automated BTD method that aids in the early diagnosis of brain cancers. This procedure has a substantial impact on improving treatment options and patient survival.Entities:
Keywords: MRI; TumorResNet; brain tumor detection; deep learning
Mesh:
Year: 2022 PMID: 36236674 PMCID: PMC9570935 DOI: 10.3390/s22197575
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.847
Overview of existing approaches for BTD.
| Author | Method | Images Details of the Dataset | Advantages | Limitation |
|---|---|---|---|---|
| Woldeyohannes et al. [ | Two-dimensional discrete wavelet transforms (2D-DWT) are used for feature extraction and SVM for classification. | 160 normal and 240 tumorous MRI | Satisfactory results on a small dataset | Testing on imbalance dataset |
| Selvaraj et al. [ | Statistical features (mean & variance), features from gray level cooccurrence matrices (entropy & contrast), and Least Squares SVM. | 833 tumorous and 267 normal MRI | Both linear and non-linear kernels are used | Testing on imbalance dataset |
| Srilatha et al. [ | LBP for feature extraction and SVM for classification. | 58 normal MRI and 100 tumorous | Computationally efficient | Results are reported for a small dataset |
| Mishra et al. [ | Graph Attention AutoEncoder-CNN | 510 tumorous and 461 normal | Good generalization ability | Computationally complex |
| Rai et al. [ | A novel Less Layered and less complex U-Net CNN | 155 tumorous and 98 normal | The model is less complex and fast | Very low accuracy on uncropped images |
| Neeraja et al. [ | CNN | 155 tumorous and 98 normal | The model is lightweight and efficient | Low classification accuracy |
| Cinar et al. [ | Resnet50 | 155 tumorous and 98 normal | The model is efficient with good generalization ability | Low classification accuracy |
| Kiraz et al. [ | weighted KNN | 300 tumorous and 300 normal | The model performs well on the combined images of two datasets | Performance is highly influenced by the location and size of brain tumors |
Figure 1Flow diagram of the proposed approach for the BTD.
Differences between the Resnet18 and TumorResnet model.
| Property | Resnet18 | TumorResnet |
|---|---|---|
| Total learnable layers | 18 | 23 |
| Total convolutional layer | 17 | 20 |
| Total fully connected layers | 1 | 3 |
| No of dropouts (0.5%) | 0 | 2 |
| No of global average pooling | 1 | 0 |
| Activation function | Relu | LreLU |
The TumorResNet Architecture.
| S No | Layers | Filter | No of filters | Padding | Stride |
|---|---|---|---|---|---|
| 1 | Convolution (BN, LreLU) | 7 × 7 | 64 | 3 × 3 | 2 × 2 |
| 2 | Max-Pooling | 3 × 3 | 1 × 1 | 2 × 2 | |
| 3 | Convolution (BN, LreLU) | 3 × 3 | 64 | 1 × 1 | |
| 4 | Convolution (BN, LreLU) | 3 × 3 | 64 | 1 × 1 | |
| 5 | Convolution (BN, LreLU) | 3 × 3 | 64 | 1 × 1 | |
| 6 | Convolution (BN, LreLU) | 3 × 3 | 64 | 1 × 1 | |
| 7 | Convolution (BN, LreLU) | 3 × 3 | 128 | 1 × 1 | 2 × 2 |
| 8 | Convolution (BN) | 3 × 3 | 128 | 1 × 1 | |
| 9 | Convolution (BN, LreLU) | 1 × 1 | 128 | 2 × 2 | |
| 10 | Convolution (BN, LreLU) | 3 × 3 | 128 | 1 × 1 | |
| 11 | Convolution (BN, LreLU) | 3 × 3 | 128 | 1 × 1 | |
| 12 | Convolution (BN) | 1 × 1 | 256 | 2 × 2 | |
| 13 | Convolution (BN, LreLU) | 3 × 3 | 256 | 1 × 1 | 2 × 2 |
| 14 | Convolution (BN, LreLU) | 3 × 3 | 256 | 1 × 1 | |
| 15 | Convolution (BN, LreLU) | 3 × 3 | 256 | 1 × 1 | |
| 16 | Convolution (BN, LreLU) | 3 × 3 | 256 | 1 × 1 | |
| 17 | Convolution (BN, LreLU) | 3 × 3 | 512 | 1 × 1 | 2 × 2 |
| 18 | Convolution (BN) | 3 × 3 | 512 | 1 × 1 | |
| 19 | Convolution (BN, LreLU) | 1 × 1 | 512 | 2 × 2 | |
| 20 | Convolution (BN, LreLU) | 3 × 3 | 512 | 1 × 1 | |
| 21 | Convolution (BN) | 3 × 3 | 512 | 1 × 1 | |
| 22 | FC + LreLU + Dropout | ||||
| 23 | FC + LreLU + Dropout | ||||
| 24 | FC + Softmax + Classification | ||||
Parameters of the proposed architecture.
| Parameter | Value |
|---|---|
| Optimization algorithm | SGD |
| Shuffle | Every epoch |
| Maximum Epochs | 30 |
| Iterations per epoch | 18 |
| Activation Function | LreLU |
| Validation frequency | 30 |
| Mini batch size | 133 |
| Verbose | false |
| learning rate | 0.01 |
| Dropout | 0.5 |
| Train Size | 0.8 |
| Test Size | 0.2 |
Figure 2Samples of BTD-MRI dataset, upper row: No tumor examples and lower row: tumorous images examples.
Details of the system used for implementation.
| Sr. No | Name | Experiment Parameters |
|---|---|---|
| 1 | CPU | IntelI Core I i5-5200U |
| 2 | System type | Windows 10, 64 bit |
| 3 | Development tool | MATLAB R2020a |
| 4 | RAM | 8 GB |
| 5 | ROM | 500 GB |
Figure 3Training and testing accuracy of TumorResNet framework (blue line represents training accuracy, the black line represents testing accuracy, and the red line represents the training loss wheras the black line in the loss section represents testing loss).
Confusion matrix of the TumorResNet framework.
| Tumor | Normal | |
|---|---|---|
| Tumor | 300 | 0 |
| Healthy or Normal | 4 | 296 |
Figure 4ROC plot of the proposed TumorResNet framework.
Changing the network to evaluate the ablation study.
| Experiment No | Activation Function | FC Layers | GAP | Dropout layers | Accuracy | Findings |
|---|---|---|---|---|---|---|
| Experiment 1 | Relu | 1 | 1 | 0 | 99.0 | Accuracy dropped |
| experiment 2 | LReLU | 1 | 0 | 0 | 98.17 | Accuracy dropped |
| Proposed method | LReLU | 3 | 0 | 2 | 99.33 | Best accuracy |
TumorResNet comparison with hybrid approaches.
| DL Model | Accuracy | Precision | Recall | Specificity | F1-Score |
|---|---|---|---|---|---|
| Shufflenet | 98.67 | 99 | 99 | 99.66 | 99 |
| Mobilenetv2 | 98.33 | 98 | 98 | 99.32 | 98 |
| Resnet18 | 96.33 | 96 | 96 | 97.28 | 96 |
| Darknet19 | 98.67 | 98.5 | 98.5 | 98.34 | 98.5 |
| Squeezenet | 99.17 | 99.5 | 99.5 | 99.66 | 99.5 |
| Alexnet | 98.17 | 98.5 | 98.5 | 99.66 | 98.5 |
| Proposed TumorResNet | 99.33 | 99.5 | 99.5 | 100 | 99.5 |
Confusion matrix of the TumorResNet framework for benign and malignant BTD.
| Predicted Class | |||
|---|---|---|---|
| Tumor | Normal | ||
| Actual Class | Malignant | 70 | 0 |
| Benign | 4 | 66 | |
Figure 5ROC plot of the proposed TumorResNet framework for benign and malignant BTD.
Comparison of the proposed work with existing methods.
| Work | Method | Dataset | Accuracy | AUC | Date |
|---|---|---|---|---|---|
| Nayak et al. [ | Spectral Data Augmentation-based Deep Autoencoder | BTD-MRI dataset | 97% | 0.9946 | 2022 |
| Lamrani et al. [ | CNN | BTD-MRI dataset | 96% | 0.96 | 2022 |
| Proposed work | TumorResNet | BTD-MRI dataset | 99.33 | 0.9997 | 2022 |