Discovering Subgroups with Exceptional Survival Characteristics
About
In many applications, it is important to identify subpopulations that survive longer or shorter than the rest of the population. In medicine, for example, it allows determining which patients benefit from treatment, and in predictive maintenance, which components are more likely to fail. Existing methods for discovering subgroups with exceptional survival characteristics rely on restrictive assumptions about the survival model (e.g. proportional hazards), require pre-discretized features, and, as they compare average statistics, tend to overlook individual heterogeneity. In this paper, we propose Sysurv, a non-parametric, fully differentiable method that discovers human-readable rules selecting subgroups with exceptional survival characteristics. Empirical evaluation on a wide range of datasets and settings, including a case study on cancer data, shows that Sysurv reveals insightful and actionable survival subgroups, outperforming the state of the art.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Subgroup Discovery | Nwtco SurvSet | Objective Value259.2 | 8 | |
| Subgroup Discovery | UnempDur SurvSet | Objective Value6.4 | 4 | |
| Subgroup Discovery | Rott2 SurvSet | Objective Value559.2 | 4 | |
| Subgroup Discovery | Aids2 SurvSet | Our Objective Score66.6 | 4 | |
| Subgroup Discovery | Support2 SurvSet | Objective Score132.8 | 4 | |
| Subgroup Discovery | DataDIVAT2 SurvSet | Our Objective286.3 | 4 | |
| Subgroup Discovery | ProstateSurvival SurvSet | Our Objective Score20.1 | 4 | |
| Subgroup Discovery | Actg SurvSet | Our Objective Score34.6 | 4 | |
| Subgroup Discovery | Scania SurvSet | Our objective172.5 | 4 | |
| Subgroup Discovery | Grace SurvSet | Our Objective Score38.6 | 4 |