arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2204.05044eess.IVcs.LGstat.ML
收藏

从现代CNN到视觉Transformer:评估深度学习模型在组织病理学中的性能、鲁棒性和分类策略

From Modern CNNs to Vision Transformers: Assessing the Performance, Robustness, and Classification Strategies of Deep Learning Models in Histopathology

Fraunhofer Heinrich Hertz Institute(弗劳恩霍夫海因里希·赫兹研究所);University of Oldenburg(奥尔登堡大学)

Maximilian Springenberg, Annika Frommholz, Markus Wenzel, Eva Weicken, Jackie Ma, Nils Strodthoff

AI总结:

本研究提出一种综合评估方法,在五个组织病理学数据集上系统比较现代CNN与视觉Transformer的性能、鲁棒性及分类策略,并利用图像翻译模型评估染色变化鲁棒性,为未来模型设计提供可迁移见解。

摘要:

尽管机器学习正在改变组织病理学领域,但该领域缺乏基于除单纯分类准确性之外的基本且互补的质量要求对最先进模型进行全面评估。为了填补这一空白,我们开发了一种新方法,广泛评估各种分类模型,包括最近的视觉Transformer和卷积神经网络,如ConvNeXt、ResNet(BiT)、Inception、ViT和Swin Transformer,并考虑有无监督或自监督预训练的情况。我们在五个广泛使用的组织病理学数据集上彻底测试了这些模型,这些数据集包含乳腺、胃和结直肠癌的全切片图像,并开发了一种使用图像到图像翻译模型的新方法,以评估癌症分类模型对染色变化的鲁棒性。此外,我们将现有的可解释性方法扩展到以前未研究的模型,并系统性地揭示了模型分类策略的见解,这些见解可迁移到未来的模型架构中。

英文摘要:

While machine learning is currently transforming the field of histopathology, the domain lacks a comprehensive evaluation of state-of-the-art models based on essential but complementary quality requirements beyond a mere classification accuracy. In order to fill this gap, we developed a new methodology to extensively evaluate a wide range of classification models, including recent vision transformers, and convolutional neural networks such as: ConvNeXt, ResNet (BiT), Inception, ViT and Swin transformer, with and without supervised or self-supervised pretraining. We thoroughly tested the models on five widely used histopathology datasets containing whole slide images of breast, gastric, and colorectal cancer and developed a novel approach using an image-to-image translation model to assess the robustness of a cancer classification model against stain variations. Further, we extended existing interpretability methods to previously unstudied models and systematically reveal insights of the models' classifications strategies that can be transferred to future model architectures.

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

↑