Adversarial Time-to-Event Modeling
About
Modern health data science applications leverage abundant molecular and electronic health data, providing opportunities for machine learning to build statistical models to support clinical practice. Time-to-event analysis, also called survival analysis, stands as one of the most representative examples of such statistical models. We present a deep-network-based approach that leverages adversarial learning to address a key challenge in modern time-to-event modeling: nonparametric estimation of event-time distributions. We also introduce a principled cost function to exploit information from censored events (events that occur subsequent to the observation window). Unlike most time-to-event models, we focus on the estimation of time-to-event distributions, rather than time ordering. We validate our model on both benchmark and real datasets, demonstrating that the proposed formulation yields significant performance gains relative to a parametric alternative, which we also propose.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Remaining Useful Life prediction | C-MAPSS FD001 | RMSE29.77 | 70 | |
| Remaining Useful Life prediction | C-MAPSS FD003 | RMSE30.96 | 69 | |
| RUL prediction | C-MAPSS FD004 | RMSE32.18 | 9 | |
| RUL prediction | C-MAPSS FD002 | RMSE31.73 | 9 | |
| Remaining Useful Life prediction | Toyota battery dataset | RMSE161.3 | 9 | |
| Remaining Useful Life prediction | NASA lithium-ion battery dataset | RMSE43.12 | 9 |