arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2507.00292cs.CVcs.AI

降低数字病理学中多实例学习方法的变异性

Reducing Variability of Multiple Instance Learning Methods for Digital Pathology

LTCI, Télécom Paris, Institut Polytechnique de Paris, France(LTCI、Telecom Paris、巴黎高等理工学院、法国);Paris Saint-Joseph Hospital, France(巴黎圣约瑟夫医院、法国)

Ali Mammadov, Loïc Le Folgoc, Guillaume Hocquet, Pietro Gori

AI总结:

针对数字病理学中多实例学习方法在不同运行间性能波动大的问题,本文提出一种多保真度模型融合策略,通过训练多个模型并基于验证分数平均稳定模型,在超过2000次实验中验证其能有效降低变异性、简化调参并提高可复现性。

摘要:

数字病理学通过将组织样本数字化为全切片图像(WSIs)彻底改变了该领域。然而,WSIs 的高分辨率和巨大尺寸给深度学习模型的应用带来了显著挑战。作为一种解决方案,WSIs 通常被分割成较小的图像块,并为每张切片赋予一个全局标签(即诊断标签),而不是进行(过于)昂贵的逐像素标注。通过将每张切片视为一个图像块包,多实例学习(MIL)方法已成为 WSI 分类的合适解决方案。MIL 方法的一个主要缺点是其在不同运行中的性能变异性很高,在测试集上的差异可达 10-15 个 AUC 点,这使得可靠地比较不同的 MIL 方法变得困难。这种变异性主要来自三个因素:i)权重初始化,ii)批次(打乱)顺序,iii)学习率。为了解决这一问题,我们引入了一种面向 MIL 方法的多保真度模型融合策略。我们首先训练多个模型若干个 epoch,并根据验证分数对最稳定且最有前景的模型进行平均。该方法可应用于任何现有的 MIL 模型以降低性能变异性。它还能简化超参数调优并提高可复现性,同时保持计算效率。我们使用 2 个不同的数据集、3 种初始化策略和 5 种 MIL 方法,在 WSI 分类任务上对我们的方法进行了广泛验证,总计进行了超过 2000 次实验。

英文摘要:

Digital pathology has revolutionized the field by enabling the digitization of tissue samples into whole slide images (WSIs). However, the high resolution and large size of WSIs present significant challenges when it comes to applying Deep Learning models. As a solution, WSIs are often divided into smaller patches with a global label (\textit{i.e., diagnostic}) per slide, instead of a (too) costly pixel-wise annotation. By treating each slide as a bag of patches, Multiple Instance Learning (MIL) methods have emerged as a suitable solution for WSI classification. A major drawback of MIL methods is their high variability in performance across different runs, which can reach up to 10-15 AUC points on the test set, making it difficult to compare different MIL methods reliably. This variability mainly comes from three factors: i) weight initialization, ii) batch (shuffling) ordering, iii) and learning rate. To address that, we introduce a Multi-Fidelity, Model Fusion strategy for MIL methods. We first train multiple models for a few epochs and average the most stable and promising ones based on validation scores. This approach can be applied to any existing MIL model to reduce performance variability. It also simplifies hyperparameter tuning and improves reproducibility while maintaining computational efficiency. We extensively validate our approach on WSI classification tasks using 2 different datasets, 3 initialization strategies and 5 MIL methods, for a total of more than 2000 experiments.

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

↑