arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.09449cs.CVcs.LG
收藏

从相关性到因果性:基于最大池化的多实例学习导致更稳健的整张滑动图像分类

From Correlation to Causation: Max-Pooling-Based Multi-Instance Learning Leads to More Robust Whole Slide Image Classification

UniSA STEM, University of South Australia(南澳大利亚大学STEM学院);School of Computer Science and Engineering, Southeast University(东南大学计算机科学与工程学院);Key Lab. of Computer Network and Information Integration (Southeast University), MoE, China(计算机网络与信息集成重点实验室(东南大学),教育部,中国);School of Information and Physical Sciences, The University of Newcastle(新castle大学信息与物理科学学院);Mohamed bin Zayed University of Artificial Intelligence (MBZUAI)(穆罕默德·本·扎耶德人工智能大学)

Xin Liu, Weijia Zhang, Wei Tang, Thuc Duy Le, Jiuyong Li, Lin Liu, Min-Ling Zhang

AI总结:

本文提出FocusMIL,通过结合最大池化与变分信息瓶颈,改进多实例学习以更稳健地处理整张滑动图像分类任务。

摘要:

在整张滑动图像(WSI)分析中,基于注意力的多实例学习(MIL)模型容易受到虚假相关性的影响,并在领域转移下性能下降。这些方法可能会将高注意力权重分配给非肿瘤区域,如染色偏差或伪影,导致肿瘤区域定位不可靠。在本文中,我们从因果角度重新审视基于最大池化的MIL方法。在温和假设下,我们的理论结果表明,最大池化鼓励模型关注因果因素并忽略偏差相关因素。此外,我们发现现有的基于最大池化的 方法可能通过机械记忆实例特征来过拟合训练集,并未能学习有意义的模式。为了解决这些问题,我们提出了FocusMIL,它将最大池化与实例级变分信息瓶颈(VIB)相结合,以学习紧凑、预测性的潜在表示,并采用多袋小批量方案来稳定优化。我们在三个真实世界数据集和一个半合成数据集上进行了全面实验。结果表明,通过捕捉因果因素,FocusMIL在分布外场景和实例级肿瘤区域定位任务中表现出显著优势。

英文摘要:

In whole slide images (WSIs) analysis, attention-based multi-instance learning (MIL) models are susceptible to spurious correlations and degrade under domain shift. These methods may assign high attention weights to non-tumor regions, such as staining biases or artifacts, leading to unreliable tumor region localization. In this paper, we revisit max-pooling-based MIL methods from a causal perspective. Under mild assumptions, our theoretical results demonstrate that max-pooling encourages the model to focus on causal factors while ignoring bias-related factors. Furthermore, we discover that existing max-pooling-based methods may overfit the training set through rote memorization of instance features and fail to learn meaningful patterns. To address these issues, we propose FocusMIL, which couples max-pooling with an instance-level variational information bottleneck (VIB) to learn compact, predictive latent representations, and employs a multi-bag mini-batch scheme to stabilize optimization. We conduct comprehensive experiments on three real-world datasets and one semi-synthetic dataset. The results show that, by capturing causal factors, FocusMIL exhibits significant advantages in out-of-distribution scenarios and instance-level tumor region localization tasks.

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

↑