MeCaMIL: Causality-Aware Multiple Instance Learning for Fair and Interpretable Whole Slide Image Diagnosis

Read full text
Authors: Yiran Song, Yikai Zhang, Shuang Zhou, Guojun Xiong, Xiaofeng Yang, Nian Wang, Fenglong Ma, Rui Zhang, Mingquan Lin
Cited by: 0
References: 52

Abstract

Multiple instance learning (MIL) has emerged as the dominant paradigm for whole slide image (WSI) analysis in computational pathology, achieving strong diagnostic performance through patch-level feature aggregation. However, existing MIL methods face critical limitations: (1) they rely on attention mechanisms that lack causal interpretability-the ability to explain why predictions vary across demographic subgroups through explicit cause-effect pathways, and (2) they fail to integrate patient demographics (age, gender, race), leading to fairness concerns across diverse populations. These shortcomings hinder clinical translation, where algorithmic bias can exacerbate health disparities. We introduce MeCaMIL, a causality-aware MIL framework that explicitly models demographic confounders through structured causal graphs. Unlike prior approaches treating demographics as auxiliary features, MeCaMIL employs principled causal inference with collider structures to disentangle disease-relevant signals from spurious demographic correlations. Extensive evaluation on three benchmarks demonstrates state-of-the-art performance across CAMELYON16 (ACC/ AUC/F1: 0.939/0.983/0.946), TCGA-Lung (0.935/0.979/0.931), and TCGA-Multi (0.977/0.993/0.970, five cancer types). Critically, MeCaMIL achieves superior fairness-demographic disparity variance drops by over 65% relative reduction on average across attributes, with notable improvements for underserved populations. The framework generalizes to survival prediction (mean C-index: 0.653, +0.017 over best baseline across five cancer types). Ablation studies confirm that the causal graph structure is essential: alternative designs yield 0.048 lower accuracy and $4.2\times $ worse fairness. These results establish MeCaMIL as a principled framework for fair, causally interpretable, and clinically actionable AI in digital pathology, where interpretability derives from explicit structural causal modeling rather than post-hoc attention visualization. We release our complete implementation at https://github.com/zongzi13545329/MeCaMIL.git.

Fields of Study & Subjects

AI in cancer detection Digital Imaging for Blood Diseases Colorectal Cancer Screening and Detection

Publication details

Journal / source
IEEE Transactions on Medical Imaging
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Volume
45
Issue
9
Pages
4861–4875
Publication date
13 July 2026
Language
EN