摘要
生存分析(SA)模型描述了事件发生前的时间,应用于医学、国防、金融和航空航天等领域。最近的研究表明,神经网络(NNs)能够有效捕捉SA中的复杂数据模式,而简单的广义线性模型在这方面往往表现不佳。然而,数据集的不确定性(如噪声测量、人为误差)会降低NN模型的性能。为此,我们利用NN验证的进展,开发了针对鲁棒、全参数SA模型的训练目标。具体而言,我们提出了一种基于Min-Max优化问题的对抗性鲁棒损失函数。我们采用CROWN-区间边界传播(CROWN-IBP)来解决解决这一Min-Max问题固有的计算挑战。在10个SurvSet数据集上的评估结果显示,我们的方法——带有对抗性正则化的生存分析(SAWAR),在各种协变量扰动下,相对于负对数似然(NegLL)、综合Brier评分(IBS)和一致性指数(CI)等指标,始终优于基线对抗性训练方法和最先进的(SOTA)深度SA模型。因此,我们证明了对抗性鲁棒性增强了SA的预测性能和校准,通过减少数据不确定性和提高跨多样数据集的泛化能力,相比基线模型最多可提升150%。