Awesome Domain Generalization for Computational Pathology

November 7, 2024 · View on GitHub

Awesome

This repository contains lists of resources (including Datasets and Code Bases) that can help domain generalization research in computational pathology. These resources and their related concepts are further explained in the following manuscript:

@misc{jahanifar2023domain,
      title={Domain Generalization in Computational Pathology: Survey and Guidelines}, 
      author={Mostafa Jahanifar and Manahil Raza and Kesi Xu and Trinh Vuong and Rob Jewsbury and Adam Shephard and Neda Zamanitajeddin and Jin Tae Kwak and Shan E Ahmed Raza and Fayyaz Minhas and Nasir Rajpoot},
      year={2023},
      eprint={2310.19656},
      archivePrefix={arXiv},
      primaryClass={eess.IV}
}

Any contribution will be appreciated. To contribute to this awesome list or suggest new resources, please make a PR and add your suggestions.

:open_file_folder: Datasets

Publicly available datasets for DG experiments in CPath. Column DS represents the type domain shift that can be studied with each dataset (1: Covariate Shift, 2: Prior Shift, 3: Posterior Shift, and 4: Class-Conditional Shift).

DatasetApplication/TaskDSDomains
Detection
ATYPIA14 [paper][download]Mitosis detection in breast cancer12 scanners
Crowdsource [paper]Nuclei detection in renal cell carcinoma36 annotators
TUPAC-Aux [paper][download]Mitosis detection in breast cancer13 centers
DigestPath [paper][download]Signet ring cell detection in colon cancer14 centers
TiGER-Cells [paper][download]TILs detection in breast cancer13 sources
EndoNuke [paper][download]Nuclei Detection in Estrogen and Progesterone Stained IHC Endometrium Scans37 annotators
MIDOG [paper][download]Mitosis detection in multiple cancer types1, 2, 37 tumors, 2 species
Classification
TUPAC-Mitosis [paper][download]BC proliferation scoring based on mitosis score13 centers
Camelyon16 [paper][download]Lymph node WSI classification for BC metastasis12 centers
PatchCamelyon [paper][download]BC tumor classification based on Camelyon1612 centers
Camelyon17 [paper][download]BC metastasis detection and pN-stage estimation15 centers
LC25000 [paper][download]Lung and colon tumor classification42 organs
Kather 100K [paper][download]Colon cancer tissue phenotype classification13 centers
WILDS [paper][download]BC tumor classification based on Camelyon1715 centers
HunCRC [paper][download]Screening status of colon cancer or normal tissue1, 44 polyps, 2 sampling
PANDA [paper][download]ISUP and Gleason grading of prostate cancer1, 2, 32 centers
Regression
TUPAC-PAM50 [paper][download]BC proliferation scoring based on PAM5013 centers
LYSTO [paper][download]Lymphocyte assessment (counting) in IHC images13 cancers, 9 centers
CoNIC (Lizard) [paper][download]Cellular composition in colon cancer1, 36 sources
TiGER-TILs [paper][download]TIL score estimation in breast cancer13 sources
Segmentation
Crowdsource [paper]Nuclear segmentation in renal cell carcinoma36 annotators
Camelyon [paper][download]BC metastasis segmentation in lymph node WSIs12 and 5 centers
DS Bowl 2018 [paper][download]Nuclear instance segmentation1, 431 sets, 5 modalities
CPM [paper][download]Nuclear instance segmentation1, 44 cancers
BCSS [paper][download]Semantic tissue segmentation in BC (from TCGA)120 centers
AIDPATH [paper]Glomeruli segmentation in Kidney biopsies13 centers
PanNuke [paper][download]Nuclear instance segmentation and classification1, 2, 419 organs
MoNuSeg [paper][download]Nuclear instance segmentation in H&E images19 organs, 18 centers
CryoNuSeg [paper][download]Nuclear segmentation in cryosectioned H&E1, 310 organs, 3 annotations
MoNuSAC [paper][download]Nuclear instance segmentation and classification1, 237 centers, 4 organs
Lizard [paper][download]Nuclear instance segmentation and classification1, 36 sources
MetaHistoSeg [paper][download]Multiple segmentation tasks in various cancers15 sources/tasks
PANDA [paper][download]Tissue segmentation in prostate cancer1, 22 centers
TiGER-BCSS [paper][download]Tissue segmentation in BC (BCSS extension)13 sources
DigestPath [paper][download]Colon tissue segmentation14 centers
NuInsSeg [paper][download]Nuclear instance segmentation pan-cancer/species1,431 organs, 2 species
Survival and gene expression prediction
TCGA [papers][download]Pan-cancer survival and gene expression prediction1, 2, 433 cancers, 20 centers
CPTAC [papers][download][tool]Pan-cancer survival and gene expression prediction1, 210 cancers, 11 centers

:computer: Code bases

ReferenceDG MethodTitle
Pretraining
Yang et al. [paper][code]Minimizing Contrastive LossCS-CO: A Hybrid Self-Supervised Visual Representation Learning Method for H&E-stained Histopathological Images
Li et al. [paper][code]Minimizing Contrastive LossLesion-Aware Contrastive Representation Learning For Histopathology Whole Slide Images Analysis
Galdran et al. [paper][code]Unsupervised/Self-supervised learningTest Time Transform Prediction for Open Set Histopathological Image Recognition
Bozorgtabar et al. [paper][code]Unsupervised/Self-supervised learningSOoD: Self-Supervised Out-of-Distribution Detection Under Domain Shift for Multi-Class Colorectal Cancer Tissue Types
Koohbanani et al. [paper][code]Multiple Pretext TasksSelf Path: Self Supervision for Classification of Histology Images with Limited Budget of Annotation
Abbet et al. [paper][code]Unsupervised/Self-supervised learningSelf-rule to multi-adapt: Generalized multi-source feature learning using unsupervised domain adaptation for colorectal cancer tissue detection
Cho et al. [paper][code]Unsupervised/Self-supervised learningCell Detection in Domain Shift Problem Using Pseudo-Cell-Position Heatmap
Chikontwe et al. [paper][code]Unsupervised/Self-supervised learningWeakly supervised segmentation on neural compressed histopathology with self-equivariant regularization
Tran et al. [paper][code]Minimizing Contrastive LossS5CL: Unifying Fully-Supervised, Self-Supervised, and Semi-Supervised Learning Through Hierarchical Contrastive Learning
Sikaroudi et al. [paper][code]Unsupervised/Self-supervised learningSupervision and Source Domain Impact on Representation Learning: A Histopathology Case Study
Wang et al. [paper][code]Unsupervised/Self-supervised learningTransformer-based unsupervised contrastive learning for histopathological image classification
Kang et al. [paper][code]Unsupervised/Self-supervised learningBenchmarking Self-Supervised Learning on Diverse Pathology Datasets
Lazard et al. [paper][code]Contrastive LearningGiga-SSL: Self-Supervised Learning for Gigapixel Images
Vuong et al. [paper][code]Contrastive LearningIMPaSh: A Novel Domain-Shift Resistant Representation for Colorectal Cancer Tissue Classification
Chen et al. [paper][code]Unsupervised/Self-supervised learningFast and scalable search of whole-slide images via self-supervised deep learning
Meta-Learning
Sikaroudi et al. [paper][code]Meta-learningMagnification Generalization For Histopathology Image Embedding
Yuan et al. [paper][code]Meta-learningMetaHistoSeg: A Python Framework for Meta Learning in Histopathology Image Segmentation
Domain Alignment
Sharma et al. [paper][code]Mutual InformationMaNi: Maximizing Mutual Information for Nuclei Cross-Domain Unsupervised Segmentation
Boyd et al. [paper][code]Generative ModelsRegion-guided CycleGANs for Stain Transfer in Whole Slide Images
Kather et al. [paper][code]Stain NormalizationDeep learning can predict microsatellite instability directly from histology in gastrointestinal cancer
Zheng et al. [paper][code]Stain NormalizationAdaptive color deconvolution for histological WSI normalization
Sebai et al. [paper][code]Stain NormalizationMaskMitosis: a deep learning framework for fully supervised, weakly supervised, and unsupervised mitosis detection in histopathology images
Zhang et al. [paper][code]Minimizing Contrastive LossStain Based Contrastive Co-training for Histopathological Image Analysis
Shahban et al. [paper][code]Generative ModelsStaingan: Stain Style Transfer for Digital Histological Images
Wagner et al. [paper][code]Generative ModelsFederated Stain Normalization for Computational Pathology
Quiros et al. [paper][code]Domain Adversarial LearningAdversarial learning of cancer tissue representations
Salehi et al. [paper][code]Minimizing the KL DivergenceUnsupervised Cross-Domain Feature Extraction for Single Blood Cell Image Classification
Wilm et al. [paper][code]Domain-Adversarial LearningDomain adversarial retinanet as a reference algorithm for the mitosis domain generalization (midog) challenge
Haan et al. [paper][code]Generative modelsDeep learning-based transformation of H&E stained tissues into special stains
Dawood et al. [paper][code]Stain NormalizationDo Tissue Source Sites leave identifiable Signatures in Whole Slide Images beyond staining?
Data Augmentation
Pohjonen et al. [paper][code]Data augmentationAugment like there’s no tomorrow: Consistently performing neural networks for medical imaging
Chang et al. [paper][code]Stain AugmentationStain Mix-up: Unsupervised Domain Generalization for Histopathology Images
Shen et al. [paper][code]Stain AugmentationRandStainNA: Learning Stain-Agnostic Features from Histology Slides by Bridging Stain Augmentation and Normalization
Koohbanani et al. [paper][code]Data augmentationNuClick: A deep learning framework for interactive segmentation of microscopic images
Wang et al. [paper][code]Data augmentationA generalizable and robust deep learning algorithm for mitosis detection in multicenter breast histopathological images
Lin et al. [paper][code]Generative ModelsInsMix: Towards Realistic Generative Data Augmentation for Nuclei Instance Segmentation
Zhang et al. [paper][code]Data augmentationBenchmarking the Robustness of Deep Neural Networks to Common Corruptions in Digital Pathology
Yamashita et al. [paper][code]Style Transfer ModelsLearning domain-agnostic visual representation for computational pathology using medically-irrelevant style transfer augmentation
Falahkheirkhah et al. [paper][code]Generative ModelsDeepfake Histologic Images for Enhancing Digital Pathology
Scalbert et al. [paper][code]Generative ModelsTest-time image-to-image translation ensembling improves out-of-distribution generalization in histopathology
Mahmood et al. [paper][code]Generative ModelsDeep Adversarial Training for Multi-Organ Nuclei Segmentation in Histopathology Images
Fan et al. [paper][code]Generative ModelsFast FF-to-FFPE Whole Slide Image Translation via Laplacian Pyramid and Contrastive Learning
Marini et al. [paper][code]Stain AugmentationData-driven color augmentation for H&E stained images in computational pathology
Faryna et al. [paper][code]RandAugment for HistologyTailoring automated data augmentation to H&E-stained histopathology
Model Design
Graham et al. [paper][code]Model designDense Steerable Filter CNNs for Exploiting Rotational Symmetry in Histology Images
Lafarge et al. [paper][code]Model designRoto-translation equivariant convolutional networks: Application to histopathology image analysis
Zhang et al. [paper][code]Model designDDTNet: A dense dual-task network for tumor-infiltrating lymphocyte detection and segmentation in histopathological images of breast cancer
Graham et al. [paper][code]Model DesignOne model is all you need: Multi-task learning enables simultaneous histology image segmentation and classification
Yu et al. [paper][code]Model DesignPrototypical multiple instance learning for predicting lymph node metastasis of breast cancer from whole-slide pathological images
Yaar et al. [paper][code]Model DesignCross-Domain Knowledge Transfer for Prediction of Chemosensitivity in Ovarian Cancer Patients
Tang et al. [paper][code]Model DesignProbeable DARTS with Application to Computational Pathology
Vuong et al. [paper][code]Model DesignJoint categorical and ordinal learning for cancer grading in pathology images
Domain Separation
Wagner et al. [paper][code]Generative ModelsHistAuGAN: Structure-Preserving Multi-Domain Stain Color Augmentation using Style-Transfer with Disentangled Representations
Chikontwe et al. [paper][code]Learning disentangled representationsFeature Re-calibration based Multiple Instance Learning for Whole Slide Image Classification
Ensemble Learning
Sohail et al. [paper][code]Ensemble learningMitotic nuclei analysis in breast cancer histopathology images using deep ensemble classifier
Regularization Strategies
Mehrtens et al. [paper][code]Regularization StrategiesBenchmarking common uncertainty estimation methods with histopathological images under domain shift and label noise
Other
Lu et al. [paper][code]OtherFederated learning for computational pathology on gigapixel whole slide images
Aubreville et al. [paper][code]OtherQuantifying the Scanner-Induced Domain Gap in Mitosis Detection
Sadafi et al. [paper][code]OtherA Continual Learning Approach for Cross-Domain White Blood Cell Classification