| Literature DB >> 35885437 |
Kwok Tai Chui1, Brij B Gupta2,3,4, Wadee Alhalabi5,6, Fatma Salih Alzahrani7.
Abstract
Alzheimer's disease (AD) is the most common type (>60%) of dementia and can wreak havoc on the psychological and physiological development of sufferers and their carers, as well as the economic and social development. Attributed to the shortage of medical staff, automatic diagnosis of AD has become more important to relieve the workload of medical staff and increase the accuracy of medical diagnoses. Using the common MRI scans as inputs, an AD detection model has been designed using convolutional neural network (CNN). To enhance the fine-tuning of hyperparameters and, thus, the detection accuracy, transfer learning (TL) is introduced, which brings the domain knowledge from heterogeneous datasets. Generative adversarial network (GAN) is applied to generate additional training data in the minority classes of the benchmark datasets. Performance evaluation and analysis using three benchmark (OASIS-series) datasets revealed the effectiveness of the proposed method, which increases the accuracy of the detection model by 2.85-3.88%, 2.43-2.66%, and 1.8-40.1% in the ablation study of GAN and TL, as well as the comparison with existing works, respectively.Entities:
Keywords: Alzheimer’s disease; MRI scans; automatic diagnosis; convolutional neural network; deep learning; dementia; generative adversarial network; imbalanced dataset; transfer learning
Year: 2022 PMID: 35885437 PMCID: PMC9318866 DOI: 10.3390/diagnostics12071531
Source DB: PubMed Journal: Diagnostics (Basel) ISSN: 2075-4418
Constituents of OASIS-1, OASIS-2, and OASIS-3.
| OASIS-1 | OASIS-2 | OASIS-3 | ||
|---|---|---|---|---|
| Number of participants | 416 | 150 | 1098 | |
| Sample size of class label | Class 0: Normal | 336 | 206 | 1210 |
| Class 1: very mild AD | 70 | 123 | 516 | |
| Class 2: Mild AD | 28 | 41 | 262 | |
| Class 3: Moderate AD | 2 | 3 | 180 | |
Imbalanced ratios across different classes of OASIS-1, OASIS-2, and OASIS-3.
| Imbalanced Ratio | |||
|---|---|---|---|
| Class Label | OASIS-1 | OASIS-2 | OASIS-3 |
| Class 0: Normal | N/A | N/A | N/A |
| Class 1: very mild AD | 4.8:1 | 1.67:1 | 2.34:1 |
| Class 2: Mild AD | 12:1 | 5.02:1 | 4.62:1 |
| Class 3: Moderate AD | 168:1 | 68.7:1 | 6.72:1 |
Figure 1Conceptual diagram of the proposed GAN-CNN-TL algorithm for AD detection.
Figure 2General architecture of GAN.
Figure 3Feature extraction and AD detection using CNN.
Figure 4Two-tier transfer learning process with domain knowledge transfer to model for OASIS-1 by OASIS-2 and OASIS-3.
Performance evaluation of the CNN-TL for OASIS-1, OASIS-2, and OASIS-3.
| Model | Accuracy of a Single Class (%) | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|---|
| CNN-TLOASIS-1 | Class 0: 94.6 | 91.0 | 94.6 | 93.8 |
| Class 1: 92.9 | ||||
| Class 2: 89.3 | ||||
| Class 3: 50 | ||||
| CNN-TLOASIS-2 | Class 0: 93.7 | 91.6 | 93.7 | 92.8 |
| Class 1: 93.5 | ||||
| Class 2: 87.8 | ||||
| Class 3: 66.7 | ||||
| CNN-TLOASIS-3 | Class 0: 95.3 | 94.1 | 95.3 | 94.8 |
| Class 1: 94.6 | ||||
| Class 2: 93.5 | ||||
| Class 3: 92.8 |
Figure 5Confusion matrices of the CNN-TLOASIS-1, CNN-TLOASIS-2, and CNN-TLOASIS-3.
Performance evaluation of the GAN-CNN for OASIS-1, OASIS-2, and OASIS-3.
| Model | Accuracy of a Single Class (%) | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|---|
| GAN-CNNOASIS-1 | Class 0: 95.2 | 93.4 | 95.2 | 94.6 |
| Class 1: 94.3 | ||||
| Class 2: 92.9 | ||||
| Class 3: 90 | ||||
| GAN-CNNOASIS-2 | Class 0: 94.7 | 93.1 | 94.7 | 93.8 |
| Class 1: 93.5 | ||||
| Class 2: 92.7 | ||||
| Class 3: 93.3 | ||||
| GAN-CNNOASIS-3 | Class 0: 95.7 | 94.6 | 95.7 | 95.1 |
| Class 1: 95.1 | ||||
| Class 2: 94.6 | ||||
| Class 3: 94.2 |
Figure 6Confusion matrices of the GAN-CNNOASIS-1, GAN-CNNOASIS-2, and GAN-CNNOASIS-3.
Performance evaluation of the GAN-CNN-TL for OASIS-1, OASIS-2, and OASIS-3.
| Model | Accuracy of a Single Class (%) | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|---|
| GAN-CNN-TLOASIS-1 | Class 0: 97.3 | 96.0 | 97.3 | 96.9 |
| Class 1: 97.1 | ||||
| Class 2: 95.8 | ||||
| Class 3: 90 | ||||
| GAN-CNN-TLOASIS-2 | Class 0: 96.6 | 95.8 | 96.6 | 96.1 |
| Class 1: 95.9 | ||||
| Class 2: 95.9 | ||||
| Class 3: 93.3 | ||||
| GAN-CNN-TLOASIS-3 | Class 0: 97.9 | 97.3 | 97.9 | 97.5 |
| Class 1: 97.5 | ||||
| Class 2: 97.3 | ||||
| Class 3: 97.0 |
Figure 7Confusion matrices of the GAN-CNN-TLOASIS-1, GAN-CNN-TLOASIS-2, and GAN-CNN-TLOASIS-3.
Performance comparison between our work and existing works.
| Work | Dataset | Class and Sample Size | Features | Algorithms | Type of Cross-Validation | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|---|---|---|---|---|
| [ | OASIS-1 | Healthy: 316 | Gradient boosted random forest | ResNet-50 | 10-fold (with an inappropriate 80:20 ratio) | N/A | N/A | 98.99 |
| [ | OASIS-1 | Healthy: 336 | BrainNet3D | 5-fold | N/A | N/A | 80 | |
| [ | OASIS-1 | Healthy: 41 | Adversarial autoencoder | No | 67 | 78 | 72.8 | |
| [ | OASIS-1 | Healthy: 100 | M-Net-axial_32 | 5-fold | N/A | N/A | 74.9 | |
|
| OASIS-1 | Healthy: 336 | GAN-CNN-TL | 2-fold | 96 | 97.2 | 96.8 | |
| [ | OASIS-2 | Healthy: 206 | Boruta | Deep neural network | No | 88.2 | 100 | 94.7 |
| [ | OASIS-2 | Healthy: 72 | CNN | No | N/A | N/A | 97 | |
| [ | OASIS-2 | Healthy: 206 | Subject ID, clinical dementia ratio, mini-mental state examination, age, magnetic resonance delay, and normalized whole brain volume | SVM | No | N/A | N/A | 68.8 |
| [ | OASIS-2 | Healthy: 41 | Voxel-size independent neural network | No | N/A | N/A | 88.2 | |
|
| OASIS-2 | Healthy: 206 | GAN-CNN-TL | 3-fold | 96.1 | 96.8 | 96.4 | |
| [ | OASIS-3 | Healthy: 100 | Gray level co-occurrence matrix and CNN | No | N/A | N/A | 90.3 | |
| [ | OASIS-3 | Healthy: 1210 | Ensemble learning of Inception-v3, DenseNet121, ResNet50, and ResNet18 | No | 83.5 | 91.4 | 87.9 | |
| [ | OASIS-3 | Healthy: 1210 | vertex-based graph-CNN | RNN | No | N/A | N/A | 82.6 |
| [ | OASIS-3 | Healthy: 1210 | Deep convolutional generative adversarial network | No | N/A | N/A | 74.4 | |
|
| OASIS-3 | Healthy: 1210 | GAN-CNN-TL | 5-fold | 97.3 | 97.9 | 97.5 | |