Machine learning algorithm for early-stage prediction of severe morbidity in COVID-19 pneumonia patients based on bio-signals

Abstract

Abstract Background Paralysis of medical systems has emerged as a major problem not only in Korea but also globally because of the COVID-19 pandemic. Therefore, early identification and treatment of COVID-19 are crucial. This study aims to develop a machine-learning algorithm based on bio-signals that predicts the infection three days in advance before it progresses from mild to severe, which may necessitate high-flow oxygen therapy or mechanical ventilation. Methods The study included 2758 hospitalized patients with mild severity COVID-19 between July 2020 and October 2021. Bio-signals, clinical information, and laboratory findings were retrospectively collected from the electronic medical records of patients. Machine learning methods included random forest, random forest ranger, gradient boosting machine, and support vector machine (SVM). Results SVM showed the best performance in terms of accuracy, kappa, sensitivity, detection rate, balanced accuracy, and run-time; the area under the receiver operating characteristic curve was also quite high at 0.96. Body temperature and SpO2 three and four days before discharge or exacerbation were ranked high among SVM features. Conclusions The proposed algorithm can predict the exacerbation of severity three days in advance in patients with mild COVID-19. This prediction can help effectively manage the reallocation of appropriate medical resources in clinical settings. Therefore, this algorithm can facilitate adequate oxygen therapy and mechanical ventilator preparation, thereby improving patient prognosis, increasing the efficiency of medical systems, and mitigating the damage caused by a global pandemic.

Keywords

• COVID-19
• Machine learning
• Bio-signal
• Morbidity, Prediction