Benchmarking weakly-supervised deep learning pipelines for whole slide classification in computational pathology

Research output: Contribution to journalResearch articleContributedpeer-review

Contributors

  • Narmin Ghaffari Laleh - , RWTH Aachen University (Author)
  • Hannah Sophie Muti - , RWTH Aachen University (Author)
  • Chiara Maria Lavinia Loeffler - , RWTH Aachen University (Author)
  • Amelie Echle - , RWTH Aachen University (Author)
  • Oliver Lester Saldanha - , RWTH Aachen University (Author)
  • Faisal Mahmood - , Harvard University (Author)
  • Ming Y. Lu - , Harvard University (Author)
  • Christian Trautwein - , RWTH Aachen University (Author)
  • Rupert Langer - , Kepler University Hospital (Author)
  • Bastian Dislich - , University of Bern (Author)
  • Roman D. Buelow - , RWTH Aachen University (Author)
  • Heike Irmgard Grabsch - , Maastricht University, University of Leeds (Author)
  • Hermann Brenner - , German Cancer Research Center (DKFZ) (Author)
  • Jenny Chang-Claude - , German Cancer Research Center (DKFZ), University of Hamburg (Author)
  • Elizabeth Alwers - , German Cancer Research Center (DKFZ) (Author)
  • Titus J. Brinker - , German Cancer Research Center (DKFZ) (Author)
  • Firas Khader - , RWTH Aachen University (Author)
  • Daniel Truhn - , RWTH Aachen University (Author)
  • Nadine T. Gaisa - , RWTH Aachen University (Author)
  • Peter Boor - , RWTH Aachen University (Author)
  • Michael Hoffmeister - , German Cancer Research Center (DKFZ) (Author)
  • Volkmar Schulz - , RWTH Aachen University, Fraunhofer Institute for Digital Medicine, Hyperion Hybrid Imaging Systems GmbH (Author)
  • Jakob Nikolas Kather - , Else Kröner Fresenius Center for Digital Health, RWTH Aachen University, University of Leeds (Author)

Abstract

Artificial intelligence (AI) can extract visual information from histopathological slides and yield biological insight and clinical biomarkers. Whole slide images are cut into thousands of tiles and classification problems are often weakly-supervised: the ground truth is only known for the slide, not for every single tile. In classical weakly-supervised analysis pipelines, all tiles inherit the slide label while in multiple-instance learning (MIL), only bags of tiles inherit the label. However, it is still unclear how these widely used but markedly different approaches perform relative to each other. We implemented and systematically compared six methods in six clinically relevant end-to-end prediction tasks using data from N=2980 patients for training with rigorous external validation. We tested three classical weakly-supervised approaches with convolutional neural networks and vision transformers (ViT) and three MIL-based approaches with and without an additional attention module. Our results empirically demonstrate that histological tumor subtyping of renal cell carcinoma is an easy task in which all approaches achieve an area under the receiver operating curve (AUROC) of above 0.9. In contrast, we report significant performance differences for clinically relevant tasks of mutation prediction in colorectal, gastric, and bladder cancer. In these mutation prediction tasks, classical weakly-supervised workflows outperformed MIL-based weakly-supervised methods for mutation prediction, which is surprising given their simplicity. This shows that new end-to-end image analysis pipelines in computational pathology should be compared to classical weakly-supervised methods. Also, these findings motivate the development of new methods which combine the elegant assumptions of MIL with the empirically observed higher performance of classical weakly-supervised approaches. We make all source codes publicly available at https://github.com/KatherLab/HIA, allowing easy application of all methods to any similar task.

Details

Original languageEnglish
Article number102474
JournalMedical Image Analysis
Volume79
Publication statusPublished - Jul 2022
Peer-reviewedYes

External IDs

PubMed 35588568

Keywords

Sustainable Development Goals

Keywords

  • Artificial intelligence, Computational pathology, Convolutional neural networks, Multiple-Instance Learning, Vision transformers, Weakly-supervised deep learning