nnMIL提升病理图像诊断的通用性与可靠性
nnMIL: A generalizable multiple instance learning framework for computational pathology
- 通过双层随机采样实现大批次优化,适配多种模型和数据集
- 在4万张切片、35项任务中均超越现有方法,生存分层准确
- 支持不确定性估计,适合临床部署与跨模型迁移
计算病理学有望改善诊断并指导治疗。尽管当前病理基础模型能从大规模全切片图像(WSIs)中提取丰富的局部特征,但现有聚合方法在设计上限制了泛化能力与可靠性。本文提出nnMIL,一种简单且通用的多实例学习框架,将局部特征映射为可靠的切片级临床预测。nnMIL在图像块和特征层面引入随机采样,支持大批次优化、任务感知采样策略,并实现跨数据集与模型架构的高效可扩展训练。轻量级聚合器采用滑动窗口推理生成集成预测,支持合理的不确定性估计。在涵盖40,000张WSI、35个临床任务及4种病理基础模型的实验中,nnMIL在疾病诊断、组织学分型、分子生物标志物检测和泛癌预后预测方面持续优于现有MIL方法。其还展现出强跨模型泛化能力、可靠不确定性量化及多个外部队列中的稳健生存分层能力。结论:nnMIL为将病理基础模型转化为临床可解释预测提供了实用且通用的解决方案,推动真实场景下可信AI系统的开发与应用。
原文摘要 · Abstract (English)
Computational pathology holds substantial promise for improving diagnosis and guiding treatment decisions. Recent pathology foundation models enable the extraction of rich patch-level representations from large-scale whole-slide images (WSIs), but current approaches for aggregating these features into slide-level predictions remain constrained by design limitations that hinder generalizability and reliability. Here we present nnMIL, a simple yet broadly applicable multiple-instance learning framework that connects patch-level foundation models to robust slide-level clinical prediction. nnMIL introduces random sampling at both the patch and feature levels, enabling large-batch optimization, task-aware sampling strategies, and efficient and scalable training across datasets and model architectures. A lightweight aggregator performs sliding-window inference to generate ensemble slide-level predictions and supports principled uncertainty estimation. Across 40,000 WSIs encompassing 35 clinical tasks and four pathology foundation models, nnMIL consistently outperformed existing MIL methods for disease diagnosis, histologic subtyping, molecular biomarker detection, and pan-cancer prognosis prediction. It further demonstrated strong cross-model generalization, reliable uncertainty quantification, and robust survival stratification in multiple external cohorts. In conclusion, nnMIL offers a practical and generalizable solution for translating pathology foundation models into clinically meaningful predictions, advancing the development and deployment of reliable AI systems in real-world settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。