Explainable learning to analyze the outcome of COVID-19 patients using clinical data

• Published in: Procedia computer science 2023, Vol.225, p.238-247
• Main Authors: Olañeta, Daniel, Morís, Daniel I., de Moura, Joaquim, Marcos, Pedro J., Rey, Enrique Míguez, Novo, Jorge, Ortega, Marcos
• Format: Article
• Language: English
• Subjects: COVID-19; decision trees; explainable machine learning; feature selection; outcome estimation
• ISSN: 1877-0509; 1877-0509
• DOI: 10.1016/j.procs.2023.10.008

Patients at high risk of contracting COVID-19 require specialized monitoring throughout their illness to ensure optimal treatment at each stage. To support this monitoring, Computer-Aided Diagnosis (CAD) methods analyze clinical data to estimate the most likely outcome for each patient, using various clinical variables such as symptoms, medical history, and laboratory results to predict outcomes. Despite the numerous proposals for COVID-19 diagnosis using CAD methods, the lack of explainability in many machine learning models poses a challenge in incorporating these methods into clinical practice. Additionally, other crucial tasks such as estimating the risk of death or severe forms of the disease must be considered to identify cases that require greater monitoring. To overcome these challenges, we propose an explainable methodology for estimating the risk of hospitalization and death in COVID-19 patients using clinical data. Our methodology employs four machine learning algorithms, three feature selection methods, and a decision tree to provide explainability. Our approach achieves an accuracy of 86.16% + 0.74% for the estimation of hospitalization risk with 29 features, and an accuracy of 86.40% + 1.80% for the estimation of the risk of death with 26 features. Moreover, our methodology provides valuable insights into the relationship between clinical variables and patient outcomes, which can inform more robust and informed clinical decision-making and improve our understanding of the disease. We demonstrate the potential of our transparent and effective CAD methods to support clinical decision-making in COVID-19 patient care and further research, offering a promising solution to overcome the challenges in incorporating CAD methods into clinical practice.