Interpretable Phenotype Discovery Using Temporal Risk-Aware Clustering

Longitudinal clinical cohorts contain heterogeneous progression patterns that are obscured by coarse diagnostic labels. Discovering these latent progression phenotypes requires models that distinguish patients not only by their outcomes, but also by how their clinical trajectories evolve. This paper proposes TRACE, an interpretable temporal phenotyping method that enables the learning of outcome-informed progression phenotypes from longitudinal trajectories. TRACE combines a trajectory- and outcome-aware contrastive objective with a probabilistic predictive phenotype layer, encouraging phenotypes to capture progression structure while remaining predictive of diagnostic state. A dual-attention mechanism further supports interpretation by identifying the visits and clinical variables most associated with each phenotype assignment. TRACE is evaluated on the Alzheimer’s Disease Neuroimaging Initiative (ADNI) cohort for cognitive disease modeling, showing improved performance in both diagnostic prediction and phenotype discovery compared with other temporal phenotyping baselines. Finally, by allowing the number of phenotypes to exceed the diagnostic labels, TRACE enables discovering additional progression patterns that are not captured by labels defined a priori.