Timezone: »

SurviVAEl: Variational Autoencoders for Clustering Time Series
Stefan Groha · Alexander Gusev · Sebastian Schmon

Fri Dec 02 01:00 PM -- 02:00 PM (PST) @
Event URL: https://openreview.net/forum?id=pREEF8_kWNT »

Multi-state models are generalizations of time-to-event models, where individuals progress through discrete states in continuous time. As opposed to classical approaches to survival analysis which include only alive-dead transitions, states can be competing in nature and transient, enabling richer modelling of complex clinical event series. Classical multi-state models, such as the Cox-Markov model, struggle to capture idiosyncratic, non-linear, time dependent, or high-dimensional covariates for which more sophisticated machine learning models are needed. Recently proposed extensions can overcome these limitations, however, they do not allow for uncertainty quantification of the model prediction, and typically have limited interpretability at the individual or population level. Here, we introduce SurviVAEl, a multi-state survival framework based on a VAE architecture, enabling uncertainty quantification and interpretable patient trajectory clustering.

Author Information

Stefan Groha (Harvard Medical School, Dana Farber Cancer Institute, Broad Institute of MIT and Harvard)
Alexander Gusev (Harvard University)
Sebastian Schmon (Improbable / Durham University)

More from the Same Authors