竞争风险生存分析:损失函数比网络结构更关键
The Loss or the Architecture? What Matters in Survival Analysis with Competing Risks
survival analysis 该怎么选 loss 还是 backbone?这篇用 23 个数据集拆开测:损失函数影响更大,表格配 RealMLP、图像配视觉编码器,组合直接拿最佳 Brier 分数。
一项研究将竞争风险生存分析中的损失函数与网络骨干拆开独立测试,覆盖 3 种损失、4 种骨干、23 个数据集(20 个表格、3 个病理图像)。结果显示直接建模累积发生率并做删失加权的损失带来的收益约是更换骨干的两倍。骨干选择需匹配模态:表格数据用 RealMLP 或 TabM 领先,图像数据用预训练视觉编码器领先,两者收益可叠加。非严格评分的 C-index 无法区分模型差异,组合出的 RealMLP/TabM 加累积发生率损失在基准中取得最佳 Brier 分数。
The Loss or the Architecture? What Matters in Survival Analysis with Competing Risks
Often, \emph{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.