Zero Shot Health Trajectory Prediction Using Transformer
About
Integrating modern machine learning and clinical decision-making has great promise for mitigating healthcare's increasing cost and complexity. We introduce the Enhanced Transformer for Health Outcome Simulation (ETHOS), a novel application of the transformer deep-learning architecture for analyzing high-dimensional, heterogeneous, and episodic health data. ETHOS is trained using Patient Health Timelines (PHTs)-detailed, tokenized records of health events-to predict future health trajectories, leveraging a zero-shot learning approach. ETHOS represents a significant advancement in foundation model development for healthcare analytics, eliminating the need for labeled data and model fine-tuning. Its ability to simulate various treatment pathways and consider patient-specific factors positions ETHOS as a tool for care optimization and addressing biases in healthcare delivery. Future developments will expand ETHOS' capabilities to incorporate a wider range of data types and data sources. Our work demonstrates a pathway toward accelerated AI development and deployment in healthcare.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| ICU Imminent Mortality | MIMIC IV | Accuracy61 | 7 | |
| ICU Imminent Discharge | MIMIC IV | Accuracy63 | 7 | |
| ED Vital Sign Development | MIMIC IV | Event F15 | 7 | |
| Next event prediction | MIMIC IV | F1 Score (macro)0.3 | 6 | |
| Next Time-Step Prediction | MIMIC-IV Restricted to data supported by ETHOS | F1 Score (macro)4 | 2 |