Skip to yearly menu bar Skip to main content


Poster

Contrastive Learning for Clinical Outcome Prediction with Partial Data Sources

Xia · Jonathan Wilson · Benjamin Goldstein · Ricardo Henao


Abstract:

The use of machine learning models to predict clinical outcomes from (longitudinal) electronic health record (EHR) data is becoming increasingly popular due to advances in deep architectures, representation learning, and the growing availability of large EHR datasets. Existing models generally assume access to the same data sources during both training and inference stages. However, this assumption is often challenged by the fact that real-world clinical datasets originate from various data sources, which though can be available for training (in a research or retrospective setting), are more realistically only partially available for inference when deployed. So motivated, we introduce Contrastive Learning for clinical Outcome Prediction with Partial data Sources (CLOPPS), that trains encoders to capture information across different data sources and then leverages them to build classifiers restricting access to a single data source. This approach can be used with existing cross-sectional or longitudinal outcome classification models. We present experiments on two real-world datasets demonstrating that CLOPPS consistently outperforms strong baselines in several practical scenarios.

Live content is unavailable. Log in and register to view live content