| Literature DB >> 35983526 |
Jingyao Liu1,2, Jiashi Zhao1,3, Liyuan Zhang1,3, Yu Miao1,3, Wei He1,3, Weili Shi1,3, Yanfang Li1,3, Bai Ji4, Ke Zhang1,3, Zhengang Jiang1,3.
Abstract
COVID-19 has become the largest public health event worldwide since its outbreak, and early detection is a prerequisite for effective treatment. Chest X-ray images have become an important basis for screening and monitoring the disease, and deep learning has shown great potential for this task. Many studies have proposed deep learning methods for automated diagnosis of COVID-19. Although these methods have achieved excellent performance in terms of detection, most have been evaluated using limited datasets and typically use a single deep learning network to extract features. To this end, the dual asymmetric feature learning network (DAFLNet) is proposed, which is divided into two modules, DAFFM and WDFM. DAFFM mainly comprises the backbone networks EfficientNetV2 and DenseNet for feature fusion. WDFM is mainly for weighted decision-level fusion and features a new pretrained network selection algorithm (PNSA) for determination of the optimal weights. Experiments on a large dataset were conducted using two schemes, DAFLNet-1 and DAFLNet-2, and both schemes outperformed eight state-of-the-art classification techniques in terms of classification performance. DAFLNet-1 achieved an average accuracy of up to 98.56% for the triple classification of COVID-19, pneumonia, and healthy images.Entities:
Mesh:
Year: 2022 PMID: 35983526 PMCID: PMC9381197 DOI: 10.1155/2022/3836498
Source DB: PubMed Journal: Comput Math Methods Med ISSN: 1748-670X Impact factor: 2.809
Figure 1Sample images of the (a) COVID-19, (b) normal, and (c) pneumonia chest category X-rays.
Data distribution in the model.
| Dataset | Test (20%) | Training (70%) | Validation (10%) | Total |
|---|---|---|---|---|
| COVID-19 | 256 | 897 | 128 | 1281 |
| Normal | 654 | 2289 | 327 | 3270 |
| Pneumonia | 931 | 3260 | 466 | 4657 |
| Total | 1841 | 6446 | 921 | 9208 |
Figure 2Structure of MBConv and Fused-MBConv.
Figure 3DenseNet structure.
Figure 4Concat for feature-level fusion.
Algorithm 1Proposed PNSA for weight selection.
Figure 5Structure of the proposed DAFLNet.
Algorithm 2Pseudocode of DAFLNet algorithm.
Hyperparameters of the DAFLNet network model.
| Hyperparameter | Value |
|---|---|
| LR | 0.0003 |
| BS | 16 |
| Epochs | 30 |
| Optimizer | Adam |
| DR | 0.4 |
Figure 6Samples from the original and the enhanced dataset.
Figure 7Comparison experiment of the concat and add methods.
Figure 8Average accuracy scores of DAFLNet-1 and DAFLNet-2 with different decision-level fusion weights.
Nine proposed networks.
| Index | Modules | Network name | Description |
|---|---|---|---|
| M1 | EfficientNetV2 | Backbone network | |
| M2 | DenseNet121 | Auxiliary network1 | |
| M3 | DenseNet169 | Auxiliary network2 | |
| M4 | ⟵FLF(M1, M2) | Eff-V2&D-121 | FLF of M1 and M2 |
| M5 | ⟵FLF(M1, M3) | Eff-V2&D-169 | FLF of M1 and M3 |
| M6 | ⟵M4 + CBAM | DAFFM-1 | Add CBAM to M4 |
| M7 | ⟵M5 + CBAM | DAFFM-2 | Add CBAM to M5 |
| M8 | ⟵DLF(M1, M2, M6) | DAFLNet-1 | DLF of M1, M2, and M6 |
| M9 | ⟵DLF(M1, M2, M7) | DAFLNet-1 | DLF of M1, M2, and M7 |
Best accuracy of nine networks in the test set (%).
| ID | Model | COVID-19 | Normal | Pneumonia | Overall Acc. |
|---|---|---|---|---|---|
| M1 | EfficientNetV2 | 98.4 | 97.50 | 97.56 | 97.23 |
| M2 | DenseNet121 | 98.08 | 97.45 | 97.18 | 96.85 |
| M3 | DenseNet169 | 97.91 | 97.12 | 97.12 | 96.58 |
| M4 | Eff-V2&D-121 | 98.35 | 97.94 | 98.15 | 97.61 |
| M5 | Eff-V2&D-169 | 99.13 | 97.77 | 97.77 | 97.34 |
| M6 | DAFFM-1 | 99.13 | 98.05 | 98.05 | 97.72 |
| M7 | DAFFM-2 | 99.51 | 98.53 | 98.48 | 98.26 |
| M8 | DAFLNet-1 | 99.62 | 99.08 | 99.02 | 98.86 |
| M9 | DAFLNet-2 | 99.62 | 99.13 | 98.86 | 98.81 |
Figure 9Optimal performance results of the nine networks.
Figure 10Classification results of DAFLNet-1 visualized using a confusion matrix.
Figure 11Classification results of DAFLNet-2 visualized using a confusion matrix.
Classification of DAFLNet networks after two kinds of validation (%).
| Models | Class | Acc (c) | Sen (c) | Pre (c) | Rec (c) | Spe (c) | F1-sc (c) | Overall Acc |
|---|---|---|---|---|---|---|---|---|
| DAFLNet-1 | COVID-19 | 99.61 ± 0.24 | 98.44 ± 1.35 | 98.75 ± 1.02 | 98.44 ± 1.35 | 99.80 ± 0.17 | 98.59 ± 0.86 | 98.33 ± 0.52 |
| Normal | 98.63 ± 0.39 | 97.55 ± 0.78 | 98.58 ± 0.47 | 97.55 ± 0.78 | 99.23 ± 0.26 | 98.06 ± 0.55 | ||
| Pneumonia | 98.41 ± 0.43 | 98.84 ± 0.33 | 98.04 ± 0.56 | 98.84 ± 0.33 | 97.98 ± 0.59 | 98.44 ± 0.42 | ||
| DAFLNet-2 | COVID-19 | 99.56 ± 0.23 | 98.44 ± 0.55 | 98.45 ± 1.43 | 98.44 ± 0.55 | 99.75 ± 0.24 | 98.44 ± 0.82 | 98.20 ± 0.55 |
| Normal | 98.52 ± 0.35 | 97.06 ± 0.88 | 98.76 ± 0.44 | 97.06 ± 0.89 | 99.33 ± 0.24 | 97.90 ± 0.50 | ||
| Pneumonia | 98.31 ± 0.54 | 98.93 ± 0.64 | 97.75 ± 0.61 | 98.93 ± 0.64 | 97.67 ± 0.63 | 98.34 ± 0.53 |
Performance comparison of the proposed DAFLNet with other studies (%).
| Method | Class | Sen (c) | Pre (c) | F1-sc (c) | Overall Acc |
|---|---|---|---|---|---|
| ECOVNet-EfficientNetB3 base [ | COVID-19 | 95.70 ± 1.95 | 96.77 ± 1.68 | 96.23 ± 1.6 | 96.67 ± 0.67 |
| Normal | 96.58 ± 0.7 | 96.20 ± 0.81 | 96.38 ± 0.66 | ||
| Pneumonia | 97.02 ± 0.80 | 97.00 ± 0.41 | 97.01 ± 0.58 | ||
| BCNN_SVM [ | COVID-19 | 97.58 ± 2.07 | 98.75 ± 0.88 | 98.15 ± 0.71 | 97.76 ± 0.42 |
| Normal | 96.82 ± 0.98 | 97.85 ± 1.01 | 97.33 ± 0.54 | ||
| Pneumonia | 98.48 ± 0.85 | 97.46 ± 0.81 | 97.96 ± 0.40 | ||
| COVNet [ | COVID-19 | 86.95 ± 3.64 | 92.19 ± 3.02 | 89.46 ± 2.68 | 93.14 ± 1.11 |
| Normal | 92.21 ± 2.24 | 92.65 ± 1.06 | 92.38 ± 1.06 | ||
| Pneumonia | 95.94 ± 1.34 | 93.38 ± 0.85 | 94.64 ± 0.88 | ||
| DTL-V19 [ | COVID-19 | 92.73 ± 1.86 | 91.99 ± 3.23 | 92.35 ± 2.52 | 91.53 ± 2.14 |
| Normal | 85.44 ± 4.09 | 92.73 ± 1.44 | 88.92 ± 2.78 | ||
| Pneumonia | 95.47 ± 1.01 | 90.70 ± 2.51 | 93.02 ± 1.76 | ||
| DenseNet121 [ | COVID-19 | 82.97 ± 5.10 | 90.25 ± 3.87 | 86.45 ± 4.53 | 91.99 ± 1.95 |
| Normal | 90.83 ± 20.2 | 91.44 ± 2.72 | 91.13 ± 2.21 | ||
| Pneumonia | 95.28 ± 1.29 | 92.80 ± 1.47 | 94.02 ± 1.24 | ||
| Resnet50V2 [ | COVID-19 | 87.27 ± 3.80 | 93.32 ± 2.21 | 90.17 ± 2.73 | 93.21 ± 1.19 |
| Normal | 91.25 ± 1.55 | 93.64 ± 1.42 | 92.43 ± 1.30 | ||
| Pneumonia | 96.22 ± 1.01 | 92.91 ± 1.20 | 94.54 ± 0.94 | ||
| Xception [ | COVID-19 | 94.61 ± 1.31 | 95.65 ± 2.64 | 95.10 ± 1.30 | 94.99 ± 1.02 |
| Normal | 92.97 ± 2.03 | 94.65 ± 1.47 | 93.79 ± 1.46 | ||
| Pneumonia | 96.52 ± 0.91 | 95.08 ± 0.85 | 95.79 ± 0.79 | ||
| MobileNetV2 [ | COVID-19 | 81.09 ± 6.16 | 91.01 ± 3.80 | 85.72 ± 4.84 | 90.60 ± 2.01 |
| Normal | 90.67 ± 1.50 | 88.54 ± 3.07 | 89.58 ± 2.11 | ||
| Pneumonia | 93.15 ± 1.54 | 91.98 ± 1.03 | 92.56 ± 1.25 | ||
| DAFLNet-1 (this work) | COVID-19 | 97.74 ± 0.43 | 99.37 ± 0.21 | 98.54 ± 0.11 | 98.56 ± 0.39 |
| Normal | 98.13 ± 1.09 | 98.47 ± 0.21 | 98.29 ± 0.53 | ||
| Pneumonia | 99.10 ± 0.14 | 98.43 ± 0.83 | 98.76 ± 0.36 | ||
| DAFLNet-2 (this work) | COVID-19 | 97.97 ± 0.89 | 99.45 ± 0.35 | 98.70 ± 0.44 | 98.41 ± 0.27 |
| Normal | 97.55 ± 0.66 | 98.82 ± 0.26 | 98.18 ± 0.35 | ||
| Pneumonia | 99.14 ± 0.15 | 97.86 ± 0.53 | 98.50 ± 0.24 |