Stain Consistency Learning: Handling Stain Variation for Automatic
Digital Pathology Segmentation
- URL: http://arxiv.org/abs/2311.06552v1
- Date: Sat, 11 Nov 2023 12:00:44 GMT
- Title: Stain Consistency Learning: Handling Stain Variation for Automatic
Digital Pathology Segmentation
- Authors: Michael Yeung, Todd Watts, Sean YW Tan, Pedro F. Ferreira, Andrew D.
Scott, Sonia Nielles-Vallespin, Guang Yang
- Abstract summary: We propose a novel framework combining stain-specific augmentation with a stain consistency loss function to learn stain colour invariant features.
We compare ten methods on Masson's trichrome and H&E stained cell and nuclei datasets, respectively.
We observed that stain normalisation methods resulted in equivalent or worse performance, while stain augmentation or stain adversarial methods demonstrated improved performance.
- Score: 3.2386272343130127
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Stain variation is a unique challenge associated with automated analysis of
digital pathology. Numerous methods have been developed to improve the
robustness of machine learning methods to stain variation, but comparative
studies have demonstrated limited benefits to performance. Moreover, methods to
handle stain variation were largely developed for H&E stained data, with
evaluation generally limited to classification tasks. Here we propose Stain
Consistency Learning, a novel framework combining stain-specific augmentation
with a stain consistency loss function to learn stain colour invariant
features. We perform the first, extensive comparison of methods to handle stain
variation for segmentation tasks, comparing ten methods on Masson's trichrome
and H&E stained cell and nuclei datasets, respectively. We observed that stain
normalisation methods resulted in equivalent or worse performance, while stain
augmentation or stain adversarial methods demonstrated improved performance,
with the best performance consistently achieved by our proposed approach. The
code is available at: https://github.com/mlyg/stain_consistency_learning
Related papers
- Unsupervised Latent Stain Adaptation for Computational Pathology [2.483372684394528]
Stain adaptation aims to reduce the generalization error between different stains by training a model on source stains that generalizes to target stains.
We propose a joint training between artificially labeled and unlabeled data including all available stained images called Unsupervised Latent Stain Adaptation (ULSA)
Our method uses stain translation to enrich labeled source images with synthetic target images in order to increase the supervised signals.
arXiv Detail & Related papers (2024-06-27T11:08:42Z) - Which Tokens to Use? Investigating Token Reduction in Vision
Transformers [64.99704164972513]
We study the reduction patterns of 10 different token reduction methods using four image classification datasets.
We find that the Top-K pruning method is a surprisingly strong baseline.
The similarity of reduction patterns is a moderate-to-strong proxy for model performance.
arXiv Detail & Related papers (2023-08-09T01:51:07Z) - A Laplacian Pyramid Based Generative H&E Stain Augmentation Network [5.841841666625825]
Generative Stain Augmentation Network (G-SAN) is a GAN-based framework that augments a collection of cell images with simulated stain variations.
Using G-SAN-augmented training data provides on average 15.7% improvement in F1 score and 7.3% improvement in panoptic quality.
arXiv Detail & Related papers (2023-05-23T17:43:18Z) - Improving Stain Invariance of CNNs for Segmentation by Fusing Channel
Attention and Domain-Adversarial Training [5.501810688265425]
Variability in staining protocols, such as different slide preparation techniques, chemicals, and scanner configurations, can result in a diverse set of whole slide images (WSIs)
This distribution shift can negatively impact the performance of deep learning models on unseen samples.
We propose a method for improving the generalizability of convolutional neural networks (CNNs) to stain changes in a single-source setting for semantic segmentation.
arXiv Detail & Related papers (2023-04-22T16:54:37Z) - Rethinking Semi-Supervised Medical Image Segmentation: A
Variance-Reduction Perspective [51.70661197256033]
We propose ARCO, a semi-supervised contrastive learning framework with stratified group theory for medical image segmentation.
We first propose building ARCO through the concept of variance-reduced estimation and show that certain variance-reduction techniques are particularly beneficial in pixel/voxel-level segmentation tasks.
We experimentally validate our approaches on eight benchmarks, i.e., five 2D/3D medical and three semantic segmentation datasets, with different label settings.
arXiv Detail & Related papers (2023-02-03T13:50:25Z) - Stain-invariant self supervised learning for histopathology image
analysis [74.98663573628743]
We present a self-supervised algorithm for several classification tasks within hematoxylin and eosin stained images of breast cancer.
Our method achieves the state-of-the-art performance on several publicly available breast cancer datasets.
arXiv Detail & Related papers (2022-11-14T18:16:36Z) - Stain-Adaptive Self-Supervised Learning for Histopathology Image
Analysis [3.8073142980733]
We propose a novel Stain-Adaptive Self-Supervised Learning(SASSL) method for histopathology image analysis.
Our SASSL integrates a domain-adversarial training module into the SSL framework to learn distinctive features that are robust to both various transformations and stain variations.
Experimental results demonstrate that the proposed method can robustly improve the feature extraction ability of the model.
arXiv Detail & Related papers (2022-08-08T09:54:46Z) - RandStainNA: Learning Stain-Agnostic Features from Histology Slides by
Bridging Stain Augmentation and Normalization [45.81689497433507]
Two proposals, namely stain normalization (SN) and stain augmentation (SA), have been spotlighted to reduce the generalization error.
To address the problems, we unify SN and SA with a novel RandStainNA scheme.
The RandStainNA constrains variable stain styles in a practicable range to train a stain agnostic deep learning model.
arXiv Detail & Related papers (2022-06-25T16:43:59Z) - Revisiting Consistency Regularization for Semi-Supervised Learning [80.28461584135967]
We propose an improved consistency regularization framework by a simple yet effective technique, FeatDistLoss.
Experimental results show that our model defines a new state of the art for various datasets and settings.
arXiv Detail & Related papers (2021-12-10T20:46:13Z) - Cross-Site Severity Assessment of COVID-19 from CT Images via Domain
Adaptation [64.59521853145368]
Early and accurate severity assessment of Coronavirus disease 2019 (COVID-19) based on computed tomography (CT) images offers a great help to the estimation of intensive care unit event.
To augment the labeled data and improve the generalization ability of the classification model, it is necessary to aggregate data from multiple sites.
This task faces several challenges including class imbalance between mild and severe infections, domain distribution discrepancy between sites, and presence of heterogeneous features.
arXiv Detail & Related papers (2021-09-08T07:56:51Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.