Uncertainty Estimation in Pathology Foundation Models via Deep Mutual Learning
Not on the map. Why?
Abstract (arXiv)
Pathology foundation models (PFMs) offer generalizable representations for whole-slide image (WSI) analysis, yet their clinical adoption remains limited. Specifically, their predictions lack reliable confidence estimates, and no single PFM is universally best across tasks, which severely undermines trust in medical settings. To overcome this, we propose DICE, a plug-and-play framework that ensembles K frozen PFMs and estimates uncertainty based on their consensus. We align the ensemble members via deep mutual learning and theoretically show that this objective controls an upper bound on epistemic uncertainty. Additionally, we demonstrate that the ensemble localizes abnormalities at the patch level without any explicit supervision. We evaluate DICE on three challenging WSI benchmarks. Notably, our framework provides reliable uncertainty estimates that accurately flag failure-prone cases under in- and out-of-distribution settings, while matching or outperforming SOTA baselines in classification, calibration, and localization. Overall, DICE takes a crucial step toward translating PFMs into uncertainty-aware decision-support systems.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 46 references that are in the map. First posted on arXiv 29 Jun 2026.
Builds on
23 of 46 references.
References · 23
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2022CoCa: Contrastive Captioners are Image-Text Foundation Modelscited by 1,274
- 2018Attention-based Deep Multiple Instance Learningcited by 1,095
- 2020Data Efficient and Weakly Supervised Computational Pathology on Whole Slide Imagescited by 745
- 2018Evidential Deep Learning to Quantify Classification Uncertaintycited by 732
- 2017Deep Mutual Learningcited by 628
- 2021TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classificationcited by 555
- 2020Dual-stream Multiple Instance Learning Network for Whole Slide Image Classification with Self-supervised Contrastive Learningcited by 478
- 2020Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awarenesscited by 389
- 2024Virchow2: Scaling Self-Supervised Mixed Magnification Models in Pathologycited by 161
11 more
- 2024HEST-1k: A Dataset for Spatial Transcriptomics and Histology Image Analysiscited by 107
- 2024PRISM: A Multi-Modal Generative Foundation Model for Slide-Level Histopathologycited by 85
- 2024A Clinical Benchmark of Public Self-Supervised Pathology Foundation Modelscited by 66
- 2025Accelerating Data Processing and Benchmarking of AI Models for Pathologycited by 65
- 2024Hibou: A Family of Foundational Vision Transformers for Pathologycited by 57
- 2024Gramian Multimodal Representation Learning and Alignmentcited by 54
- 2024Benchmarking foundation models as feature extractors for weakly-supervised computational pathologycited by 49
- 2025Fusion of Multi-scale Heterogeneous Pathology Foundation Models for Whole Slide Image Analysiscited by 5
- 2025Ensemble learning of pathology foundation models for precision oncologycited by 4
- 2025SGPMIL: Sparse Gaussian Process Multiple Instance Learningcited by 2
- 2026GrapHist: Graph Self-Supervised Learning for Histopathologycited by 2
Built on by
Not cited yet.