TraCeR: Transformer-Based Competing Risk Analysis with Longitudinal Covariates
This addresses a persistent challenge in survival analysis for researchers and practitioners by enabling better modeling of time-to-event data with longitudinal measurements and improved evaluation.
The authors tackled the challenge of incorporating longitudinal covariates and assessing calibration in survival analysis by introducing TraCeR, a transformer-based framework that achieved substantial and statistically significant performance improvements over state-of-the-art methods on multiple real-world datasets.
Survival analysis is a critical tool for modeling time-to-event data. Recent deep learning-based models have reduced various modeling assumptions including proportional hazard and linearity. However, a persistent challenge remains in incorporating longitudinal covariates, with prior work largely focusing on cross-sectional features, and in assessing calibration of these models, with research primarily focusing on discrimination during evaluation. We introduce TraCeR, a transformer-based survival analysis framework for incorporating longitudinal covariates. Based on a factorized self-attention architecture, TraCeR estimates the hazard function from a sequence of measurements, naturally capturing temporal covariate interactions without assumptions about the underlying data-generating process. The framework is inherently designed to handle censored data and competing events. Experiments on multiple real-world datasets demonstrate that TraCeR achieves substantial and statistically significant performance improvements over state-of-the-art methods. Furthermore, our evaluation extends beyond discrimination metrics and assesses model calibration, addressing a key oversight in literature.