| Literature DB >> 34749098 |
Sajad Ahmadian1, Seyed Mohammad Jafar Jalali2, Syed Mohammed Shamsul Islam3, Abbas Khosravi4, Ebrahim Fazli5, Saeid Nahavandi4.
Abstract
COVID-19 has had a detrimental impact on normal activities, public safety, and the global financial system. To identify the presence of this disease within communities and to commence the management of infected patients early, positive cases should be diagnosed as quickly as possible. New results from X-ray imaging indicate that images provide key information about COVID-19. Advanced deep-learning (DL) models can be applied to X-ray radiological images to accurately diagnose this disease and to mitigate the effects of a shortage of skilled medical personnel in rural areas. However, the performance of DL models strongly depends on the methodology used to design their architectures. Therefore, deep neuroevolution (DNE) techniques are introduced to automatically design DL architectures accurately. In this paper, a new paradigm is proposed for the automated diagnosis of COVID-19 from chest X-ray images using a novel two-stage improved DNE Algorithm. The proposed DNE framework is evaluated on a real-world dataset and the results demonstrate that it provides the highest classification performance in terms of different evaluation metrics.Entities:
Keywords: COVID-19 diagnosis; Convolutional neural network; Evolutionary computation; Improved salp swarm algorithm
Mesh:
Year: 2021 PMID: 34749098 PMCID: PMC8558149 DOI: 10.1016/j.compbiomed.2021.104994
Source DB: PubMed Journal: Comput Biol Med ISSN: 0010-4825 Impact factor: 4.589
Fig. 1The flowchart of the proposed BSSA Algorithm.
Fig. 2Overall procedure of COVID-19 diagnosis using the proposed BSSA-CNN model.
Fig. 3Two samples of available images in the dataset related to the patients with COVID-19.
Fig. 4Two samples of available images in the dataset related to normal persons.
List of CNN hyperparameter symbols and their values.
| Symbol | Value |
|---|---|
| K | [1, 30] |
| N | [1, 500] |
| N | [1, 400] |
| B | [10, 20, …, 200] |
| MP | [1, 20] |
| D | [0.2, 0.25, …, 0.65] |
| L | [0.001, 0.006,.., 0.1] |
| M | [0.05, 0.1,.., 0.95] |
| N | [1, 2, …, 20] |
Fig. 5The influence of different initialized population sizes on the performance of our proposed model.
Quantitative performance results of evaluation metrics for evolutionary algorithms over the binary COVID-19 dataset. The best results among different DNEs are shown by bold black fonts.
| Metric | GA-CNN | PSO–CNN | DE-CNN | GWO-CNN | MFO-CNN | SSA-CNN | OSSA-CNN | CSSA-CNN | BSSA-Softmax | BSSA-CEL | DON | ADOPT | Proposed | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ACC | AVG | 0.917442 | 0.931232 | 0.942693 | 0.919771 | 0.928367 | 0.936963 | 0.974212 | 0.977077 | 0.975479 | 0.969248 | 0.973228 | 0.978555 | |
| STD | 0.008957 | 0.008912 | 0.009358 | 0.007384 | 0.006933 | 0.007019 | 0.007566 | 0.007044 | 0.006446 | 0.006653 | 0.007182 | 0.00692 | ||
| Best | 0.937685 | 0.943488 | 0.948802 | 0.932559 | 0.938876 | 0.945667 | 0.978081 | 0.983667 | 0.982541 | 0.977113 | 0.981822 | 0.988703 | ||
| Worst | 0.894537 | 0.903663 | 0.916766 | 0.895053 | 0.913443 | 0.908897 | 0.965445 | 0.968003 | 0.966743 | 0.961199 | 0.969822 | 0.971634 | ||
| Precision | AVG | 0.948718 | 0.961538 | 0.920455 | 0.884615 | 0.955414 | 0.929412 | 0.981818 | 0.981928 | 0.980302 | 0.970118 | 0.980213 | 0.981228 | |
| STD | 0.018232 | 0.017033 | 0.017163 | 0.009408 | 0.018145 | 0.029241 | 0.005666 | 0.003367 | 0.003812 | 0.003135 | 0.004119 | 0.003291 | ||
| Best | 0.958434 | 0.966768 | 0.930221 | 0.902443 | 0.961113 | 0.938986 | 0.984554 | 0.983227 | 0.982105 | 0.976142 | 0.982716 | 0.98312 | ||
| Worst | 0.911679 | 0.923572 | 0.901088 | 0.879044 | 0.913008 | 0.909077 | 0.963114 | 0.968922 | 0.967688 | 0.961457 | 0.970005 | 0.973649 | ||
| Recall | AVG | 0.880952 | 0.892857 | 0.964286 | 0.958333 | 0.892857 | 0.940476 | 0.964286 | 0.970238 | 0.968704 | 0.962842 | 0.963433 | 0.968099 | |
| STD | 0.028157 | 0.037647 | 0.036811 | 0.017883 | 0.019884 | 0.027636 | 0.016889 | 0.013776 | 0.012288 | 0.011026 | 0.013433 | 0.010199 | ||
| Best | 0.897411 | 0.913444 | 0.974521 | 0.959222 | 0.920557 | 0.959088 | 0.974407 | 0.976885 | 0.975762 | 0.970231 | 0.975062 | 0.971388 | ||
| Worst | 0.858999 | 0.867621 | 0.943222 | 0.916588 | 0.858804 | 0.912223 | 0.961282 | 0.961434 | 0.96019 | 0.954067 | 0.959989 | 0.965622 | ||
| F-measure | AVG | 0.913582 | 0.925926 | 0.94186 | 0.919778 | 0.923077 | 0.934911 | 0.972973 | 0.976048 | 0.974712 | 0.968389 | 0.975451 | 0.978044 | |
| STD | 0.012807 | 0.013191 | 0.011576 | 0.011867 | 0.011989 | 0.009428 | 0.010339 | 0.009676 | 0.008416 | 0.007154 | 0.009545 | 0.016522 | ||
| Best | 0.922748 | 0.934389 | 0.955619 | 0.925561 | 0.925655 | 0.948888 | 0.975668 | 0.980033 | 0.978891 | 0.972268 | 0.980488 | 0.984731 | ||
| Worst | 0.891833 | 0.883533 | 0.921319 | 0.888346 | 0.883044 | 0.923346 | 0.958988 | 0.965543 | 0.964342 | 0.958019 | 0.967812 | 0.969596 | ||
| AUC | AVG | 0.918377 | 0.929854 | 0.943469 | 0.921156 | 0.927092 | 0.937089 | 0.973856 | 0.976832 | 0.975346 | 0.969883 | 0.975557 | 0.978522 | |
| STD | 0.008993 | 0.011899 | 0.010256 | 0.009545 | 0.009943 | 0.006575 | 0.009918 | 0.008774 | 0.007352 | 0.006429 | 0.008151 | 0.006333 | ||
| Best | 0.932845 | 0.936767 | 0.963133 | 0.928461 | 0.935449 | 0.954478 | 0.978888 | 0.982359 | 0.981239 | 0.974616 | 0.981911 | 0.979477 | ||
| Worst | 0.909657 | 0.901532 | 0.908748 | 0.900355 | 0.891001 | 0.920004 | 0.964544 | 0.970669 | 0.969455 | 0.962693 | 0.971331 | 0.970036 |
Fig. 6Confusion matrices of the best (proposed model) and the second-best (CSSA-CNN) algorithms based on the accuracy metric.
Fig. 7Box plots of different DNEs for the accuracy metric.
Fig. 8Convergence curves of different DNEs based on the accuracy metric.
Quantitative performance results of evaluation metrics for state-of-the-art algorithms for the binary COVID-19 dataset. The best results among different classifiers are shown by bold black fonts.
| Metric | MobileNet | DesnseNet121 | InceptionV3 | ResNet50 | VGG16 | ResNet50V2 | ResNet152V2 | nCOVnet | CTnet-10 | Proposed | |
|---|---|---|---|---|---|---|---|---|---|---|---|
| ACC | AVG | 0.888252 | 0.882521 | 0.902579 | 0.836676 | 0.885387 | 0.785111 | 0.908309 | 0.896731 | 0.886574 | |
| STD | 0.008435 | 0.009437 | 0.007891 | 0.040947 | 0.018467 | 0.047282 | 0.056214 | 0.043103 | 0.032791 | ||
| Best | 0.901883 | 0.903922 | 0.913226 | 0.893533 | 0.905344 | 0.826711 | 0.948979 | 0.936001 | 0.924471 | ||
| Worst | 0.865554 | 0.869047 | 0.879011 | 0.803216 | 0.827056 | 0.748022 | 0.853432 | 0.842854 | 0.832498 | ||
| Precision | AVG | 0.852459 | 0.950355 | 0.848958 | 0.930233 | 0.890244 | 0.854962 | 0.914634 | 0.904323 | 0.894084 | |
| STD | 0.046995 | 0.009899 | 0.042101 | 0.030749 | 0.033152 | 0.042653 | 0.110804 | 0.100705 | 0.090687 | ||
| Best | 0.883007 | 0.965652 | 0.858922 | 0.954767 | 0.915038 | 0.884578 | 0.937876 | 0.927388 | 0.917053 | ||
| Worst | 0.815561 | 0.937689 | 0.803574 | 0.874521 | 0.854538 | 0.821366 | 0.794022 | 0.782444 | 0.770778 | ||
| Recall | AVG | 0.928571 | 0.797619 | 0.970238 | 0.714286 | 0.869048 | 0.666667 | 0.892857 | 0.881279 | 0.869121 | |
| STD | 0.009231 | 0.073547 | 0.000213 | 0.086395 | 0.027655 | 0.069103 | 0.090967 | 0.078957 | 0.066918 | ||
| Best | 0.928895 | 0.814423 | 0.978845 | 0.786454 | 0.891044 | 0.736754 | 0.952612 | 0.943496 | 0.933685 | ||
| Worst | 0.897554 | 0.730567 | 0.966328 | 0.659113 | 0.837529 | 0.608932 | 0.811042 | 0.797931 | 0.785092 | ||
| F-measure | AVG | 0.888889 | 0.867314 | 0.905556 | 0.808081 | 0.879518 | 0.749164 | 0.903614 | 0.892503 | 0.880941 | |
| STD | 0.015655 | 0.027781 | 0.008944 | 0.053299 | 0.037424 | 0.061454 | 0.046513 | 0.037597 | 0.028482 | ||
| Best | 0.900767 | 0.893563 | 0.922155 | 0.846568 | 0.899045 | 0.795423 | 0.939082 | 0.929966 | 0.920728 | ||
| Worst | 0.841911 | 0.789022 | 0.899294 | 0.748989 | 0.845652 | 0.681921 | 0.857939 | 0.847628 | 0.837406 | ||
| AUC | AVG | 0.889711 | 0.879473 | 0.905009 | 0.832281 | 0.884836 | 0.780847 | 0.907755 | 0.896839 | 0.885949 | |
| STD | 0.010494 | 0.011373 | 0.009879 | 0.042439 | 0.035194 | 0.048984 | 0.053612 | 0.043531 | 0.031866 | ||
| Best | 0.910898 | 0.893315 | 0.921883 | 0.868494 | 0.908935 | 0.834326 | 0.943089 | 0.925978 | 0.908786 | ||
| Worst | 0.875497 | 0.869454 | 0.883533 | 0.773545 | 0.867435 | 0.738782 | 0.854025 | 0.841809 | 0.828476 |
Fig. 9Confusion matrices of the best (proposed model) and the second-best (ResNet152V2) classification algorithms based on the accuracy metric.
Fig. 10Box plots of different classifiers based on the accuracy metric.
Performance comparison of our proposed model against other EA-based deep-learning models based on the Wilcoxon paired signed-rank test with confidence intervals of 95%.
| GA-CNN | PSO–CNN | DE-CNN | GWO-CNN | MFO-CNN | SSA-CNN | OSSA-CNN | CSSA-CNN | BSSA-Softmax | BSSA-CEL | DON | ADOPT |
|---|---|---|---|---|---|---|---|---|---|---|---|
| 9.646E-6 | 9.725E-5 | 7.545E-5 | 9.121E-6 | 4.412E-5 | 8.333E-5 | 4.509E-4 | 4.661E-4 | 5.408E-4 | 2.394E-5 | 5.661E-4 | 4.495E-4 |
Performance comparison of our proposed model against other deep-learning models based on the Wilcoxon paired signed-rank test with confidence intervals of 95%.
| MobileNet | DesnseNet121 | InceptionV3 | ResNet50 | VGG16 | ResNet50V2 | ResNet152V2 | nCOVnet | CTnet-10 |
|---|---|---|---|---|---|---|---|---|
| 7.369E-6 | 7.825E-6 | 6.131E-6 | 7.375E-7 | 7.694E-6 | 5.661E-8 | 5.656E-6 | 6.779E-6 | 7.551E-6 |
The run-time consumed in seconds for all algorithms over the COVID-19 dataset.
| Model | Optimization time | Training time | Test time |
|---|---|---|---|
| MobileNet | – | 641 | 239 |
| DesnseNet121 | – | 624 | 231 |
| InceptionV3 | – | 583 | 198 |
| ResNet50 | – | 612 | 213 |
| VGG16 | – | 567 | 192 |
| ResNet50V2 | – | 539 | 195 |
| ResNet152V2 | – | 542 | 183 |
| nCOVnet | – | 531 | 166 |
| CTnet-10 | – | 512 | 171 |
| GA-CNN | 2846 | 497 | 169 |
| PSO–CNN | 2689 | 451 | 142 |
| DE-CNN | 2710 | 482 | 157 |
| GWO-CNN | 2635 | 423 | 162 |
| MFO-CNN | 2647 | 411 | 153 |
| SSA-CNN | 2479 | 394 | 149 |
| OSSA-CNN | 2422 | 361 | 132 |
| CSSA-CNN | 2461 | 389 | 128 |
| BSSA-Softmax | 2453 | 376 | 117 |
| BSSA-CEL | 2411 | 348 | 106 |
| Proposed |