Skip to content

Latest commit

 

History

History
51 lines (51 loc) · 1.95 KB

2022-12-31-han22a.md

File metadata and controls

51 lines (51 loc) · 1.95 KB
abstract booktitle title volume year layout series publisher issn id month tex_title firstpage lastpage page order cycles bibtex_author author date address container-title genre issued pdf extras
Survival analysis, the art of time-to-event modeling, plays an important role in clinical treatment decisions. Recently, continuous time models built from neural ODEs have been proposed for survival analysis. However, the training of neural ODEs is slow due to the high computational complexity of neural ODE solvers. Here, we propose an efficient alternative for flexible continuous time models, called Survival Mixture Density Networks (Survival MDNs). Survival MDN applies an invertible positive function to the output of Mixture Density Networks (MDNs). While MDNs produce flexible real-valued distributions, the invertible positive function maps the model into the time-domain while preserving a tractable density. Using four datasets, we show that Survival MDN performs better than, or similarly to continuous and discrete time baselines on concordance, integrated Brier score and integrated binomial log-likelihood. Meanwhile, Survival MDNs are also faster than ODE-based models and circumvent binning issues in discrete models.
Proceedings of the 7th Machine Learning for Healthcare Conference
Survival Mixture Density Networks
182
2022
inproceedings
Proceedings of Machine Learning Research
PMLR
2640-3498
han22a
0
Survival Mixture Density Networks
224
248
224-248
224
false
Han, Xintian and Goldstein, Mark and Ranganath, Rajesh
given family
Xintian
Han
given family
Mark
Goldstein
given family
Rajesh
Ranganath
2022-12-31
Proceedings of the 7th Machine Learning for Healthcare Conference
inproceedings
date-parts
2022
12
31