Retrospective development and external validation of machine learning models for COVID-19 severity prediction in lung and hematological malignancies: evidence of subtype-specific feature selection bias in joint training
Year: 2026
Type: article
Study design: Observational Study
Abstract: Patients with lung or hematological malignancies face an elevated risk of severe COVID-19 outcomes. While machine learning frameworks offer predictive utility, training a single joint clinical-biological model on heterogeneous cancer populations can introduce structural feature selection bias, limiting clinical generalizability. Utilizing a multicenter retrospective cohort (n = 340) for model development and an independent external cohort (103) for validation, we evaluated an end-to-end machine learning pipeline. Initially, a joint 7-variable signature—comprising age, ECOG performance status, lymphopenia, BMI, albumin, creatinine, and SARS-CoV-2 serology—was analyzed using logistic regression (LR) and random forest (RF) algorithms. Phenotypic deconstruction via cluster-level ablation and cancer-type-stratified optimal feature searches were subsequently executed to evaluate population-specific feature mechanics. The primary joint LR model achieved an optimism-corrected internal AUC of 0.75 (95% CI: 0.69–0.79) and an external validation AUC of 0.72 (95% CI: 0.62–0.82). However, cluster ablation and subgroup stratification revealed a critical biological divergence between the cancer types. The joint model's feature selection prioritized variables that acted as predictive noise for specific populations; notably, the apparent predictive contribution of albumin and creatinine in the joint model diminished or reversed after subtype-specific stratification, suggesting these variables largely reflect mixed-population effects rather than robust subtype-specific biological signal; their removal improved lung cancer model performance (albumin ΔAUC +0.043, creatinine +0.036) and did not materially benefit hematological patients either. The two cancer-type-optimal signatures shared age, ECOG performance status and lymphopenia, but diverged in their remaining components — BMI and albumin for hematological patients versus diabetes and immunodepression status for lung patients — indicating that the joint model's discriminative variables are only partially transferable across subtypes. While a unified 7-variable framework provides acceptable predictive baseline performance, joint training on mixed cancer populations introduces systematic feature selection bias. These exploratory findings suggest that cancer-type-stratified models may better capture subtype-specific biological signatures than a single joint model; however, independent prospective validation with pre-specified subtype stratification is required before any clinical deployment. Not applicable.
Source: BMC Infectious Diseases
Authors: Frederik Trinkmann, Nikos Paragios, Jean-Yves Blay, Hugo Crochet, Pierre-Etienne Heudel, Samir Bouam, Marie-Pierre Revel, Guillaume Chassagnon, Camille Bigenwald, Laurence Albiges, Fabrice Andre, Enzo Batistella, Olivier Teboul, Daniel Pescia, Basile Bertrand, Catherine Martineau-Huynh, Enzo D’Amico-Gandia, Walid Kandeil, Marie Guillet, Fungwe Jah, Amanda Remorino, David Dellamonica, Souad Assaad
Institutions: Heidelberg University, German Center for Lung Research, University Medical Centre Mannheim, Medizinische Fakultät Mannheim, Université Paris-Saclay, CentraleSupélec, Lyon 1 Université, Centre Léon Bérard, Hôpital Cochin, Assistance Publique – Hôpitaux de Paris, Groupe Hospitalier Cochin - Port-Royal, Hôtel-Dieu, Broca - La Collégiale, Université Paris Cité, Inserm, Institut Gustave Roussy, Immunologie anti-tumorale et immunothérapie des cancers, Fondation Gustave Roussy, Broad Institute, AstraZeneca (United Kingdom)
Language: English
Subfield: Infectious Diseases
Field: Medicine
Domain: Health Sciences
Open Access status: gold
Estimated APC paid: $3,090
OpenAlex ID: W7204474296
FWCI:
Cites:
Cited by:
Related to: