Cargando…

An innovative ensemble model based on deep learning for predicting COVID-19 infection

Nowadays, global public health crises are occurring more frequently, and accurate prediction of these diseases can reduce the burden on the healthcare system. Taking COVID-19 as an example, accurate prediction of infection can assist experts in effectively allocating medical resources and diagnosing...

Descripción completa

Detalles Bibliográficos
Autores principales: Su, Xiaoying, Sun, Yanfeng, Liu, Hongxi, Lang, Qiuling, Zhang, Yichen, Zhang, Jiquan, Wang, Chaoyong, Chen, Yanan
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10387055/
https://www.ncbi.nlm.nih.gov/pubmed/37516796
http://dx.doi.org/10.1038/s41598-023-39408-8
_version_ 1785081803138138112
author Su, Xiaoying
Sun, Yanfeng
Liu, Hongxi
Lang, Qiuling
Zhang, Yichen
Zhang, Jiquan
Wang, Chaoyong
Chen, Yanan
author_facet Su, Xiaoying
Sun, Yanfeng
Liu, Hongxi
Lang, Qiuling
Zhang, Yichen
Zhang, Jiquan
Wang, Chaoyong
Chen, Yanan
author_sort Su, Xiaoying
collection PubMed
description Nowadays, global public health crises are occurring more frequently, and accurate prediction of these diseases can reduce the burden on the healthcare system. Taking COVID-19 as an example, accurate prediction of infection can assist experts in effectively allocating medical resources and diagnosing diseases. Currently, scholars worldwide use single model approaches or epidemiology models more often to predict the outbreak trend of COVID-19, resulting in poor prediction accuracy. Although a few studies have employed ensemble models, there is still room for improvement in their performance. In addition, there are only a few models that use the laboratory results of patients to predict COVID-19 infection. To address these issues, research efforts should focus on improving disease prediction performance and expanding the use of medical disease prediction models. In this paper, we propose an innovative deep learning model Whale Optimization Convolutional Neural Networks (CNN), Long-Short Term Memory (LSTM) and Artificial Neural Network (ANN) called WOCLSA which incorporates three models ANN, CNN and LSTM. The WOCLSA model utilizes the Whale Optimization Algorithm to optimize the neuron number, dropout and batch size parameters in the integrated model of ANN, CNN and LSTM, thereby finding the global optimal solution parameters. WOCLSA employs 18 patient indicators as predictors, and compares its results with three other ensemble deep learning models. All models were validated with train-test split approaches. We evaluate and compare our proposed model and other models using accuracy, F1 score, recall, AUC and precision metrics. Through many studies and tests, our results show that our prediction models can identify patients with COVID-19 infection at the AUC of 91%, 91%, and 93% respectively. Other prediction results achieve a respectable accuracy of 92.82%, 92.79%, and 91.66% respectively, f1-score of 93.41%, 92.79%, and 92.33% respectively, precision of 93.41%, 92.79%, and 92.33% respectively, recall of 93.41%, 92.79%, and 92.33% respectively. All of these exceed 91%, surpassing those of comparable models. The execution time of WOCLSA is also an advantage. Therefore, the WOCLSA ensemble model can be used to assist in verifying laboratory research results and predict and to judge various diseases in public health events.
format Online
Article
Text
id pubmed-10387055
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-103870552023-07-31 An innovative ensemble model based on deep learning for predicting COVID-19 infection Su, Xiaoying Sun, Yanfeng Liu, Hongxi Lang, Qiuling Zhang, Yichen Zhang, Jiquan Wang, Chaoyong Chen, Yanan Sci Rep Article Nowadays, global public health crises are occurring more frequently, and accurate prediction of these diseases can reduce the burden on the healthcare system. Taking COVID-19 as an example, accurate prediction of infection can assist experts in effectively allocating medical resources and diagnosing diseases. Currently, scholars worldwide use single model approaches or epidemiology models more often to predict the outbreak trend of COVID-19, resulting in poor prediction accuracy. Although a few studies have employed ensemble models, there is still room for improvement in their performance. In addition, there are only a few models that use the laboratory results of patients to predict COVID-19 infection. To address these issues, research efforts should focus on improving disease prediction performance and expanding the use of medical disease prediction models. In this paper, we propose an innovative deep learning model Whale Optimization Convolutional Neural Networks (CNN), Long-Short Term Memory (LSTM) and Artificial Neural Network (ANN) called WOCLSA which incorporates three models ANN, CNN and LSTM. The WOCLSA model utilizes the Whale Optimization Algorithm to optimize the neuron number, dropout and batch size parameters in the integrated model of ANN, CNN and LSTM, thereby finding the global optimal solution parameters. WOCLSA employs 18 patient indicators as predictors, and compares its results with three other ensemble deep learning models. All models were validated with train-test split approaches. We evaluate and compare our proposed model and other models using accuracy, F1 score, recall, AUC and precision metrics. Through many studies and tests, our results show that our prediction models can identify patients with COVID-19 infection at the AUC of 91%, 91%, and 93% respectively. Other prediction results achieve a respectable accuracy of 92.82%, 92.79%, and 91.66% respectively, f1-score of 93.41%, 92.79%, and 92.33% respectively, precision of 93.41%, 92.79%, and 92.33% respectively, recall of 93.41%, 92.79%, and 92.33% respectively. All of these exceed 91%, surpassing those of comparable models. The execution time of WOCLSA is also an advantage. Therefore, the WOCLSA ensemble model can be used to assist in verifying laboratory research results and predict and to judge various diseases in public health events. Nature Publishing Group UK 2023-07-29 /pmc/articles/PMC10387055/ /pubmed/37516796 http://dx.doi.org/10.1038/s41598-023-39408-8 Text en © The Author(s) 2023 https://creativecommons.org/licenses/by/4.0/Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Su, Xiaoying
Sun, Yanfeng
Liu, Hongxi
Lang, Qiuling
Zhang, Yichen
Zhang, Jiquan
Wang, Chaoyong
Chen, Yanan
An innovative ensemble model based on deep learning for predicting COVID-19 infection
title An innovative ensemble model based on deep learning for predicting COVID-19 infection
title_full An innovative ensemble model based on deep learning for predicting COVID-19 infection
title_fullStr An innovative ensemble model based on deep learning for predicting COVID-19 infection
title_full_unstemmed An innovative ensemble model based on deep learning for predicting COVID-19 infection
title_short An innovative ensemble model based on deep learning for predicting COVID-19 infection
title_sort innovative ensemble model based on deep learning for predicting covid-19 infection
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10387055/
https://www.ncbi.nlm.nih.gov/pubmed/37516796
http://dx.doi.org/10.1038/s41598-023-39408-8
work_keys_str_mv AT suxiaoying aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT sunyanfeng aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT liuhongxi aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT langqiuling aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT zhangyichen aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT zhangjiquan aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT wangchaoyong aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT chenyanan aninnovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT suxiaoying innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT sunyanfeng innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT liuhongxi innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT langqiuling innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT zhangyichen innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT zhangjiquan innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT wangchaoyong innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection
AT chenyanan innovativeensemblemodelbasedondeeplearningforpredictingcovid19infection