Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

MambaMIL+: Modeling Long-Term Contextual Patterns for Gigapixel Whole Slide Image

About

Whole-slide images (WSIs) are an important data modality in computational pathology, yet their gigapixel resolution and lack of fine-grained annotations challenge conventional deep learning models. Multiple instance learning (MIL) offers a solution by treating each WSI as a bag of patch-level instances, but effectively modeling ultra-long sequences with rich spatial context remains difficult. Recently, Mamba has emerged as a promising alternative for long sequence learning, scaling linearly to thousands of tokens. However, despite its efficiency, it still suffers from limited spatial context modeling and memory decay, constraining its effectiveness to WSI analysis. To address these limitations, we propose MambaMIL+, a new MIL framework that explicitly integrates spatial context while maintaining long-range dependency modeling without memory forgetting. Specifically, MambaMIL+ introduces 1) overlapping scanning, which restructures the patch sequence to embed spatial continuity and instance correlations; 2) a selective stripe position encoder (S2PE) that encodes positional information while mitigating the biases of fixed scanning orders; and 3) a contextual token selection (CTS) mechanism, which leverages supervisory knowledge to dynamically enlarge the contextual memory for stable long-range modeling. Extensive experiments on 20 benchmarks across diagnostic classification, molecular prediction, and survival analysis demonstrate that MambaMIL+ consistently achieves state-of-the-art performance under three feature extractors (ResNet-50, PLIP, and CONCH), highlighting its effectiveness and robustness for large-scale computational pathology

Qian Zeng, Yihui Wang, Shu Yang, Yingxue Xu, Fengtao Zhou, Jiabo Ma, Dejia Cai, Zhengyu Zhang, Lijuan Qu, Yu Wang, Li Liang, Hao Chen• 2025

Related benchmarks

TaskDatasetResultRank
ClassificationCRC-KRAS TCGA cohort
AUC72.6
84
WSI ClassificationBCNB-HER2
Accuracy65.2
54
Survival PredictionBLCA
C-Index0.632
46
ClassificationCRC-Molecular TCGA cohort
AUC83.6
42
ClassificationFuzhou-CRC-KRAS
AUC (%)77.9
42
Diagnostic ClassificationBRACS-7
AUC0.879
42
Molecular predictionBRCA-Molecular
AUC78.7
42
Molecular predictionBRCA-TP53
AUC (%)84.6
42
Molecular predictionBCNB-ER
AUC89.8
42
Molecular predictionBCNB-PR
AUC83.2
42
Showing 10 of 19 rows

Other info

Follow for update