Multistain Pretraining for Slide Representation Learning in Pathology
About
Developing self-supervised learning (SSL) models that can learn universal and transferable representations of H&E gigapixel whole-slide images (WSIs) is becoming increasingly valuable in computational pathology. These models hold the potential to advance critical tasks such as few-shot classification, slide retrieval, and patient stratification. Existing approaches for slide representation learning extend the principles of SSL from small images (e.g., 224 x 224 patches) to entire slides, usually by aligning two different augmentations (or views) of the slide. Yet the resulting representation remains constrained by the limited clinical and biological diversity of the views. Instead, we postulate that slides stained with multiple markers, such as immunohistochemistry, can be used as different views to form a rich task-agnostic training signal. To this end, we introduce Madeleine, a multimodal pretraining strategy for slide representation learning. Madeleine is trained with a dual global-local cross-stain alignment objective on large cohorts of breast cancer samples (N=4,211 WSIs across five stains) and kidney transplant samples (N=12,070 WSIs across four stains). We demonstrate the quality of slide representations learned by Madeleine on various downstream evaluations, ranging from morphological and molecular classification to prognostic prediction, comprising 21 tasks using 7,299 WSIs from multiple medical centers. Code is available at https://github.com/mahmoodlab/MADELEINE.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Few-shot classification | CPTAC | LUNG ST Accuracy90 | 45 | |
| Few-shot classification | CPTAC k=25 positive samples (test) | LUNG ST Accuracy93.4 | 45 | |
| Classification | CPTAC k=5 positive samples | Lung ST Performance78.1 | 30 | |
| Computational Pathology Classification | CPTAC k=10 (test) | LUNG ST80.6 | 30 | |
| Subtyping | CPTAC NSCLC | AUPRC94.6 | 23 | |
| Multi-cancer Classification | CPTAC Aggregate | Avg F1 Score28.2 | 23 | |
| Classification | CPTAC | NSCLC ST Accuracy95.4 | 19 | |
| Classification | CPTAC 9x magnification (test) | ST (NSCLC) Accuracy95.6 | 19 | |
| Few-shot tissue subtyping and mutation prediction | CPTAC k=10 positive samples (test) | LUNG-ST80.2 | 15 | |
| Classification | CPTAC 1.0 (k=25) | LUNG ST85.1 | 15 |