Abstract:Multiple instance learning (MIL) stands as a powerful approach in weakly supervised learning, regularly employed in histological whole slide image (WSI) classification for detecting tumorous lesions. However, existing mainstream MIL methods focus on modeling correlation between instances while overlooking the inherent diversity among instances. However, few MIL methods have aimed at diversity modeling, which empirically show inferior performance but with a high computational cost. To bridge this gap, we propose a novel MIL aggregation method based on diverse global representation (DGR-MIL), by modeling diversity among instances through a set of global vectors that serve as a summary of all instances. First, we turn the instance correlation into the similarity between instance embeddings and the predefined global vectors through a cross-attention mechanism. This stems from the fact that similar instance embeddings typically would result in a higher correlation with a certain global vector. Second, we propose two mechanisms to enforce the diversity among the global vectors to be more descriptive of the entire bag: (i) positive instance alignment and (ii) a novel, efficient, and theoretically guaranteed diversification learning paradigm. Specifically, the positive instance alignment module encourages the global vectors to align with the center of positive instances (e.g., instances containing tumors in WSI). To further diversify the global representations, we propose a novel diversification learning paradigm leveraging the determinantal point process. The proposed model outperforms the state-of-the-art MIL aggregation models by a substantial margin on the CAMELYON-16 and the TCGA-lung cancer datasets. The code is available at \url{<a class="link-external link-https" href="https://github.com/ChongQingNoSubway/DGR-MIL" rel="external noopener nofollow">this https URL</a>}.

DualGCN-MIL: Whole Slide Image Classification Based on Double Relationship Graph Learning

Clustering-Based Multi-instance Learning Network for Whole Slide Image Classification

GCN-based MIL: Multi-Instance Learning Utilizing Structural Relationships among Instances

TDT-MIL: a framework with a dual-channel spatial positional encoder for weakly-supervised whole slide image classification

Dual-Attention Multiple Instance Learning Framework for Pathology Whole-Slide Image Classification

CaMIL: Causal Multiple Instance Learning for Whole Slide Image Classification

SlideGCD: Slide-based Graph Collaborative Training with Knowledge Distillation for Whole Slide Image Classification

A Regional Multiple Instance Learning Network for Whole Slide Image Segmentation

MamMIL: Multiple Instance Learning for Whole Slide Images with State Space Models

Multi-scale representation attention based deep multiple instance learning for gigapixel whole slide image analysis

DGMIL: Distribution Guided Multiple Instance Learning for Whole Slide Image Classification

Local-to-Global Spatial Learning for Whole-Slide Image Representation and Classification

DGR-MIL: Exploring Diverse Global Representation in Multiple Instance Learning for Whole Slide Image Classification

Learning Hybrid Negative Probability Model for Weakly-Supervised Whole Slide Image Recognition.

Iteratively Coupled Multiple Instance Learning from Instance to Bag Classifier for Whole Slide Image Classification

Multi-scale Multi-Instance Contrastive Learning for Whole Slide Image Classification

TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classification

Deep Hierarchical Multiple Instance Learning for Whole Slide Image Classification

Identifying Key Factors in Multi-Instance Learning for Whole-Slide Pathological Images Designing Assumption-Guided Feature Extractor

DTFD-MIL: Double-Tier Feature Distillation Multiple Instance Learning for Histopathology Whole Slide Image Classification

Multiple Instance Learning with Critical Instance for Whole Slide Image Classification.