Saved in:
Bibliographic Details
Main Authors: Song, Ziyang, Lu, Qingcheng, Zhu, He, Buckeridge, David, Li, Yue
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.02133
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913528944787456
author Song, Ziyang
Lu, Qingcheng
Zhu, He
Buckeridge, David
Li, Yue
author_facet Song, Ziyang
Lu, Qingcheng
Zhu, He
Buckeridge, David
Li, Yue
contents In many domains, such as healthcare, time-series data is often irregularly sampled with varying intervals between observations. This poses challenges for classical time-series models that require equally spaced data. To address this, we propose a novel time-series Transformer called Trajectory Generative Pre-trained Transformer (TrajGPT). TrajGPT employs a novel Selective Recurrent Attention (SRA) mechanism, which utilizes a data-dependent decay to adaptively filter out irrelevant past information based on contexts. By interpreting TrajGPT as discretized ordinary differential equations (ODEs), it effectively captures the underlying continuous dynamics and enables time-specific inference for forecasting arbitrary target timesteps. Experimental results demonstrate that TrajGPT excels in trajectory forecasting, drug usage prediction, and phenotype classification without requiring task-specific fine-tuning. By evolving the learned continuous dynamics, TrajGPT can interpolate and extrapolate disease risk trajectories from partially-observed time series. The visualization of predicted health trajectories shows that TrajGPT forecasts unseen diseases based on the history of clinically relevant phenotypes (i.e., contexts).
format Preprint
id arxiv_https___arxiv_org_abs_2410_02133
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TrajGPT: Irregular Time-Series Representation Learning for Health Trajectory Analysis
Song, Ziyang
Lu, Qingcheng
Zhu, He
Buckeridge, David
Li, Yue
Machine Learning
In many domains, such as healthcare, time-series data is often irregularly sampled with varying intervals between observations. This poses challenges for classical time-series models that require equally spaced data. To address this, we propose a novel time-series Transformer called Trajectory Generative Pre-trained Transformer (TrajGPT). TrajGPT employs a novel Selective Recurrent Attention (SRA) mechanism, which utilizes a data-dependent decay to adaptively filter out irrelevant past information based on contexts. By interpreting TrajGPT as discretized ordinary differential equations (ODEs), it effectively captures the underlying continuous dynamics and enables time-specific inference for forecasting arbitrary target timesteps. Experimental results demonstrate that TrajGPT excels in trajectory forecasting, drug usage prediction, and phenotype classification without requiring task-specific fine-tuning. By evolving the learned continuous dynamics, TrajGPT can interpolate and extrapolate disease risk trajectories from partially-observed time series. The visualization of predicted health trajectories shows that TrajGPT forecasts unseen diseases based on the history of clinically relevant phenotypes (i.e., contexts).
title TrajGPT: Irregular Time-Series Representation Learning for Health Trajectory Analysis
topic Machine Learning
url https://arxiv.org/abs/2410.02133