atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

SI-MIL: Taming Deep MIL for Self-Interpretability in Gigapixel Histopathology

CVPR 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Prototype and concept dictionary models · 15 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SI-MIL, a dual-branch MIL that co-trains a deep ABMIL branch with a linear, self-interpretable branch over handcrafted nuclei-based PathExpert features to give feature-level explanations for WSI predictions.

Problem

Whole slide image classification with MIL models is typically black-box; existing interpretability only highlights salient patches and post-hoc feature analyses disconnect from the deep model's computations, so pathologists get little feature-level rationale.

Approach

Co-learn a deep additive ABMIL branch and a linear predictor on handcrafted PathExpert nuclei features, linked by a differentiable perturbed Top-K patch selection guided by MIL attention and a knowledge distillation loss, so the final prediction is a linear function of interpretable features.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add differentiable PAG Top-K patch selection (ablation)SI-MIL with Additive ABMIL · TCGA-Lungaccuracy: 0.859 -> 0.884Helped
Add differentiable PAG Top-K patch selection (ablation)SI-MIL with Additive ABMIL · TCGA-BRCAaccuracy: 0.915 -> 0.944Helped
Add differentiable PAG Top-K patch selection (ablation)SI-MIL with Additive ABMIL · TCGA-CRCaccuracy: 0.876 -> 0.884Helped
Add knowledge distillation (ablation)SI-MIL with Additive ABMIL · TCGA-Lungaccuracy: 0.853 -> 0.884Helped
Add knowledge distillation (ablation)SI-MIL with Additive ABMIL · TCGA-BRCAaccuracy: 0.932 -> 0.944Helped
Add knowledge distillation (ablation)SI-MIL with Additive ABMIL · TCGA-CRCaccuracy: 0.878 -> 0.884Helped
Add PAG Top-K patch selection and knowledge distillation (ablation)SI-MIL with Additive ABMIL · TCGA-Lungaccuracy: 0.857 -> 0.884Helped
Add PAG Top-K patch selection and knowledge distillation (ablation)SI-MIL with Additive ABMIL · TCGA-BRCAaccuracy: 0.915 -> 0.944Helped
Add PAG Top-K patch selection and knowledge distillation (ablation)SI-MIL with Additive ABMIL · TCGA-CRCaccuracy: 0.879 -> 0.884Helped

Surprising findings

Limitations

Experimental setup

Datasets: TCGA-BRCA TCGA-Lung TCGA-CRC PanNuke

Benchmarks: TCGA-BRCA TCGA-Lung TCGA-CRC

Metrics: accuracy AUC Silhouette score JS divergence user-friendliness faithfulness

Baselines: Additive ABMIL with IN ViT-S RetCCL CTransPath DINO ViT-S PathFeat 2-stage training CLAM TransMIL

Builds on: ABMIL CLAM TransMIL DINO HoVer-Net perturbed top-k

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy