CEHR-XGPT: A Scalable Multi-Task Foundation Model for Electronic Health Records

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Pang, Chao, Park, Jiheum, Jiang, Xinzhuo, Pavinkurve, Nishanth Parameshwar, Kalluri, Krishna S., Joshi, Shalmali, Elhadad, Noémie, Natarajan, Karthik
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916935474610176
author Pang, Chao
Park, Jiheum
Jiang, Xinzhuo
Pavinkurve, Nishanth Parameshwar
Kalluri, Krishna S.
Joshi, Shalmali
Elhadad, Noémie
Natarajan, Karthik
author_facet Pang, Chao
Park, Jiheum
Jiang, Xinzhuo
Pavinkurve, Nishanth Parameshwar
Kalluri, Krishna S.
Joshi, Shalmali
Elhadad, Noémie
Natarajan, Karthik
contents Electronic Health Records (EHRs) provide a rich, longitudinal view of patient health and hold significant potential for advancing clinical decision support, risk prediction, and data-driven healthcare research. However, most artificial intelligence (AI) models for EHRs are designed for narrow, single-purpose tasks, limiting their generalizability and utility in real-world settings. Here, we present CEHR-XGPT, a general-purpose foundation model for EHR data that unifies three essential capabilities - feature representation, zero-shot prediction, and synthetic data generation - within a single architecture. To support temporal reasoning over clinical sequences, CEHR-XGPT incorporates a novel time-token-based learning framework that explicitly encodes patients' dynamic timelines into the model structure. CEHR-XGPT demonstrates strong performance across all three tasks and generalizes effectively to external datasets through vocabulary expansion and fine-tuning. Its versatility enables rapid model development, cohort discovery, and patient outcome forecasting without the need for task-specific retraining.
format Preprint
id arxiv_https___arxiv_org_abs_2509_03643
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CEHR-XGPT: A Scalable Multi-Task Foundation Model for Electronic Health Records
Pang, Chao
Park, Jiheum
Jiang, Xinzhuo
Pavinkurve, Nishanth Parameshwar
Kalluri, Krishna S.
Joshi, Shalmali
Elhadad, Noémie
Natarajan, Karthik
Machine Learning
Artificial Intelligence
Electronic Health Records (EHRs) provide a rich, longitudinal view of patient health and hold significant potential for advancing clinical decision support, risk prediction, and data-driven healthcare research. However, most artificial intelligence (AI) models for EHRs are designed for narrow, single-purpose tasks, limiting their generalizability and utility in real-world settings. Here, we present CEHR-XGPT, a general-purpose foundation model for EHR data that unifies three essential capabilities - feature representation, zero-shot prediction, and synthetic data generation - within a single architecture. To support temporal reasoning over clinical sequences, CEHR-XGPT incorporates a novel time-token-based learning framework that explicitly encodes patients' dynamic timelines into the model structure. CEHR-XGPT demonstrates strong performance across all three tasks and generalizes effectively to external datasets through vocabulary expansion and fine-tuning. Its versatility enables rapid model development, cohort discovery, and patient outcome forecasting without the need for task-specific retraining.
title CEHR-XGPT: A Scalable Multi-Task Foundation Model for Electronic Health Records
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.03643