Attention-Challenging Multiple Instance Learning for Whole Slide Image Classification
Not on the map. Why?
Abstract (arXiv)
In the application of Multiple Instance Learning (MIL) methods for Whole Slide Image (WSI) classification, attention mechanisms often focus on a subset of discriminative instances, which are closely linked to overfitting. To mitigate overfitting, we present Attention-Challenging MIL (ACMIL). ACMIL combines two techniques based on separate analyses for attention value concentration. Firstly, UMAP of instance features reveals various patterns among discriminative instances, with existing attention mechanisms capturing only some of them. To remedy this, we introduce Multiple Branch Attention (MBA) to capture more discriminative instances using multiple attention branches. Secondly, the examination of the cumulative value of Top-K attention scores indicates that a tiny number of instances dominate the majority of attention. In response, we present Stochastic Top-K Instance Masking (STKIM), which masks out a portion of instances with Top-K attention values and allocates their attention values to the remaining instances. The extensive experimental results on three WSI datasets with two pre-trained backbones reveal that our ACMIL outperforms state-of-the-art methods. Additionally, through heatmap visualization and UMAP visualization, this paper extensively illustrates ACMIL's effectiveness in suppressing attention value concentration and overcoming the overfitting challenge. The source code is available at urlhttps://github.com/dazhangyu123/ACMIL.
Key facts. Cited by 61 papers in the map (data through 14 Oct 2026). Builds on 38 of its 72 references that are in the map. First posted on arXiv 13 Nov 2023.
Builds on
38 of 72 references.
References · 38
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2018UMAP: Uniform Manifold Approximation and Projection for Dimension Reductioncited by 2,898
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2017Improved Regularization of Convolutional Neural Networks with Cutoutcited by 1,628
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,432
- 2017Random Erasing Data Augmentationcited by 1,239
- 2018Attention-based Deep Multiple Instance Learningcited by 1,095
- 2020Data Efficient and Weakly Supervised Computational Pathology on Whole Slide Imagescited by 745
- 2016Language Modeling with Gated Convolutional Networkscited by 702
- 2021TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classificationcited by 555
26 more
- 2020Dual-stream Multiple Instance Learning Network for Whole Slide Image Classification with Self-supervised Contrastive Learningcited by 478
- 2020Self-Challenging Improves Cross-Domain Generalizationcited by 426
- 2022Scaling Vision Transformers to Gigapixel Images via Hierarchical Self-Supervised Learningcited by 359
- 2022DTFD-MIL: Double-Tier Feature Distillation Multiple Instance Learning for Histopathology Whole Slide Image Classificationcited by 256
- 2020Whole Slide Images based Cancer Survival Prediction using Attention Guided Deep Multiple Instance Learning Networkscited by 178
- 2015Patch-based Convolutional Neural Network for Whole Slide Tissue Image Classificationcited by 159
- 2022Benchmarking Self-Supervised Learning on Diverse Pathology Datasetscited by 150
- 2021BRACS: A Dataset for BReAst Carcinoma Subtyping in H&E Histology Imagescited by 132
- 2016Deep Learning for Identifying Metastatic Breast Cancercited by 93
- 2023Interventional Bag Multi-Instance Learning On Whole-Slide Pathological Imagescited by 91
- 2018Neural Image Compression for Gigapixel Histopathology Image Analysiscited by 65
- 2023Multiple Instance Learning Framework with Masked Hard Instance Mining for Whole Slide Image Classificationcited by 60
- 2023Task-specific Fine-tuning via Variational Information Bottleneck for Weakly-supervised Pathology Whole Slide Image Classificationcited by 54
- 2023Histopathology Whole Slide Image Analysis with Heterogeneous Graph Representation Learningcited by 47
- 2020Self-Supervision Closes the Gap Between Weak and Strong Supervision in Histologycited by 43
- 2022Bi-directional Weakly Supervised Knowledge Distillation for Whole Slide Image Classificationcited by 43
- 2019Streaming convolutional neural networks for end-to-end learning with multi-megapixel imagescited by 41
- 2022ReMix: A General and Efficient Framework for Multiple Instance Learning based Whole Slide Image Classificationcited by 30
- 2023Diagnose Like a Pathologist: Transformer-Enabled Hierarchical Attention-Guided Multiple Instance Learning for Whole Slide Image Classificationcited by 27
- 2022Benchmarking the Robustness of Deep Neural Networks to Common Corruptions in Digital Pathologycited by 26
- 2020Predicting molecular phenotypes from histopathology images: a transcriptome-wide expression-morphology analysis in breast cancercited by 21
- 2018Cancer Metastasis Detection With Neural Conditional Random Fieldcited by 17
- 2021Efficient Classification of Very Large Images with Tiny Objectscited by 17
- 2022Iterative Patch Selection for High-Resolution Image Recognitioncited by 14
- 2023Iteratively Coupled Multiple Instance Learning from Instance to Bag Classifier for Whole Slide Image Classificationcited by 8
- 2023Dual-Query Multiple Instance Learning for Dynamic Meta-Embedding based Tumor Classificationcited by 1
Built on by
Citing papers per year: 2023: 2; 2024: 13; 2025: 25; 2026: 21.
61 papers.
Sort:
By area:
- 2025CPathAgent: An Agent-based Foundation Model for Interpretable High-Resolution Pathology Image Analysis Mimicking Pathologists' Diagnostic Logiccited by 29
- 2024PathGen-1.6M: 1.6 Million Pathology Image-text Pairs Generation through Multi-agent Collaborationcited by 28
- 2024HMIL: Hierarchical Multi-Instance Learning for Fine-Grained Whole Slide Image Classificationcited by 13
- 2023Shapley Values-enabled Progressive Pseudo Bag Augmentation for Whole Slide Image Classificationcited by 12
- 2024Rethinking Transformer for Long Contextual Histopathology Whole Slide Image Analysiscited by 12
- 2024AEM: Attention Entropy Maximization for Multiple Instance Learning based Whole Slide Image Classificationcited by 10
- 2024From Correlation to Causation: Max-Pooling-Based Multi-Instance Learning Leads to More Robust Whole Slide Image Classificationcited by 7
- 2025Revisiting End-to-End Learning with Slide-level Supervision in Computational Pathologycited by 6
- 2025Efficient Whole Slide Pathology VQA via Token Compressioncited by 6
- 2025LoC-Path: Learning to Compress for Pathology Multimodal Large Language Modelscited by 6
- 2026ASMIL: Attention-Stabilized Multiple Instance Learning for Whole Slide Imagingcited by 6
- 2026Act Like a Pathologist: Tissue-Aware Whole Slide Image Reasoningcited by 5
38 more
- 2024RankByGene: Gene-Guided Histopathology Representation Learning Through Cross-Modal Ranking Consistencycited by 4
- 2023Long-MIL: Scaling Long Contextual Multiple Instance Learning for Histopathology Whole Slide Image Analysiscited by 3
- 2025Can We Simplify Slide-level Fine-tuning of Pathology Foundation Models?cited by 3
- 2025PathVQ: Reforming Computational Pathology Foundation Model for Whole Slide Image Analysis via Vector Quantizationcited by 3
- 2025Cracking Instance Jigsaw Puzzles: An Alternative to Multiple Instance Learning for Whole Slide Image Analysiscited by 3
- 2024Biochemical Prostate Cancer Recurrence Prediction: Thinking Fast & Slowcited by 2
- 2025HyperPath: Knowledge-Guided Hyperbolic Semantic Hierarchy Modeling for WSI Analysiscited by 2
- 2025Sequential Attention-based Sampling for Histopathological Analysiscited by 2
- 2024Beyond accuracy: quantifying the reliability of Multiple Instance Learning for Whole Slide Image classificationcited by 1
- 2024Beyond Labels: A Self-Supervised Framework with Masked Autoencoders and Random Cropping for Breast Cancer Subtype Classificationcited by 1
- 2025MExD: An Expert-Infused Diffusion Model for Whole-Slide Image Classificationcited by 1
- 2025SMILE: a Scale-aware Multiple Instance Learning Method for Multicenter STAS Lung Cancer Histopathology Diagnosiscited by 1
- 2025Advancing Multiple Instance Learning with Continual Learning for Whole Slide Imagingcited by 1
- 2025SlideMamba: Entropy-Based Adaptive Fusion of GNN and Mamba for Enhanced Representation Learning in Digital Pathologycited by 1
- 2025MAPLE: Multi-scale Attribute-enhanced Prompt Learning for Few-shot Whole Slide Image Classificationcited by 1
- 2026A Deployment-Friendly Foundational Framework for Efficient Computational Pathologycited by 1
- 2026Exploiting Label-Independent Regularization from Spatial Dependencies for Whole Slide Image Analysiscited by 1
- 2026Are Compact Rationales Free? Measuring Tile Selection Headroom in Frozen WSI-MILcited by 1
- 2026QG-MIL: A Gated Transformer Aggregator for Domain-Agnostic Multiple Instance Learning in Medical Imagingcited by 1
- 2024MicroMIL: Graph-Based Multiple Instance Learning for Context-Aware Diagnosis with Microscopic Imagescited by 0
- 2024Efficient Whole Slide Image Classification through Fisher Vector Representationcited by 0
- 2024Multimodal Outer Arithmetic Block Dual Fusion of Whole Slide Images and Omics Data for Precision Oncologycited by 0
- 2024On the Generalizability of Iterative Patch Selection for Memory-Efficient High-Resolution Image Classificationcited by 0
- 2025LadderMIL: Multiple Instance Learning with Coarse-to-Fine Self-Distillationcited by 0
- 2025Denoising Mutual Knowledge Distillation in Bi-Directional Multiple Instance Learningcited by 0
- 2025DLiPath: A Benchmark for the Comprehensive Assessment of Donor Liver Based on Histopathological Image Datasetcited by 0
- 2025Reducing Variability of Multiple Instance Learning Methods for Digital Pathologycited by 0
- 2025Contrastive Cross-Bag Augmentation for Multiple Instance Learning-based Whole Slide Image Classificationcited by 0
- 2025Revisiting Data Challenges of Computational Pathology: A Pack-based Multiple Instance Learning Training Frameworkcited by 0
- 2025A Hierarchical Geometry-guided Transformer for Histological Subtyping of Primary Liver Cancercited by 0
- 2025Fourier Transform Multiple Instance Learning for Whole Slide Image Classificationcited by 0
- 2025Positive Semi-definite Latent Factor Grouping-Boosted Cluster-reasoning Instance Disentangled Learning for WSI Representationcited by 0
- 2025Cross-Stain Contrastive Learning for Paired Immunohistochemistry and Histopathology Slide Representation Learningcited by 0
- 2025CAPRMIL: Context-Aware Patch Representations for Multiple Instance Learningcited by 0
- 2026Prototype Instance-semantic Disentanglement with Low-rank Regularized Subspace Clustering for WSIs Explainable Recognitioncited by 0
- 2026MacNet: An End-to-End Manifold-Constrained Adaptive Clustering Network for Interpretable Whole Slide Image Classificationcited by 0
- 2026AINet: Anchor Instances Learning for Regional Heterogeneity in Whole Slide Imagecited by 0
- 2026Domain Adaptation Without the Compute Burden for Efficient Whole Slide Image Analysiscited by 0
50 most cited of 61 All 61 (JSON)
- 2026Weakly Supervised Spatial Grounding for Discriminative Attention-Based Ultrasound-Histopathology Alignment in Prostate Cancer Grading14 Sep 2026
- 2026CAR-MIL: Counterfactual Attention Regularization for Multiple Instance Learning8 Sep 2026
- 2026Semantic-Aware Subgraph State Space Model for WSI Classification in Histopathology3 Sep 2026
- 2026Learning Unified Video and Image Representation for Video Face Forgery Detection13 Aug 2026
- 2026A Multi-Cohort Validation of Censoring-Aware Conformal Lower Predictive Bounds for Pathology Survival Models23 Jul 2026
- 2026QG-MIL: A Gated Transformer Aggregator for Domain-Agnostic Multiple Instance Learning in Medical Imaging18 Jun 2026
- 2026Semantic-Anchored Evidential Fusion for Domain-Robust Whole-Slide Survival Analysis18 Jun 2026
- 2026In-Context Multiple Instance Learning4 Jun 2026
- 2026GCE-MIL: Faithful and Recoverable Evidence for Multiple Instance Learning in Whole-Slide Imaging17 May 2026
- 2026Spatial Blindness in Whole-Slide Multiple Instance Learning17 May 2026
- 2026Diffusion Attention Expert Model for Predicting and Semi-automatic Localizing STAS in Lung Cancer Histopathological Images15 May 2026
- 2026Are Compact Rationales Free? Measuring Tile Selection Headroom in Frozen WSI-MIL12 May 2026
38 more
- 2026Federated Distillation for Whole Slide Image via Gaussian-Mixture Feature Alignment and Curriculum Integration1 May 2026
- 2026Domain Adaptation Without the Compute Burden for Efficient Whole Slide Image Analysis16 Mar 2026
- 2026ASMIL: Attention-Stabilized Multiple Instance Learning for Whole Slide Imaging1 Mar 2026
- 2026Act Like a Pathologist: Tissue-Aware Whole Slide Image Reasoning28 Feb 2026
- 2026Exploiting Label-Independent Regularization from Spatial Dependencies for Whole Slide Image Analysis23 Feb 2026
- 2026AINet: Anchor Instances Learning for Regional Heterogeneity in Whole Slide Image21 Feb 2026
- 2026MacNet: An End-to-End Manifold-Constrained Adaptive Clustering Network for Interpretable Whole Slide Image Classification16 Feb 2026
- 2026Prototype Instance-semantic Disentanglement with Low-rank Regularized Subspace Clustering for WSIs Explainable Recognition16 Feb 2026
- 2026A Deployment-Friendly Foundational Framework for Efficient Computational Pathology15 Feb 2026
- 2025CAPRMIL: Context-Aware Patch Representations for Multiple Instance Learning16 Dec 2025
- 2025LoC-Path: Learning to Compress for Pathology Multimodal Large Language Models5 Dec 2025
- 2025Cross-Stain Contrastive Learning for Paired Immunohistochemistry and Histopathology Slide Representation Learning3 Dec 2025
- 2025Positive Semi-definite Latent Factor Grouping-Boosted Cluster-reasoning Instance Disentangled Learning for WSI Representation3 Nov 2025
- 2025Fourier Transform Multiple Instance Learning for Whole Slide Image Classification16 Oct 2025
- 2025A Hierarchical Geometry-guided Transformer for Histological Subtyping of Primary Liver Cancer7 Oct 2025
- 2025MAPLE: Multi-scale Attribute-enhanced Prompt Learning for Few-shot Whole Slide Image Classification30 Sep 2025
- 2025SlideMamba: Entropy-Based Adaptive Fusion of GNN and Mamba for Enhanced Representation Learning in Digital Pathology25 Sep 2025
- 2025Revisiting Data Challenges of Computational Pathology: A Pack-based Multiple Instance Learning Training Framework25 Sep 2025
- 2025Contrastive Cross-Bag Augmentation for Multiple Instance Learning-based Whole Slide Image Classification5 Aug 2025
- 2025Efficient Whole Slide Pathology VQA via Token Compression19 Jul 2025
- 2025Cracking Instance Jigsaw Puzzles: An Alternative to Multiple Instance Learning for Whole Slide Image Analysis10 Jul 2025
- 2025Sequential Attention-based Sampling for Histopathological Analysis7 Jul 2025
- 2025Reducing Variability of Multiple Instance Learning Methods for Digital Pathology30 Jun 2025
- 2025HyperPath: Knowledge-Guided Hyperbolic Semantic Hierarchy Modeling for WSI Analysis19 Jun 2025
- 2025Revisiting End-to-End Learning with Slide-level Supervision in Computational Pathology3 Jun 2025
- 2025DLiPath: A Benchmark for the Comprehensive Assessment of Donor Liver Based on Histopathological Image Dataset30 May 2025
- 2025CPathAgent: An Agent-based Foundation Model for Interpretable High-Resolution Pathology Image Analysis Mimicking Pathologists' Diagnostic Logic26 May 2025
- 2025Denoising Mutual Knowledge Distillation in Bi-Directional Multiple Instance Learning17 May 2025
- 2025Advancing Multiple Instance Learning with Continual Learning for Whole Slide Imaging15 May 2025
- 2025SMILE: a Scale-aware Multiple Instance Learning Method for Multicenter STAS Lung Cancer Histopathology Diagnosis18 Mar 2025
- 2025MExD: An Expert-Infused Diffusion Model for Whole-Slide Image Classification16 Mar 2025
- 2025PathVQ: Reforming Computational Pathology Foundation Model for Whole Slide Image Analysis via Vector Quantization9 Mar 2025
- 2025Can We Simplify Slide-level Fine-tuning of Pathology Foundation Models?28 Feb 2025
- 2025LadderMIL: Multiple Instance Learning with Coarse-to-Fine Self-Distillation4 Feb 2025
- 2024On the Generalizability of Iterative Patch Selection for Memory-Efficient High-Resolution Image Classification15 Dec 2024
- 2024Multimodal Outer Arithmetic Block Dual Fusion of Whole Slide Images and Omics Data for Precision Oncology26 Nov 2024
- 2024RankByGene: Gene-Guided Histopathology Representation Learning Through Cross-Modal Ranking Consistency22 Nov 2024
- 2024Efficient Whole Slide Image Classification through Fisher Vector Representation13 Nov 2024
50 newest of 61 All 61 (JSON)
- 2024PathGen-1.6M: 1.6 Million Pathology Image-text Pairs Generation through Multi-agent Collaborationcited by 28
- 2025Efficient Whole Slide Pathology VQA via Token Compressioncited by 6
- 2025LoC-Path: Learning to Compress for Pathology Multimodal Large Language Modelscited by 6
- 2025MAPLE: Multi-scale Attribute-enhanced Prompt Learning for Few-shot Whole Slide Image Classificationcited by 1
- 2025LoC-Path: Learning to Compress for Pathology Multimodal Large Language Models5 Dec 2025
- 2025MAPLE: Multi-scale Attribute-enhanced Prompt Learning for Few-shot Whole Slide Image Classification30 Sep 2025
- 2025Efficient Whole Slide Pathology VQA via Token Compression19 Jul 2025
- 2024PathGen-1.6M: 1.6 Million Pathology Image-text Pairs Generation through Multi-agent Collaboration28 Jun 2024
- 2025CPathAgent: An Agent-based Foundation Model for Interpretable High-Resolution Pathology Image Analysis Mimicking Pathologists' Diagnostic Logiccited by 29
- 2024HMIL: Hierarchical Multi-Instance Learning for Fine-Grained Whole Slide Image Classificationcited by 13
- 2023Shapley Values-enabled Progressive Pseudo Bag Augmentation for Whole Slide Image Classificationcited by 12
- 2024Rethinking Transformer for Long Contextual Histopathology Whole Slide Image Analysiscited by 12
- 2024AEM: Attention Entropy Maximization for Multiple Instance Learning based Whole Slide Image Classificationcited by 10
- 2024From Correlation to Causation: Max-Pooling-Based Multi-Instance Learning Leads to More Robust Whole Slide Image Classificationcited by 7
- 2025Revisiting End-to-End Learning with Slide-level Supervision in Computational Pathologycited by 6
- 2026ASMIL: Attention-Stabilized Multiple Instance Learning for Whole Slide Imagingcited by 6
- 2024RankByGene: Gene-Guided Histopathology Representation Learning Through Cross-Modal Ranking Consistencycited by 4
- 2023Long-MIL: Scaling Long Contextual Multiple Instance Learning for Histopathology Whole Slide Image Analysiscited by 3
- 2025Can We Simplify Slide-level Fine-tuning of Pathology Foundation Models?cited by 3
- 2025PathVQ: Reforming Computational Pathology Foundation Model for Whole Slide Image Analysis via Vector Quantizationcited by 3
12 most cited of 53 All 53 (JSON)
- 2026Weakly Supervised Spatial Grounding for Discriminative Attention-Based Ultrasound-Histopathology Alignment in Prostate Cancer Grading14 Sep 2026
- 2026CAR-MIL: Counterfactual Attention Regularization for Multiple Instance Learning8 Sep 2026
- 2026Semantic-Aware Subgraph State Space Model for WSI Classification in Histopathology3 Sep 2026
- 2026Learning Unified Video and Image Representation for Video Face Forgery Detection13 Aug 2026
- 2026A Multi-Cohort Validation of Censoring-Aware Conformal Lower Predictive Bounds for Pathology Survival Models23 Jul 2026
- 2026QG-MIL: A Gated Transformer Aggregator for Domain-Agnostic Multiple Instance Learning in Medical Imaging18 Jun 2026
- 2026Semantic-Anchored Evidential Fusion for Domain-Robust Whole-Slide Survival Analysis18 Jun 2026
- 2026In-Context Multiple Instance Learning4 Jun 2026
- 2026GCE-MIL: Faithful and Recoverable Evidence for Multiple Instance Learning in Whole-Slide Imaging17 May 2026
- 2026Spatial Blindness in Whole-Slide Multiple Instance Learning17 May 2026
- 2026Diffusion Attention Expert Model for Predicting and Semi-automatic Localizing STAS in Lung Cancer Histopathological Images15 May 2026
- 2026Are Compact Rationales Free? Measuring Tile Selection Headroom in Frozen WSI-MIL12 May 2026
12 newest of 53 All 53 (JSON)