xMIL: Insightful Explanations for Multiple Instance Learning in Histopathology
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866913638658342912 |
|---|---|
| author | Hense, Julius Idaji, Mina Jamshidi Eberle, Oliver Schnake, Thomas Dippel, Jonas Ciernik, Laure Buchstab, Oliver Mock, Andreas Klauschen, Frederick Müller, Klaus-Robert |
| author_facet | Hense, Julius Idaji, Mina Jamshidi Eberle, Oliver Schnake, Thomas Dippel, Jonas Ciernik, Laure Buchstab, Oliver Mock, Andreas Klauschen, Frederick Müller, Klaus-Robert |
| contents | Multiple instance learning (MIL) is an effective and widely used approach for weakly supervised machine learning. In histopathology, MIL models have achieved remarkable success in tasks like tumor detection, biomarker prediction, and outcome prognostication. However, MIL explanation methods are still lagging behind, as they are limited to small bag sizes or disregard instance interactions. We revisit MIL through the lens of explainable AI (XAI) and introduce xMIL, a refined framework with more general assumptions. We demonstrate how to obtain improved MIL explanations using layer-wise relevance propagation (LRP) and conduct extensive evaluation experiments on three toy settings and four real-world histopathology datasets. Our approach consistently outperforms previous explanation attempts with particularly improved faithfulness scores on challenging biomarker prediction tasks. Finally, we showcase how xMIL explanations enable pathologists to extract insights from MIL models, representing a significant advance for knowledge discovery and model debugging in digital histopathology. Codes are available at: https://github.com/bifold-pathomics/xMIL. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2406_04280 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | xMIL: Insightful Explanations for Multiple Instance Learning in Histopathology Hense, Julius Idaji, Mina Jamshidi Eberle, Oliver Schnake, Thomas Dippel, Jonas Ciernik, Laure Buchstab, Oliver Mock, Andreas Klauschen, Frederick Müller, Klaus-Robert Machine Learning Computer Vision and Pattern Recognition Multiple instance learning (MIL) is an effective and widely used approach for weakly supervised machine learning. In histopathology, MIL models have achieved remarkable success in tasks like tumor detection, biomarker prediction, and outcome prognostication. However, MIL explanation methods are still lagging behind, as they are limited to small bag sizes or disregard instance interactions. We revisit MIL through the lens of explainable AI (XAI) and introduce xMIL, a refined framework with more general assumptions. We demonstrate how to obtain improved MIL explanations using layer-wise relevance propagation (LRP) and conduct extensive evaluation experiments on three toy settings and four real-world histopathology datasets. Our approach consistently outperforms previous explanation attempts with particularly improved faithfulness scores on challenging biomarker prediction tasks. Finally, we showcase how xMIL explanations enable pathologists to extract insights from MIL models, representing a significant advance for knowledge discovery and model debugging in digital histopathology. Codes are available at: https://github.com/bifold-pathomics/xMIL. |
| title | xMIL: Insightful Explanations for Multiple Instance Learning in Histopathology |
| topic | Machine Learning Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2406.04280 |