TEMPO: Transformers for Temporal Disease Progression from Cross-Sectional Data
About
Event-Based Models (EBMs) infer biomarker progression from cross-sectional data but typically only as ordinal sequences and rely on rigid model assumptions. We propose \textsc{Tempo}, a Transformer architecture that learns both ordinal and continuous event sequences through simulation-based supervised learning. \textsc{Tempo} uses two Transformer modules: one treats biomarkers as tokens to infer event sequencing; the other treats patients as tokens, representing each by their per-biomarker abnormality profile, to infer patients' disease stages. On synthetic benchmarks, \textsc{Tempo} reduces normalized Kendall's Tau distance by 52.89\% and staging MAE by 25.33\% compared to state-of-the-art SA-EBM, with larger reductions in high-dimensional settings (58.88\% and 61.10\%). Applied to ADNI, \textsc{Tempo} recovers a biologically plausible Alzheimer's progression: early medial temporal atrophy, followed by amyloid accumulation and cognitive decline, and late-stage tau pathology with terminal acceleration of global neurodegeneration -- broadly consistent with established disease models. \textsc{Tempo} also eliminates the need to derive custom inference algorithms and enables rapid empirical comparison of generative hypotheses.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Sequencing | High Dimensional Synthetic Data B = 100 biomarkers v1 | Tau Distance0.042 | 18 | |
| Staging | High Dimensional Synthetic Data B = 100 biomarkers v1 | Staging MAE2.61 | 18 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 1 | Tau Distance0.047 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 2 | Tau Distance0.072 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 3 Uniform stage prior | Tau Distance0.013 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 4, Uniform stage prior | Tau Distance0.037 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 5 | Tau Distance0.041 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 6 | Tau Distance0.057 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 7 | Tau Distance0.085 | 2 | |
| Temporal Disease Progression | Low Dimensional Synthetic Experiment 8 | Tau Distance0.073 | 2 |