arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.04558cs.LGcs.AIcs.CV

降低自监督学习复杂度可提升计算病理学中的弱监督分类性能

Reducing self-supervised learning complexity improves weakly-supervised classification performance in computational pathology

University Hospital and Faculty of Medicine Carl Gustav Carus, TUD Dresden University of Technology(德累斯顿工业大学卡尔·古斯塔夫·卡鲁斯大学医院及医学院);National Center for Tumor Diseases (NCT), University Hospital Heidelberg(海德堡大学医院国家肿瘤疾病中心);Leeds Institute of Medical Research at St James’s, University of Leeds(利兹大学圣詹姆斯利兹医学研究所)

Tim Lenz, Omar S. M. El Nahhas, Marta Ligero, Jakob Nikolas Kather

AI总结:

针对计算病理学中自监督学习资源需求高的问题,研究通过调整数据量、架构与算法,在消费级硬件上实现训练时长减少90%的同时提升弱监督分类性能,提出适配低资源环境的方案。

摘要:

深度学习模型已被成功用于从常规可用的组织学数据中提取具有临床指导意义的见解。通常而言,这些模型需要临床医生完成标注,而这类标注资源稀缺且生成成本高昂。自监督学习(SSL)方法的出现消除了这一障碍,使得对未标注数据开展大规模分析成为可能。然而,近期的自监督学习方法采用了愈发庞大的模型架构与更大规模的数据集,导致数据量、硬件要求以及总体成本快速攀升,仅有少数机构能够获取这些资源。为此,我们在消费级硬件条件下,研究了计算病理学中对比式自监督学习的复杂度与分类性能的关联。具体而言,我们分析了数据量、架构和算法的调整对下游分类任务的影响,重点关注其对计算资源的作用。我们在一个大型公共患者队列上训练了乳腺癌基础模型,并以弱监督方式在两个外部公共患者队列的多项下游分类任务上对模型进行了验证。我们的实验表明,我们能够在将自监督学习训练时长减少90%的同时,提升下游分类性能。综上,我们提出了一系列调整方案,使自监督学习能够在资源不充裕的环境下应用于计算病理学领域。

英文摘要:

Deep Learning models have been successfully utilized to extract clinically actionable insights from routinely available histology data. Generally, these models require annotations performed by clinicians, which are scarce and costly to generate. The emergence of self-supervised learning (SSL) methods remove this barrier, allowing for large-scale analyses on non-annotated data. However, recent SSL approaches apply increasingly expansive model architectures and larger datasets, causing the rapid escalation of data volumes, hardware prerequisites, and overall expenses, limiting access to these resources to few institutions. Therefore, we investigated the complexity of contrastive SSL in computational pathology in relation to classification performance with the utilization of consumer-grade hardware. Specifically, we analyzed the effects of adaptations in data volume, architecture, and algorithms on downstream classification tasks, emphasizing their impact on computational resources. We trained breast cancer foundation models on a large public patient cohort and validated them on various downstream classification tasks in a weakly supervised manner on two external public patient cohorts. Our experiments demonstrate that we can improve downstream classification performance whilst reducing SSL training duration by 90%. In summary, we propose a set of adaptations which enable the utilization of SSL in computational pathology in non-resource abundant environments.

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

↑