MeCaMIL: Causality-Aware Multiple Instance Learning for Fair and Interpretable Whole Slide Image Diagnosis.
Journal:
IEEE transactions on medical imaging
Published Date:
Sep 1, 2026
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.
Authors
Keywords
No keywords available for this article.