Do Multiple Instance Learning Models Transfer?

TL;DR

多实例学习模型在不同器官任务上迁移表现优异,提升9.8%。

cs.CV 🟡 进阶级 2025-06-11 10 次浏览
Daniel Shao Richard J. Chen Andrew H. Song Joel Runevic Ming Y. Lu Tong Ding Faisal Mahmood
多实例学习 迁移学习 计算病理学 深度学习 医学影像

核心发现

方法论

研究评估了11种多实例学习(MIL)模型在21个预训练任务中的迁移能力,特别是形态学和分子亚型预测。使用预训练编码器对全片图像(WSI)进行特征提取,并通过可训练的聚合器进行特征汇总。实验采用了ABMIL、TransMIL等架构。

关键结果

  • 预训练的MIL模型在不同器官任务上表现优异,平均提升9.8%。
  • 在跨器官和任务的泛化能力上,预训练的pancancer数据集模型优于从头训练的模型。
  • 使用PC-108预训练的模型在19个任务中平均提升3.3%。

研究意义

该研究揭示了MIL模型在计算病理学中的强大适应性,尤其是在小数据集的情况下。通过迁移学习,模型在不同器官和任务中表现出色,表明该方法在医学影像分析中具有广泛应用潜力。

技术贡献

研究提供了标准化的MIL模型实现和预训练模型权重集合,展示了MIL模型在不同任务间的迁移能力,尤其是使用pancancer数据集进行预训练的有效性。

新颖性

首次系统性评估了MIL模型在计算病理学中的迁移能力,尤其是在不同器官任务间的迁移表现,填补了该领域的研究空白。

局限性

  • 模型在某些特定任务上的迁移效果可能不如预期,尤其是数据集差异较大的情况下。
  • 对实例级别的辅助损失依赖较大,可能影响模型的收敛性。

未来方向

未来研究可探索更复杂的MIL架构在迁移学习中的表现,以及在更大规模和多样化的数据集上的应用。

AI 总览摘要

多实例学习(MIL)模型在计算病理学中用于从全片图像生成临床意义的嵌入。然而,MIL在小规模、弱监督的临床数据集上常常表现不佳。与自然语言处理和传统计算机视觉领域不同,MIL模型的迁移能力尚不明确。

本研究系统性地评估了预训练MIL模型的迁移学习能力,分析了11种模型在21个预训练任务中的表现。结果显示,即使在不同器官的任务上,预训练的MIL模型也能显著优于从头训练的模型。此外,使用pancancer数据集进行预训练的模型在跨器官和任务的泛化能力上表现出色,且所需的预训练数据显著减少。

这些发现强调了MIL模型的强大适应性,并展示了利用迁移学习提升计算病理学性能的优势。研究还提供了标准化的MIL模型实现和预训练模型权重集合,便于在流行的计算病理学任务中应用。

深度分析

研究背景

多实例学习(MIL)是计算病理学中的一种重要方法,用于从全片图像生成有临床意义的嵌入。近年来,随着深度学习的发展,MIL在病理图像分析中的应用越来越广泛。然而,由于病理数据集通常较小且标注困难,MIL模型的性能受到限制。

核心问题

MIL模型在小规模、弱监督的临床数据集上表现不佳,限制了其在实际应用中的效果。与自然语言处理和计算机视觉领域不同,MIL模型的迁移能力尚未得到充分研究。

核心创新

本研究首次系统性地评估了MIL模型在计算病理学中的迁移能力,特别是在不同器官任务间的迁移表现。通过使用pancancer数据集进行预训练,模型在跨器官和任务的泛化能力上表现出色。

方法详解

  • �� 使用预训练编码器对全片图像进行特征提取
  • �� 使用可训练的聚合器进行特征汇总
  • �� 评估11种MIL架构在21个任务中的迁移能力
  • �� 使用AUROC、加权kappa等指标进行性能评估

实验设计

实验设计包括使用不同的MIL架构在21个预训练任务中的迁移能力评估。使用的指标包括AUROC、加权kappa和平衡准确率。实验还涉及不同预训练数据集的对比分析。

结果分析

研究发现,预训练的MIL模型在不同器官任务上表现优异,平均提升9.8%。特别是使用pancancer数据集进行预训练的模型在跨器官和任务的泛化能力上表现出色。

应用场景

MIL模型在计算病理学中的应用包括癌症分类、癌症分级和分子亚型预测。其强大的迁移能力使其在小数据集的情况下也能表现出色。

局限与展望

尽管研究展示了MIL模型的强大迁移能力,但在某些特定任务上的效果可能不如预期。此外,对实例级别的辅助损失依赖较大,可能影响模型的收敛性。

通俗解读 非专业人士也能看懂

想象你在厨房里做饭。每个食材就像全片图像中的一个小块,而菜谱就是模型的预训练任务。通过学习不同的菜谱,你可以更好地理解如何搭配食材,即使面对新的食材组合,你也能做出美味的菜肴。这就像MIL模型通过预训练在不同任务中的迁移能力。

简单解释 像给14岁少年讲一样

想象你在玩一个超级复杂的拼图游戏。每个拼图块都是一个小图像,而整个拼图就是一个全片图像。通过练习不同的拼图,你可以更快地完成新的拼图。这就像MIL模型通过预训练在不同任务中的迁移能力。是不是很酷?

术语表

多实例学习 (MIL)

一种机器学习方法,模型在没有单个实例标签的情况下对实例集合进行训练。

用于从全片图像生成临床意义的嵌入。

迁移学习

一种机器学习技术,通过在一个任务中学习到的知识来提高在另一个任务中的表现。

评估MIL模型在不同任务间的迁移能力。

全片图像 (WSI)

指数字化的人体组织切片图像,通常具有超高分辨率。

MIL模型用于从WSI中提取特征。

pancancer数据集

包含多种癌症类型的数据集,用于训练模型以提高其泛化能力。

用于评估模型在不同器官和任务间的迁移能力。

AUROC

受试者工作特征曲线下面积,用于评估分类模型的性能。

用于评估MIL模型在二分类任务中的表现。

开放问题 这项研究留下的未解疑问

  • 1 MIL模型在极小数据集上的迁移能力仍需进一步研究。
  • 2 如何优化实例级别的辅助损失以提高模型的收敛性。

应用场景

近期应用

癌症分类

通过迁移学习提升MIL模型在小数据集上的分类性能,帮助医生更准确地诊断癌症。

远期愿景

跨器官病理分析

开发能够在不同器官间高效迁移的MIL模型,推动个性化医学的发展。

原文摘要

Multiple Instance Learning (MIL) is a cornerstone approach in computational pathology (CPath) for generating clinically meaningful slide-level embeddings from gigapixel tissue images. However, MIL often struggles with small, weakly supervised clinical datasets. In contrast to fields such as NLP and conventional computer vision, where transfer learning is widely used to address data scarcity, the transferability of MIL models remains poorly understood. In this study, we systematically evaluate the transfer learning capabilities of pretrained MIL models by assessing 11 models across 21 pretraining tasks for morphological and molecular subtype prediction. Our results show that pretrained MIL models, even when trained on different organs than the target task, consistently outperform models trained from scratch. Moreover, pretraining on pancancer datasets enables strong generalization across organs and tasks, outperforming slide foundation models while using substantially less pretraining data. These findings highlight the robust adaptability of MIL models and demonstrate the benefits of leveraging transfer learning to boost performance in CPath. Lastly, we provide a resource which standardizes the implementation of MIL models and collection of pretrained model weights on popular CPath tasks, available at https://github.com/mahmoodlab/MIL-Lab

cs.CV