Cost-sensitive ordinal classification methods to predict SARS-CoV-2 pneumonia severity

Fernando Garcia-Garcia,Dae-Jin Lee,Pedro Pablo Espana Yandiola,Isabel Urrutia Landa,Joaquin Martinez-Minaya,Miren Hayet-Otero,Monica Nieves Ermecheo,Jose Maria Quintana,Rosario Menendez,Antoni Torres,Rafael Zalacain Jorge,Fernando García-García,Pedro Pablo España Yandiola,Joaquín Martínez-Minaya,Mónica Nieves Ermecheo,José María Quintana,Rosario Menéndez
DOI: https://doi.org/10.1109/jbhi.2024.3363765
IF: 7.7
2024-01-01
IEEE Journal of Biomedical and Health Informatics
Abstract:OBJECTIVE: <AbstractText Label="OBJECTIVE" NlmCategory="OBJECTIVE">To study the suitability of costsensitive ordinal artificial intelligence-machine learning (AIML) strategies in the prognosis of SARS-CoV-2 pneumonia severity.</AbstractText>MATERIALS & METHODS: <AbstractText Label="MATERIALS & METHODS" NlmCategory="METHODS">Observational, retrospective, longitudinal, cohort study in 4 hospitals in Spain. Information regarding demographic and clinical status was supplemented by socioeconomic data and air pollution exposures. We proposed AI-ML algorithms for ordinal classification via ordinal decomposition and for cost-sensitive learning via resampling techniques. For performance-based model selection, we defined a custom score including per-class sensitivities and asymmetric misprognosis costs. 260 distinct AI-ML models were evaluated via 10 repetitions of 5×5 nested cross-validation with hyperparameter tuning. Model selection was followed by the calibration of predicted probabilities. Final overall performance was compared against five well-established clinical severity scores and against a 'standard' (non-cost sensitive, non-ordinal) AI-ML baseline. In our best model, we also evaluated its explainability with respect to each of the input variables.</AbstractText>RESULTS: <AbstractText Label="RESULTS" NlmCategory="RESULTS">The study enrolled n = 1548 patients: 712 experienced low, 238 medium, and 598 high clinical severity. d = 131 variables were collected, becoming d ' = 148 features after categorical encoding. Model selection resulted in our best-performing AI-ML pipeline having: a) no imputation of missing data, b) no feature selection (i.e. using the full set of d ' features), c) 'Ordered Partitions' ordinal decomposition, d) cost-based reimbalance, and e) a Histogram-based Gradient Boosting classifier. This best model (calibrated) obtained a median accuracy of 68.1% [67.3%, 68.8%] (95% confidence interval), a balanced accuracy of 57.0% [55.6%, 57.9%], and an overall area under the curve (AUC) 0.802 [0.795, 0.808]. In our dataset, it outperformed all five clinical severity scores and the 'standard' AI-ML baseline.</AbstractText>DISCUSSION & CONCLUSION: <AbstractText Label="DISCUSSION & CONCLUSION" NlmCategory="CONCLUSIONS">We conducted an exhaustive exploration of AI-ML methods designed for both ordinal and cost-sensitive classification, motivated by a real-world application domain (clinical severity prognosis) in which these topics arise naturally. Our model with the best classification performance exploited successfully the ordering information of ground truth classes, coping with imbalance and asymmetric costs. However, these ordinal and cost-sensitive aspects are seldom explored in the literature.</AbstractText>
computer science, interdisciplinary applications,mathematical & computational biology,medical informatics, information systems
What problem does this paper attempt to address?