PaReGTA: a temporally aware LLM-based patient representation framework for EHR analytics
Temporal information in structured electronic health records (EHRs) is often lost in sparse one-hot or count-based representations, while sequence models can be costly and data-hungry. We propose PaReGTA, an LLM-based encoding framework that (i) converts longitudinal EHR events into visit-level templated text with explicit temporal cues, (ii) learns domain-adapted visit embeddings via lightweight contrastive fine-tuning of a sentence-embedding model, and (iii) aggregates visit embeddings into a fixed-dimensional patient representation using hybrid temporal pooling that captures both recency and globally informative visits. The resulting fixed-dimensional patient representations can be used with conventional downstream machine-learning models. To examine factor-level sensitivity, we use PaReGTA-RSS (Representation Shift Score), a prespecified factor-removal analysis that recomputes patient representations after removing clinically defined factor groups and quantifies the resulting change in the fitted logit of a fixed logistic-regression model. We evaluated PaReGTA in a cohort of 39 088 patients with migraine from the All of Us Research Program (AoU) on a retrospective patient-level classification task with an EHR-derived target. In an exploratory comparison on the fixed held-out test cohort of 7818 patients, PaReGTA-Gap + LightGBM, used as a post hoc analytical reference, had the highest reported AUC, accuracy, and F1 values among the evaluated sparse, BERT-based EHR, and recurrent approaches in this cohort.
Authors
- Todd J. Schwedt (ORCID: https://orcid.org/0000-0002-7780-7086)
- Catherine Daniela Chong (ORCID: https://orcid.org/0000-0002-9035-8748)
- Chia‐Chun Chiang (ORCID: https://orcid.org/0000-0001-7802-7172)
- Yoon Kihyuk
- Lingchao Mao
- Jing Li
Institutions
- Georgia Institute of Technology (US)
- Mayo Clinic (US)
- Mayo Clinic in Arizona (US)
- Ulsan National Institute of Science and Technology (KR)
Publication Details
- Journal
- IISE Transactions on Healthcare Systems Engineering
- Published
- 2026-11-01
- DOI
- https://doi.org/10.1080/24725579.2026.2740600
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Ministry of Trade, Industry and Energy