Search papers, labs, and topics across Lattice.
This paper introduces a framework for evaluating the quality of Multiple Instance Learning (MIL) heatmaps in computational histopathology without requiring additional labels. Through a large-scale benchmark experiment, they assessed six explanation methods across various histopathology tasks, MIL architectures (Attention, Transformer, Mamba), and patch encoders (UNI2, Virchow2). The results indicate that perturbation-based methods (Single), LRP, and integrated gradients (IG) consistently outperform attention-based and gradient-based saliency heatmaps in reflecting model decision mechanisms.
Attention heatmaps in MIL models for histopathology are often misleading, and simpler methods like perturbation or LRP provide more faithful explanations.
Multiple instance learning (MIL) has enabled substantial progress in computational histopathology, where a large amount of patches from gigapixel whole slide images are aggregated into slide-level predictions. Heatmaps are widely used to validate MIL models and to discover tissue biomarkers. Yet, the validity of these heatmaps has barely been investigated. In this work, we introduce a general framework for evaluating the quality of MIL heatmaps without requiring additional labels. We conduct a large-scale benchmark experiment to assess six explanation methods across histopathology task types (classification, regression, survival), MIL model architectures (Attention-, Transformer-, Mamba-based), and patch encoder backbones (UNI2, Virchow2). Our results show that explanation quality mostly depends on MIL model architecture and task type, with perturbation ("Single"), layer-wise relevance propagation (LRP), and integrated gradients (IG) consistently outperforming attention-based and gradient-based saliency heatmaps, which often fail to reflect model decision mechanisms. We further demonstrate the advanced capabilities of the best-performing explanation methods: (i) We provide a proof-of-concept that MIL heatmaps of a bulk gene expression prediction model can be correlated with spatial transcriptomics for biological validation, and (ii) showcase the discovery of distinct model strategies for predicting human papillomavirus (HPV) infection from head and neck cancer slides. Our work highlights the importance of validating MIL heatmaps and establishes that improved explainability can enable more reliable model validation and yield biological insights, making a case for a broader adoption of explainable AI in digital pathology. Our code is provided in a public GitHub repository: https://github.com/bifold-pathomics/xMIL/tree/xmil-journal
