Time-Aware Attention for Enhanced Electronic Health Records Modeling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yu, Junhan, Feng, Zhunyi, Lu, Junwei, Cai, Tianxi, Zhou, Doudou
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909696753926144
author Yu, Junhan
Feng, Zhunyi
Lu, Junwei
Cai, Tianxi
Zhou, Doudou
author_facet Yu, Junhan
Feng, Zhunyi
Lu, Junwei
Cai, Tianxi
Zhou, Doudou
contents Electronic Health Records (EHR) contain valuable clinical information for predicting patient outcomes and guiding healthcare decisions. However, effectively modeling Electronic Health Records (EHRs) requires addressing data heterogeneity and complex temporal patterns. Standard approaches often struggle with irregular time intervals between clinical events. We propose TALE-EHR, a Transformer-based framework featuring a novel time-aware attention mechanism that explicitly models continuous temporal gaps to capture fine-grained sequence dynamics. To complement this temporal modeling with robust semantics, TALE-EHR leverages embeddings derived from standardized code descriptions using a pre-trained Large Language Model (LLM), providing a strong foundation for understanding clinical concepts. Experiments on the MIMIC-IV and PIC dataset demonstrate that our approach outperforms state-of-the-art baselines on tasks such as disease progression forecasting. TALE-EHR underscores the benefit of integrating explicit, continuous temporal modeling with strong semantic representations provides a powerful solution for advancing EHR analysis.
format Preprint
id arxiv_https___arxiv_org_abs_2507_14847
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Time-Aware Attention for Enhanced Electronic Health Records Modeling
Yu, Junhan
Feng, Zhunyi
Lu, Junwei
Cai, Tianxi
Zhou, Doudou
Machine Learning
Electronic Health Records (EHR) contain valuable clinical information for predicting patient outcomes and guiding healthcare decisions. However, effectively modeling Electronic Health Records (EHRs) requires addressing data heterogeneity and complex temporal patterns. Standard approaches often struggle with irregular time intervals between clinical events. We propose TALE-EHR, a Transformer-based framework featuring a novel time-aware attention mechanism that explicitly models continuous temporal gaps to capture fine-grained sequence dynamics. To complement this temporal modeling with robust semantics, TALE-EHR leverages embeddings derived from standardized code descriptions using a pre-trained Large Language Model (LLM), providing a strong foundation for understanding clinical concepts. Experiments on the MIMIC-IV and PIC dataset demonstrate that our approach outperforms state-of-the-art baselines on tasks such as disease progression forecasting. TALE-EHR underscores the benefit of integrating explicit, continuous temporal modeling with strong semantic representations provides a powerful solution for advancing EHR analysis.
title Time-Aware Attention for Enhanced Electronic Health Records Modeling
topic Machine Learning
url https://arxiv.org/abs/2507.14847