arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2407.00664cs.CVcs.AI
收藏

SCMIL:用于预测全切片图像中癌症生存概率分布的稀疏上下文感知多实例学习

SCMIL: Sparse Context-aware Multiple Instance Learning for Predicting Cancer Survival Probability Distribution in Whole Slide Images

Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);University of Chinese Academy of Sciences(中国科学院大学)

Zekang Yang, Hong Liu, Xiangdong Wang

AI总结:

SCMIL通过稀疏上下文感知多实例学习、SoftFilter斑块过滤模块和基于寄存器的混合密度网络,有效建模WSI中任务相关斑块的交互并预测癌症生存概率分布,在TCGA的LUAD和KIRC数据集上优于现有方法。

摘要:

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

英文摘要:

Cancer survival prediction is a challenging task that involves analyzing of the tumor microenvironment within Whole Slide Image (WSI). Previous methods cannot effectively capture the intricate interaction features among instances within the local area of WSI. Moreover, existing methods for cancer survival prediction based on WSI often fail to provide better clinically meaningful predictions. To overcome these challenges, we propose a Sparse Context-aware Multiple Instance Learning (SCMIL) framework for predicting cancer survival probability distributions. SCMIL innovatively segments patches into various clusters based on their morphological features and spatial location information, subsequently leveraging sparse self-attention to discern the relationships between these patches with a context-aware perspective. Considering many patches are irrelevant to the task, we introduce a learnable patch filtering module called SoftFilter, which ensures that only interactions between task-relevant patches are considered. To enhance the clinical relevance of our prediction, we propose a register-based mixture density network to forecast the survival probability distribution for individual patients. We evaluate SCMIL on two public WSI datasets from the The Cancer Genome Atlas (TCGA) specifically focusing on lung adenocarcinom (LUAD) and kidney renal clear cell carcinoma (KIRC). Our experimental results indicate that SCMIL outperforms current state-of-the-art methods for survival prediction, offering more clinically meaningful and interpretable outcomes. Our code is accessible at https://github.com/yang-ze-kang/SCMIL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑