The Loss or the Architecture? What Matters in Survival Analysis with Competing Risks
Organizations: SODA Team, Inria Saclay, France
Abstract
Often, eg in healthcare and predictive maintenance, what matters is which of several competing causes will strike first, and with what probability: survival analysis with competing risks. Corresponding deep learning methods are developed as bundles: a new architecture together with a new training loss, tested on one to five datasets against a handful of baselines. Thus, a state-of-the-art result cannot be attributed to the architecture, the loss, or the evaluation protocol. The coupling is not accidental: with one incidence curve per cause, a network's head is tied to the losses expressible on it. We vary the two axes independently, three losses, two of them strictly proper, on four backbones, across two modalities that tease the axes apart: 20 tabular and 3 pathology-image datasets. One loss wins in both: modelling cumulative incidence directly under censoring weighting. It brings twice as much as benefits a good choice of backbone on competing-risks data, and no more than it on single-event data: the loss is decisive for competing risks. The backbone matters too, and must match the modality: tabular networks lead on tables, pretrained vision encoders on images. The two choices compound rather than substitute. Proper scoring rules reveal these dynamics; the concordance index (C-index), which is not proper, does not separate models: neither axis moves it, not even tuning on the loss built to improve it. This is an avenue of progress: composed, the two give a model nobody has tried, RealMLP or TabM trained with the cumulative-incidence loss, which takes the best Brier score in our benchmark.
Figures & tables
| # Real Life CR | # | # CR | |
| Publication | Datasets | baselines | metrics |
| Ishwaran et al. 2014 | 5 | 5 | 2 |
| Lee et al. 2018 | 1 | 3 | 1 |
| Wang & Sun 2022 | 1 | 6 | 1 |
| Jeanselme et al. 2023 | 3 | 5 | 3 |
| Alberge et al. 2025 | 1 | 7 | 3 |
| Competing Risks | ||
| Dataset | Samples | Cens. |
| Framingham | 4,240 | 58% |
| Metabric (CR) | 3,237 | 33% |
| MIMIC IV | 65,366 | 58% |
| MIMIC III | 38,511 | 54% |
| SEER Breast | 476,746 | 63% |
Appendix figures & tables20 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset name | CR | Samples | Nb features | Censoring rate | Github |
| Framingham | ✓ | 4,240 | 24 | 58% | |
| Metabric (competing events) | ✓ | 3,237 | 73 | 33% | ✗ |
| MIMIC IV | ✓ | 65,366 | 172 | 58% | ✗ |
| MIMIC III | ✓ | 38,511 | 81 | 54% | ✗ |
| SEER Breast Cancer | ✓ | 476,746 | 18 | 63% | ✗ |
| SEER Digestion Cancer | ✓ | 887,552 | 304 | 19% | ✗ |
| Package | Language | used for |
| pycox | Python | Downloading datasets, DeepHit Architecture |
| lifelines | Python | Aalen–Johansen estimator ( Aalen & Johansen, 1978 ) |
| auton-survival | Python | Downloading Datasets |
| hazardous | Python | Computing CR metrics (IBS, C-index, calibrations), SurvivalBoost |
| cbioportal | R | Downloading Datasets |
| cmprsks | R | Fine and Gray estimator ( Fine & Gray, 1999 ) , RSF |
| model | parameter | range |
|---|---|---|
| Backbones (each trained under all three losses) | ||
| MLP | batch size | 128, 256 |
| dropout | None, 0.1, 0.2, 0.3 | |
| epochs | randint (256, 1024) | |
| lr | loguniform (1e-05, 0.001) | |
| nb output times | 40, 60, 80, 100 | |
| Selection | Backbone | Median | Min | Max | ||
| IBS | MLP | |||||
| RealMLP | ||||||
| TabM | ||||||
| SurvTRACE | ||||||
| All backbones | ||||||
| C-index | MLP |
| cells | median Cox vs. KM | median stratified vs. KM | |
| Stable | 24 | ||
| Blown up ( worse) | 11 |
| Datasets | Loss effect | Backbone effect | Ratio | ||
| median | IQR | median | IQR | ||
| Competing risks | 0.053 | [0.020, 0.084] | 0.027 | [0.021, 0.062] | 1.93 |
| Single event | 0.072 | [0.045, 0.132] | 0.086 | [0.039, 0.208] | 0.83 |
| All 20 | 0.065 | [0.030, 0.100] | 0.059 | [0.023, 0.136] | 1.10 |
| Dataset | Type | Loss spread | Backbone spread | Best loss |
| framingham | CR | 0.036 | 0.059 | CRNLL |
| metabric_competing | CR | 0.068 | 0.022 | CRNLL |
| mimic | CR | 0.012 | – | CRNLL |
| mimiciii | CR | 0.011 | 0.020 | CRNLL |
| seer | CR | 0.043 | 0.021 | CRNLL |
| seer_digestion | CR | 0.029 | – | CRNLL |
| C-index spread | C-index / IBS spread | ||||
| Spread across | CR | single event | CR | single event | |
| the 3 losses | [0.015, 0.067] | [0.015, 0.068] | [0.46, 1.25] | [0.16, 1.05] | |
| the 4 backbones | [0.026, 0.113] | [0.024, 0.075] | [0.76, 4.09] | [0.14, 1.40] | |
| all cells of a dataset | [0.061, 0.182] | [0.048, 0.133] | [0.62, 1.76] | [0.18, 0.74] | |
| Competing risks (10 datasets) | Single event (10 datasets) | |||||||
| Loss | MLP | RealMLP | TabM | SurvTRACE | MLP | RealMLP | TabM | SurvTRACE |
| CRNLL | 0.261 | 0.250 | 0.351 | 0.258 | 0.146 | 0.150 | 0.152 | 0.125 |
| HazardLikelihood | 0.117 | 0.198 | 0.268 | 0.120 | 0.084 | 0.043 | 0.128 | 0.117 |
| HLRanking | 0.120 | 0.093 | 0.180 | 0.120 | 0.063 | 0.049 | 0.126 | 0.117 |
| Backbone | med. IBS | IBS worse | med. C-idx | C-idx better | |
| MLP | 22 | 17/22 | 13/22 | ||
| RealMLP | 19 | 14/19 | 12/19 | ||
| TabM | 20 | 17/20 | 13/20 | ||
| SurvTRACE | 13 | 9/13 | 7/13 | ||
| Pooled | 74 | 57/74 (77%) | 45/74 (61%) |
Survival Analysis
Survival analysis models when events such as disease progression, failure, or death occur, accounting for incomplete follow-up and censoring.
See all 106 papers