Contrastive Transfer Learning for Prediction of Adverse Events in Hospitalized Patients

Hojjat Salehinejad, Anne M. Meehan, Pedro J. Caraballo, Bijan J. Borah

Research output: Contribution to journalArticlepeer-review

Abstract

Objective: Deterioration index (DI) is a computer-generated score at a specific frequency that represents the overall condition of hospitalized patients using a variety of clinical, laboratory and physiologic data. In this paper, a contrastive transfer learning method is proposed and validated for early prediction of adverse events in hospitalized patients using DI scores. Methods and procedures: An unsupervised contrastive learning (CL) model with a classifier is proposed to predict adverse outcome using a single temporal variable (DI scores). The model is pretrained on an unsupervised fashion with large-scale time series data and fine-tuned with retrospective DI score data. Results: The performance of this model is compared with supervised deep learning models for time series classification. Results show that unsupervised contrastive transfer learning with a classifier outperforms supervised deep learning solutions. Pretraining of the proposed CL model with large-scale time series data and fine-tuning that with DI scores can enhance prediction accuracy. Conclusion: A relationship exists between longitudinal DI scores of a patient and the corresponding outcome. DI scores and contrastive transfer learning can be used to predict and prevent adverse outcomes in hospitalized patients. Clinical impact: This paper successfully developed an unsupervised contrastive transfer learning algorithm for prediction of adverse events in hospitalized patients. The proposed model can be deployed in hospitals as an early warning system for preemptive intervention in hospitalized patients, which can mitigate the likelihood of adverse outcomes.

Original languageEnglish (US)
Pages (from-to)215-224
Number of pages10
JournalIEEE Journal of Translational Engineering in Health and Medicine
Volume12
DOIs
StatePublished - 2024

Keywords

  • Contrastive learning
  • deterioration index
  • early warning system
  • transfer learning

ASJC Scopus subject areas

  • Biomedical Engineering

Fingerprint

Dive into the research topics of 'Contrastive Transfer Learning for Prediction of Adverse Events in Hospitalized Patients'. Together they form a unique fingerprint.

Cite this