Zero shot health trajectory prediction using transformer
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.
Authors
- Jarosław Wąs (ORCID: https://orcid.org/0000-0003-2964-745X)
- David Westfall Bates (ORCID: https://orcid.org/0000-0001-6268-1540)
- Quanzheng Li (ORCID: https://orcid.org/0000-0002-9651-5820)
- Anthony E. Samir (ORCID: https://orcid.org/0000-0002-7801-8724)
- Yugang Jia (ORCID: https://orcid.org/0009-0008-0628-5461)
- Arkadiusz Sitek (ORCID: https://orcid.org/0000-0002-0677-4002)
- Paweł Renc (ORCID: https://orcid.org/0000-0002-0487-7454)
Institutions
- Jagiellonian University (PL)
- Brigham and Women's Hospital (US)
- Harvard University (US)
- Massachusetts General Hospital (US)
- Massachusetts Institute of Technology (US)
- AGH University of Krakow (PL)
Publication Details
- Journal
- npj Digital Medicine
- Published
- 2024-09-19
- DOI
- https://doi.org/10.1038/s41746-024-01235-0
- Citations
- 37
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 14.96
Funders
- Office of Extramural Research, National Institutes of Health