Uncertainty-aware and causal test-time adaptive foundation model for robust colorectal cancer pathology diagnosis
Colorectal cancer (CRC) is a leading malignancy worldwide, where histopathological assessment of hematoxylin and eosin (H&E) stained whole-slide images remains the diagnostic gold standard. However, current computational pathology models suffer from domain shifts, unreliable uncertainty estimation, and spurious correlations, limiting clinical reliability. We present UAD-FM, an Uncertainty-Aware and Causally Adaptive Foundation Model that integrates epistemic-aleatoric uncertainty decomposition, causal test-time adaptation using do-interventions, and post-hoc calibration for trustworthy inference. Across five public CRC datasets (TCGA-COAD/READ, CRAG, DigestPath 2019, NCT-CRC-HE-100K, and LC25000), UAD-FM achieves superior accuracy, calibration, and domain robustness compared with existing foundation models and adaptation baselines. The model also produces interpretable uncertainty maps to support human-AI collaboration. UAD-FM provides a unified, transparent framework for reliable and generalizable CRC pathology diagnosis across heterogeneous clinical settings.
Authors
- Hongxue Meng (ORCID: https://orcid.org/0000-0002-1831-2737)
- Xinyue Zhang (ORCID: https://orcid.org/0000-0002-4176-9516)
- Meihong Yan
- Siyuan Ma (ORCID: https://orcid.org/0000-0002-9207-1062)
- Lei Cao (ORCID: https://orcid.org/0000-0002-8666-0127)
- Shenghan Lou (ORCID: https://orcid.org/0000-0001-8826-3160)
- Xiao Zhang (ORCID: https://orcid.org/0009-0007-3220-8892)
- Gary Mo (ORCID: https://orcid.org/0000-0001-7805-5077)
- Keru Ma (ORCID: https://orcid.org/0000-0003-3397-5667)
- C. Li (ORCID: https://orcid.org/0000-0003-2323-9471)
- Yuze Huang (ORCID: https://orcid.org/0000-0003-4580-2237)
- Hao Li (ORCID: https://orcid.org/0000-0001-7191-5118)
- Hao Wang (ORCID: https://orcid.org/0000-0003-2661-5495)
- Huiying Li (ORCID: https://orcid.org/0000-0003-2345-5908)
- Peng Han (ORCID: https://orcid.org/0009-0002-6742-0539)
- Haonan Xie (ORCID: https://orcid.org/0000-0001-5623-2845)
Institutions
- Harbin Medical University (CN)
- Nanyang Technological University (SG)
- Third Affiliated Hospital of Harbin Medical University (CN)
- China Rural Technology Development Center (CN)
- Education Department of Heilongjiang Province (CN)
- Heilongjiang Institute of Wood Science (CN)
- Heilongjiang Institute of Technology (CN)
- Heilongjiang University (CN)
Publication Details
- Journal
- npj Digital Medicine
- Published
- 2025-12-06
- DOI
- https://doi.org/10.1038/s41746-025-02149-1
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Harbin Medical University
- Health Commission of Heilongjiang Province
- National Natural Science Foundation of China
- China Postdoctoral Science Foundation