Abstract
The COVID-19 pandemic is continuously evolving with drastically changing epidemiological situations which are approached with different decisions: from the reduction of fatalities to even the selection of patients with the highest probability of survival in critical clinical situations. Motivated by this, a battery of mortality prediction models with different performances has been developed to assist physicians and hospital managers. Logistic regression, one of the most popular classifiers within the clinical field, has been chosen as the basis for the generation of our models. Whilst a standard logistic regression only learns a single model focusing on improving accuracy, we propose to extend the possibilities of logistic regression by focusing on sensitivity and specificity. Hence, the log-likelihood function, used to calculate the coefficients in the logistic model, is split into two objective functions: one representing the survivors and the other for the deceased class. A multi-objective optimization process is undertaken on both functions in order to find the Pareto set, composed of models not improved by another model in both objective functions simultaneously. The individual optimization of either sensitivity (deceased patients) or specificity (survivors) criteria may be conflicting objectives because the improvement of one can imply the worsening of the other. Nonetheless, this conflict guarantees the output of a battery of diverse prediction models. Furthermore, a specific methodology for the evaluation of the Pareto models is proposed. As a result, a battery of COVID-19 mortality prediction models is obtained to assist physicians in decision-making for specific epidemiological situations.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
World health organization (2022). https://www.who.int/emergencies/diseases/novel-coronavirus-2019/situation-reports
Armañanzas, R., Díaz, A., Martínez-García, M., Mazuelas, S.: Derivation of a cost-sensitive COVID-19 mortality risk indicator using a multistart framework. In: IEEE International Conference on Bioinformatics and Biomedicine (BIBM), pp. 2179–2186 (2021). https://doi.org/10.1109/BIBM52615.2021.9669288
Blank, J., Deb, K.: Pymoo: multi-objective optimization in python. IEEE Access 8, 89497–89509 (2020). https://doi.org/10.1109/ACCESS.2020.2990567
Deb, K., Pratap, A., Agarwal, S., Meyarivan, T.: A fast and elitist multiobjective genetic algorithm: NSGA-ii. IEEE Trans. Evol. Comput. 6(2), 182–197 (2002). https://doi.org/10.1109/4235.996017
Gupta, R.K., Harrison, E.M., Ho, A., Docherty, A.B., Knight, S.R.: Development and validation of the ISARIC 4C deterioration model for adults hospitalised with COVID-19: a prospective cohort study. Lancet Respir. Med. 9(4), 349–359 (2021). https://doi.org/10.1016/S2213-2600(20)30559-2
Ircio, J., Lojo, A., Mori, U., Lozano, J.A.: A multivariate time series streaming classifier for predicting hard drive failures. IEEE Comput. Intell. Mag. 17(1), 102–114 (2022). https://doi.org/10.1109/MCI.2021.3129962
Knight, S.R., Ho, A., Pius, R., Buchan, I.: Risk stratification of patients admitted to hospital with COVID-19 using the ISARIC who clinical characterisation protocol: development and validation of the 4C mortality score, 370 (2020). https://doi.org/10.1136/bmj.m3339
Troyanskaya, O., Cantor, M., Sherlock, G., Brown, P.: Missing value estimation methods for DNA microarrays. Bioinformatics 17(6), 520–525 (2001). https://doi.org/10.1093/bioinformatics/17.6.520
Wynants, L., Van Calster, B., Collins, G.S., Riley, R.D.: Prediction models for diagnosis and prognosis of COVID-19: systematic review and critical appraisal, 369 (2020). https://doi.org/10.1136/bmj.m1328
Yan, L., Zhang, H.T., Goncalves, J., Xiao, Y.: An interpretable mortality prediction model for COVID-19 patients. Nat. Mach. Intell. 2(5), 283–288 (2020). https://doi.org/10.1038/s42256-020-0180-7
Acknowledgements
This research is supported by the Basque Government (IT1504-22, Elkartek) through the BERC 2022–2025 program and BMTF project, and by the Ministry of Science, Innovation and Universities: BCAM Severo Ochoa accreditation SEV-2017-0718 and PID2019-104966GB-I00. Furthermore, the work is also supported by the AXA Research Fund project “Early prognosis of COVID-19 infections via machine learning”.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
AAppendix
AAppendix
A detailed explanation of the variables used in the model is shown in Table 1.
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Martínez-García, M., García-Gutierrez, S., Armañanzas, R., Díaz, A., Inza, I., Lozano, J.A. (2022). Learning a Battery of COVID-19 Mortality Prediction Models by Multi-objective Optimization. In: Michalowski, M., Abidi, S.S.R., Abidi, S. (eds) Artificial Intelligence in Medicine. AIME 2022. Lecture Notes in Computer Science(), vol 13263. Springer, Cham. https://doi.org/10.1007/978-3-031-09342-5_32
Download citation
DOI: https://doi.org/10.1007/978-3-031-09342-5_32
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-09341-8
Online ISBN: 978-3-031-09342-5
eBook Packages: Computer ScienceComputer Science (R0)