May, 2024

通过多标签学习和深度模型解释推进头颈癌生存预测

TL;DR我们提出了 IMLSP,一个可解释的多标签多模态深度存活预测框架,用于同时预测多个头颈癌存活结果,并提供深度预测过程的时间事件特定的可视化解释。我们采用多任务逻辑回归层将存活预测从回归问题转化为多时间点分类任务,并能够同时预测多个相关存活结果。通过公开可用的 RADCURE HNC 数据集的评估,我们的方法在所有存活结果上优于相应的单模态模型和单标签模型。生成的激活图显示,该模型在做出决策时主要关注肿瘤和淋巴结体积,并且对于高风险患者和低风险患者的感兴趣体积有所不同。我们证明多标签学习策略可以提高学习效率和预后性能,而可解释的存活预测模型有助于理解人工智能的决策过程,并促进个性化治疗。