Title: Preparing for the next pandemic via transfer learning from existing diseases with hierarchical multi-modal BERT: a study on COVID-19 outcome prediction

Journal Article · · Scientific Reports

Abstract Developing prediction models for emerging infectious diseases from relatively small numbers of cases is a critical need for improving pandemic preparedness. Using COVID-19 as an exemplar, we propose a transfer learning methodology for developing predictive models from multi-modal electronic healthcare records by leveraging information from more prevalent diseases with shared clinical characteristics. Our novel hierarchical, multi-modal model ( $$$${\textsc {TransMED}}$$$$ T R A N S MED ) integrates baseline risk factors from the natural language processing of clinical notes at admission, time-series measurements of biomarkers obtained from laboratory tests, and discrete diagnostic, procedure and drug codes. We demonstrate the alignment of $$$${\textsc {TransMED}}$$$$ T R A N S MED ’s predictions with well-established clinical knowledge about COVID-19 through univariate and multivariate risk factor driven sub-cohort analysis. $$$${\textsc {TransMED}}$$$$ T R A N S MED ’s superior performance over state-of-the-art methods shows that leveraging patient data across modalities and transferring prior knowledge from similar disorders is critical for accurate prediction of patient outcomes, and this approach may serve as an important tool in the early response to future pandemics.

Research Organization:
Pacific Northwest National Laboratory (PNNL), Richland, WA (United States)
Sponsoring Organization:
National Science Foundation (NSF); USDOE
Contributing Organization:
Weintz Family COVID-19 Research Fund
Grant/Contract Number:
AC05-76RL01830
OSTI ID:
1873772
Report Number(s):
PNNL-SA-166555; 10748; PII: 13072
Journal Information:
Scientific Reports, Journal Name: Scientific Reports Journal Issue: 1 Vol. 12; ISSN 2045-2322
Publisher:
Nature Publishing GroupCopyright Statement
Country of Publication:
United Kingdom
Language:
English

References (28)

Mental health care for medical staff in China during the COVID-19 outbreak journal April 2020
Performance of pneumonia severity index and CURB-65 in predicting 30-day mortality in patients with COVID-19 journal September 2020
Learning from past respiratory failure patients to triage COVID-19 patient ventilator needs: A multi-institutional study journal July 2021
Spark NLP: Natural Language Understanding at Scale journal May 2021
U1 snRNP regulates cancer cell migration and invasion in vitro journal January 2020
Developing a COVID-19 mortality risk prediction model when individual-level data are not available journal September 2020
Machine learning based early warning system enables accurate mortality risk prediction for COVID-19 journal October 2020
Deep phenotyping of 34,128 adult patients hospitalised with COVID-19 in an international network study journal October 2020
Longitudinal symptom dynamics of COVID-19 infection journal December 2020
Anomalous collapses of Nares Strait ice arches leads to enhanced export of Arctic sea ice journal January 2021
Integrating deep learning CT-scan model, biological and clinical variables to predict severity of COVID-19 patients journal January 2021
Responding to the hidden pandemic for healthcare workers: stress journal April 2020
Multitask learning and benchmarking with clinical time series data journal June 2019
Re-epithelialization and immune cell behaviour in an ex vivo human skin model journal January 2020
A novel severity score to predict inpatient mortality in COVID-19 patients journal October 2020
Multimodal deep learning models for early detection of Alzheimer’s disease stage journal February 2021
Plasma Hsp90 levels in patients with systemic sclerosis and relation to lung and skin involvement: a cross-sectional and longitudinal study journal January 2021
A multilayer multimodal detection and prediction model based on explainable artificial intelligence for Alzheimer’s disease journal January 2021
Blockchain vehicles for efficient Medical Record management journal January 2020
A validated, real-time prediction model for favorable outcomes in hospitalized COVID-19 patients journal October 2020
Deep representation learning of electronic health records to unlock patient stratification at scale journal July 2020
Predicting COVID-19 mortality with electronic medical records journal February 2021
Med-BERT: pretrained contextualized embeddings on large-scale structured electronic health records for disease prediction journal May 2021
An interpretable mortality prediction model for COVID-19 patients journal May 2020
Developing epidemic forecasting models to assist disease surveillance for influenza with electronic health records journal August 2019
GRAM: Graph-based Attention Model for Healthcare Representation Learning
  • Choi, Edward; Bahadori, Mohammad Taha; Song, Le
  • KDD '17: The 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining https://doi.org/10.1145/3097983.3098126
conference August 2017
KGAT: Knowledge Graph Attention Network for Recommendation
  • Wang, Xiang; He, Xiangnan; Cao, Yixin
  • KDD '19: The 25th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining https://doi.org/10.1145/3292500.3330989
conference July 2019
Self-Supervised Learning of Contextual Embeddings for Link Prediction in Heterogeneous Networks conference June 2021