摘要
癌症生存预测是一项具有挑战性的任务,涉及分析全切片图像 (WSI) 中的肿瘤微环境。以往的方法无法有效地捕获 WSI 局部区域内实例之间的复杂交互特征。此外,现有的基于 WSI 的癌症生存预测方法往往无法提供更好的临床意义上的预测。为了克服这些挑战,我们提出了一种稀疏上下文感知多实例学习 (SCMIL) 框架,用于预测癌症生存概率分布。SCMIL 创新性地根据形态特征和空间位置信息将图像块分割成不同的簇,随后利用稀疏自注意力机制从上下文感知的角度来识别这些图像块之间的关系。考虑到许多图像块与任务无关,我们引入了一个可学习的图像块过滤模块 SoftFilter,它确保只考虑任务相关图像块之间的交互。为了增强我们预测的临床相关性,我们提出了一种基于寄存器的混合密度网络,用于预测单个患者的生存概率分布。我们在来自癌症基因组图谱 (TCGA) 的两个公共 WSI 数据集上评估了 SCMIL,重点关注肺腺癌 (LUAD) 和肾透明细胞癌 (KIRC)。我们的实验结果表明,SCMIL 在生存预测方面优于当前最先进的方法,提供了更具临床意义和可解释性的结果。我们的代码可在 https://github.com/yang-ze-kang/SCMIL 上获取。