Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Uncertainty Estimation in Pathology Foundation Models via Deep Mutual Learning

About

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 $\mathtt{DICE}$, a plug-and-play framework that ensembles $K$ frozen PFMs and models their disagreement as a proxy for uncertainty estimation. To ensure this proxy yields meaningful estimates, we align the ensemble members via deep mutual learning, and theoretically show that this objective upper-bounds the model uncertainty. Additionally, we demonstrate that the ensemble's consensus localizes abnormalities at the patch level without any explicit supervision. We evaluate $\mathtt{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, $\mathtt{DICE}$ takes a crucial step toward translating PFMs into uncertainty-aware decision-support systems.

Gb\`egninougbo Aurel Davy Tchokponhoue, Sevda \"O\u{g}\"ut, Ali Idri, Dorina Thanou, Pascal Frossard• 2026

Related benchmarks

TaskDatasetResultRank
Whole Slide Image classificationCAMELYON16 (test)--
171
Slide-level calibrationPANDA (test)
NLL48.9
22
Slide-level calibrationCAMELYON16 (test)
NLL11.5
22
Slide-level calibrationCAMELYON17 (test)
NLL50.5
22
Slide-level classificationPANDA held-out (test)
F1 Score76.6
22
Slide-level classificationCamelyon 17 (test)
F1 Score79.3
22
Patch-level Tumor LocalizationCAMELYON16 (test)
AUC93
11
Patch-level Tumor LocalizationPANDA (test)
AUC81.1
11
Patch-level Tumor LocalizationCAMELYON17 (test)
AUC96.2
11
Slide-level classificationCAMELYON16 held-out (test)
AUC98.7
11
Showing 10 of 10 rows

Other info

Follow for update