Dataset Information

Contrastive learning improves critical event prediction in COVID-19 patients.

ABSTRACT: Deep learning (DL) models typically require large-scale, balanced training data to be robust, generalizable, and effective in the context of healthcare. This has been a major issue for developing DL models for the coronavirus disease 2019 (COVID-19) pandemic, where data are highly class imbalanced. Conventional approaches in DL use cross-entropy loss (CEL), which often suffers from poor margin classification. We show that contrastive loss (CL) improves the performance of CEL, especially in imbalanced electronic health records (EHR) data for COVID-19 analyses. We use a diverse EHR dataset to predict three outcomes: mortality, intubation, and intensive care unit (ICU) transfer in hospitalized COVID-19 patients over multiple time windows. To compare the performance of CEL and CL, models are tested on the full dataset and a restricted dataset. CL models consistently outperform CEL models, with differences ranging from 0.04 to 0.15 for area under the precision and recall curve (AUPRC) and 0.05 to 0.1 for area under the receiver-operating characteristic curve (AUROC).

SUBMITTER: Wanyan T

PROVIDER: S-EPMC8542449 | biostudies-literature | 2021 Dec

REPOSITORIES: biostudies-literature

ACCESS DATA

Publications

Contrastive learning improves critical event prediction in COVID-19 patients.

Wanyan Tingyi T Honarvar Hossein H Jaladanki Suraj K SK Zang Chengxi C Naik Nidhi N Somani Sulaiman S De Freitas Jessica K JK Paranjpe Ishan I Vaid Akhil A Zhang Jing J Miotto Riccardo R Wang Zhangyang Z Nadkarni Girish N GN Zitnik Marinka M Azad Ariful A Wang Fei F Ding Ying Y Glicksberg Benjamin S BS

Patterns (New York, N.Y.) 20211025 12

Deep learning (DL) models typically require large-scale, balanced training data to be robust, generalizable, and effective in the context of healthcare. This has been a major issue for developing DL models for the coronavirus disease 2019 (COVID-19) pandemic, where data are highly class imbalanced. Conventional approaches in DL use cross-entropy loss (CEL), which often suffers from poor margin classification. We show that contrastive loss (CL) improves the performance of CEL, especially in imbal ...[more]

PMID: 34723227

Similar Datasets

Project description:MotivationPatients with novel coronavirus disease 2019 (COVID-19) worsen into critical illness suddenly is a matter of great concern. Early identification and effective triaging of patients with a high risk of developing critical illness COVID-19 upon admission can aid in improving patient care, increasing the cure rate, and mitigating the burden on the medical care system. This study proposed and extended classical least absolute shrinkage and selection operator (LASSO) logistic regression to objectively identify clinical determination and risk factors for the early identification of patients at high risk of progression to critical illness at the time of hospital admission.MethodsIn this retrospective multicenter study, data of 1,929 patients with COVID-19 were assessed. The association between laboratory characteristics measured at admission and critical illness was screened with logistic regression. LASSO logistic regression was utilized to construct predictive models for estimating the risk that a patient with COVID-19 will develop a critical illness.ResultsThe development cohort consisted of 1,363 patients with COVID-19 with 133 (9.7%) patients developing the critical illness. Univariate logistic regression analysis revealed 28 variables were prognosis factors for critical illness COVID-19 (p < 0.05). Elevated CK-MB, neutrophils, PCT, α-HBDH, D-dimer, LDH, glucose, PT, APTT, RDW (SD and CV), fibrinogen, and AST were predictors for the early identification of patients at high risk of progression to critical illness. Lymphopenia, a low rate of basophils, eosinophils, thrombopenia, red blood cell, hematocrit, hemoglobin concentration, blood platelet count, and decreased levels of K, Na, albumin, albumin to globulin ratio, and uric acid were clinical determinations associated with the development of critical illness at the time of hospital admission. The risk score accurately predicted critical illness in the development cohort [area under the curve (AUC) = 0.83, 95% CI: 0.78-0.86], also in the external validation cohort (n = 566, AUC = 0.84).ConclusionA risk prediction model based on laboratory findings of patients with COVID-19 was developed for the early identification of patients at high risk of progression to critical illness. This cohort study identified 28 indicators associated with critical illness of patients with COVID-19. The risk model might contribute to the treatment of critical illness disease as early as possible and allow for optimized use of medical resources.

Dataset Information

Contrastive learning improves critical event prediction in COVID-19 patients.

Publications

Contrastive learning improves critical event prediction in COVID-19 patients.

Similar Datasets

OmicsDI is part of the ELIXIR infrastructure

Tweets