Neural Topic Models with Survival Supervision: Jointly Predicting
  Time-to-Event Outcomes and Learning How Clinical Features Relate
  In time-to-event prediction problems, a standard approach to estimating an
interpretable model is to use Cox proportional hazards, where features are
selected based on lasso regularization or stepwise regression. However, these
Cox-based models do not learn how different features relate. As an alternative,
we present an interpretable neural network approach to jointly learn a survival
model to predict time-to-event outcomes while simultaneously learning how
features relate in terms of a topic model. In particular, we model each subject
as a distribution over "topics", which are learned from clinical features as to
help predict a time-to-event outcome. From a technical standpoint, we extend
existing neural topic modeling approaches to also minimize a survival analysis
loss function. We study the effectiveness of this approach on seven healthcare
datasets on predicting time until death as well as hospital ICU length of stay,
where we find that neural survival-supervised topic models achieves competitive
accuracy with existing approaches while yielding interpretable clinical
"topics" that explain feature relationships.
