Cargando…
Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images
PURPOSE: Confronting the pandemic of COVID-19 is nowadays one of the most prominent challenges of the human species. A key factor in slowing down the virus propagation is the rapid diagnosis and isolation of infected patients. The standard method for COVID-19 identification, the Reverse transcriptio...
Autores principales: | , , , , , , , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
Springer International Publishing
2021
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8055440/ http://dx.doi.org/10.1007/s42600-021-00151-6 |
_version_ | 1783680450051440640 |
---|---|
author | Luz, Eduardo Silva, Pedro Silva, Rodrigo Silva, Ludmila Guimarães, João Miozzo, Gustavo Moreira, Gladston Menotti, David |
author_facet | Luz, Eduardo Silva, Pedro Silva, Rodrigo Silva, Ludmila Guimarães, João Miozzo, Gustavo Moreira, Gladston Menotti, David |
author_sort | Luz, Eduardo |
collection | PubMed |
description | PURPOSE: Confronting the pandemic of COVID-19 is nowadays one of the most prominent challenges of the human species. A key factor in slowing down the virus propagation is the rapid diagnosis and isolation of infected patients. The standard method for COVID-19 identification, the Reverse transcription polymerase chain reaction method, is time-consuming and in short supply due to the pandemic. Thus, researchers have been looking for alternative screening methods, and deep learning applied to chest X-rays of patients has been showing promising results. Despite their success, the computational cost of these methods remains high, which imposes difficulties to their accessibility and availability. Thus, the main goal of this work is to propose an accurate yet efficient method in terms of memory and processing time for the problem of COVID-19 screening in chest X-rays. METHODS: To achieve the defined objective, we propose a new family of models based on the EfficientNet family of deep artificial neural networks which are known for their high accuracy and low footprints. We also exploit the underlying taxonomy of the problem with a hierarchical classifier. A dataset of 13,569 X-ray images divided into healthy, non-COVID-19 pneumonia, and COVID-19 patients is used to train the proposed approaches and other 5 competing architectures. We also propose a cross-dataset evaluation with a second dataset to evaluate the method generalization power. RESULTS: The results show that the proposed approach was able to produce a high-quality model, with an overall accuracy of 93.9%, COVID-19 sensitivity of 96.8%, and positive prediction of 100% while having from 5 to 30 times fewer parameters than the other tested architectures. Larger and more heterogeneous databases are still needed for validation before claiming that deep learning can assist physicians in the task of detecting COVID-19 in X-ray images, since the cross-dataset evaluation shows that even state-of-the-art models suffer from a lack of generalization power. CONCLUSIONS: We believe the reported figures represent state-of-the-art results, both in terms of efficiency and effectiveness, for the COVIDx database, a database of 13,800 X-ray images, 183 of which are from patients affected by COVID-19. The current proposal is a promising candidate for embedding in medical equipment or even physicians’ mobile phones. |
format | Online Article Text |
id | pubmed-8055440 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2021 |
publisher | Springer International Publishing |
record_format | MEDLINE/PubMed |
spelling | pubmed-80554402021-04-20 Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images Luz, Eduardo Silva, Pedro Silva, Rodrigo Silva, Ludmila Guimarães, João Miozzo, Gustavo Moreira, Gladston Menotti, David Res. Biomed. Eng. Original Article PURPOSE: Confronting the pandemic of COVID-19 is nowadays one of the most prominent challenges of the human species. A key factor in slowing down the virus propagation is the rapid diagnosis and isolation of infected patients. The standard method for COVID-19 identification, the Reverse transcription polymerase chain reaction method, is time-consuming and in short supply due to the pandemic. Thus, researchers have been looking for alternative screening methods, and deep learning applied to chest X-rays of patients has been showing promising results. Despite their success, the computational cost of these methods remains high, which imposes difficulties to their accessibility and availability. Thus, the main goal of this work is to propose an accurate yet efficient method in terms of memory and processing time for the problem of COVID-19 screening in chest X-rays. METHODS: To achieve the defined objective, we propose a new family of models based on the EfficientNet family of deep artificial neural networks which are known for their high accuracy and low footprints. We also exploit the underlying taxonomy of the problem with a hierarchical classifier. A dataset of 13,569 X-ray images divided into healthy, non-COVID-19 pneumonia, and COVID-19 patients is used to train the proposed approaches and other 5 competing architectures. We also propose a cross-dataset evaluation with a second dataset to evaluate the method generalization power. RESULTS: The results show that the proposed approach was able to produce a high-quality model, with an overall accuracy of 93.9%, COVID-19 sensitivity of 96.8%, and positive prediction of 100% while having from 5 to 30 times fewer parameters than the other tested architectures. Larger and more heterogeneous databases are still needed for validation before claiming that deep learning can assist physicians in the task of detecting COVID-19 in X-ray images, since the cross-dataset evaluation shows that even state-of-the-art models suffer from a lack of generalization power. CONCLUSIONS: We believe the reported figures represent state-of-the-art results, both in terms of efficiency and effectiveness, for the COVIDx database, a database of 13,800 X-ray images, 183 of which are from patients affected by COVID-19. The current proposal is a promising candidate for embedding in medical equipment or even physicians’ mobile phones. Springer International Publishing 2021-04-20 2022 /pmc/articles/PMC8055440/ http://dx.doi.org/10.1007/s42600-021-00151-6 Text en © Sociedade Brasileira de Engenharia Biomedica 2021 This article is made available via the PMC Open Access Subset for unrestricted research re-use and secondary analysis in any form or by any means with acknowledgement of the original source. These permissions are granted for the duration of the World Health Organization (WHO) declaration of COVID-19 as a global pandemic. |
spellingShingle | Original Article Luz, Eduardo Silva, Pedro Silva, Rodrigo Silva, Ludmila Guimarães, João Miozzo, Gustavo Moreira, Gladston Menotti, David Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images |
title | Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images |
title_full | Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images |
title_fullStr | Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images |
title_full_unstemmed | Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images |
title_short | Towards an effective and efficient deep learning model for COVID-19 patterns detection in X-ray images |
title_sort | towards an effective and efficient deep learning model for covid-19 patterns detection in x-ray images |
topic | Original Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8055440/ http://dx.doi.org/10.1007/s42600-021-00151-6 |
work_keys_str_mv | AT luzeduardo towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT silvapedro towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT silvarodrigo towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT silvaludmila towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT guimaraesjoao towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT miozzogustavo towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT moreiragladston towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages AT menottidavid towardsaneffectiveandefficientdeeplearningmodelforcovid19patternsdetectioninxrayimages |