Federated Stain Normalization for Computational Pathology
arXiv:2209.14849 · doi:10.1007/978-3-031-16434-7_2
Abstract
Although deep federated learning has received much attention in recent years, progress has been made mainly in the context of natural images and barely for computational pathology. However, deep federated learning is an opportunity to create datasets that reflect the data diversity of many laboratories. Further, the effort of dataset construction can be divided among many. Unfortunately, existing algorithms cannot be easily applied to computational pathology since previous work presupposes that data distributions of laboratories must be similar. This is an unlikely assumption, mainly since different laboratories have different staining styles. As a solution, we propose BottleGAN, a generative model that can computationally align the staining styles of many laboratories and can be trained in a privacy-preserving manner to foster federated learning in computational pathology. We construct a heterogenic multi-institutional dataset based on the PESO segmentation dataset and improve the IOU by 42\% compared to existing federated learning algorithms. An implementation of BottleGAN is available at https://github.com/MECLabTUDA/BottleGAN
arXiv commentAccepted for Poster at MICCAI2022
References in corpus (8)
- Distilling the Knowledge in a Neural Network
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence
- SCAFFOLD: Stochastic Controlled Averaging for Federated Learning
- Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classification
- Ensemble Distillation for Robust Model Fusion in Federated Learning
- HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clients
- Epithelium segmentation using deep learning in H&E-stained prostate specimens with immunohistochemistry as reference standard
- Cronus: Robust and Heterogeneous Collaborative Learning with Black-Box Knowledge Transfer