C$^2$MIL: Synchronizing Semantic and Topological Causalities in Multiple Instance Learning for Robust and Interpretable Survival Analysis

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cen, Min, Zhuang, Zhenfeng, Zhang, Yuzhe, Zeng, Min, Magnier, Baptiste, Yu, Lequan, Zhang, Hong, Wang, Liansheng
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915511519936512
author Cen, Min
Zhuang, Zhenfeng
Zhang, Yuzhe
Zeng, Min
Magnier, Baptiste
Yu, Lequan
Zhang, Hong
Wang, Liansheng
author_facet Cen, Min
Zhuang, Zhenfeng
Zhang, Yuzhe
Zeng, Min
Magnier, Baptiste
Yu, Lequan
Zhang, Hong
Wang, Liansheng
contents Graph-based Multiple Instance Learning (MIL) is widely used in survival analysis with Hematoxylin and Eosin (H\&E)-stained whole slide images (WSIs) due to its ability to capture topological information. However, variations in staining and scanning can introduce semantic bias, while topological subgraphs that are not relevant to the causal relationships can create noise, resulting in biased slide-level representations. These issues can hinder both the interpretability and generalization of the analysis. To tackle this, we introduce a dual structural causal model as the theoretical foundation and propose a novel and interpretable dual causal graph-based MIL model, C$^2$MIL. C$^2$MIL incorporates a novel cross-scale adaptive feature disentangling module for semantic causal intervention and a new Bernoulli differentiable causal subgraph sampling method for topological causal discovery. A joint optimization strategy combining disentangling supervision and contrastive learning enables simultaneous refinement of both semantic and topological causalities. Experiments demonstrate that C$^2$MIL consistently improves generalization and interpretability over existing methods and can serve as a causal enhancement for diverse MIL baselines. The code is available at https://github.com/mimic0127/C2MIL.
format Preprint
id arxiv_https___arxiv_org_abs_2509_20152
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle C$^2$MIL: Synchronizing Semantic and Topological Causalities in Multiple Instance Learning for Robust and Interpretable Survival Analysis
Cen, Min
Zhuang, Zhenfeng
Zhang, Yuzhe
Zeng, Min
Magnier, Baptiste
Yu, Lequan
Zhang, Hong
Wang, Liansheng
Computer Vision and Pattern Recognition
Graph-based Multiple Instance Learning (MIL) is widely used in survival analysis with Hematoxylin and Eosin (H\&E)-stained whole slide images (WSIs) due to its ability to capture topological information. However, variations in staining and scanning can introduce semantic bias, while topological subgraphs that are not relevant to the causal relationships can create noise, resulting in biased slide-level representations. These issues can hinder both the interpretability and generalization of the analysis. To tackle this, we introduce a dual structural causal model as the theoretical foundation and propose a novel and interpretable dual causal graph-based MIL model, C$^2$MIL. C$^2$MIL incorporates a novel cross-scale adaptive feature disentangling module for semantic causal intervention and a new Bernoulli differentiable causal subgraph sampling method for topological causal discovery. A joint optimization strategy combining disentangling supervision and contrastive learning enables simultaneous refinement of both semantic and topological causalities. Experiments demonstrate that C$^2$MIL consistently improves generalization and interpretability over existing methods and can serve as a causal enhancement for diverse MIL baselines. The code is available at https://github.com/mimic0127/C2MIL.
title C$^2$MIL: Synchronizing Semantic and Topological Causalities in Multiple Instance Learning for Robust and Interpretable Survival Analysis
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2509.20152