Can I read Unsupervised Pre-Training on Patient Population Graphs for Patient-Level Predictions on EtoBox?

Unsupervised Pre-Training on Patient Population Graphs for Patient-Level Predictions by Pellegrini, Chantal; Kazi, Anees; Navab, Nassir is a scholarly article available to read on EtoBox.

What is Unsupervised Pre-Training on Patient Population Graphs for Patient-Level Predictions about?

Pre-training has shown success in different areas of machine learning, such as Computer Vision (CV), Natural Language Processing (NLP) and medical imaging. However, it has not been fully explored for clinical data analysis. Even though an immense amount of Electronic Health Record (EHR) data is recorded, data and labels can be scarce if the data is collected in small hospitals or deals with rare diseases. In such scenarios, pre-training on a larger set of EHR data could improve the model performance. In this paper, we apply unsupervised pre-training to heterogeneous, multi-modal EHR data for patient outcome prediction. To model this data, we leverage graph deep learning over population graphs. We first design a network architecture based on graph transformer designed to handle various input feature types occurring in EHR data, like continuous, discrete, and time-series features, allowing better multi-modal data fusion. Further, we design pre-training methods based on masked imputation to pre-train our network before fine-tuning on different end tasks. Pre-training is done in a fully unsupervised fashion, which lays the groundwork for pre-training on large public datasets with diffe

Author

Pellegrini, Chantal; Kazi, Anees; Navab, Nassir

Published

2022

Language

EN