Abstract:Training a deep learning model to classify histopathological images is challenging, because of the color and shape variability of the cells and tissues, and the reduced amount of available data, which does not allow proper learning of those variations. Variations can come from the image acquisition process, for example, due to different cell staining protocols or tissue deformation. To tackle this challenge, Data Augmentation (DA) can be used during training to generate additional samples by applying transformations to existing ones, to help the model become invariant to those color and shape transformations. The problem with DA is that it is not only dataset-specific but it also requires domain knowledge, which is not always available. Without this knowledge, selecting the right transformations can only be done using heuristics or through a computationally demanding search. To address this, we propose an automatic DA learning method. In this method, the DA parameters, i.e. the transformation parameters needed to improve the model training, are considered learnable and are learned automatically using a bilevel optimization approach in a quick and efficient way using truncated backpropagation. We validated the method on six different datasets. Experimental results show that our model can learn color and affine transformations that are more helpful to train an image classifier than predefined DA transformations, which are also more expensive as they need to be selected before the training by grid search on a validation set. We also show that similarly to a model trained with RandAugment, our model has also only a few method-specific hyperparameters to tune but is performing better. This makes our model a good solution for learning the best DA parameters, especially in the context of histopathological images, where defining potentially useful transformation heuristically is not trivial.

Data Augmentation Based on DiscrimDiff for Histopathology Image Classification

Boosting Unsupervised Contrastive Learning Using Diffusion-Based Data Augmentation from Scratch

Synthetic Augmentation with Large-scale Unconditional Pre-training

A Data Augmentation Methodology to Reduce the Class Imbalance in Histopathology Images

DAug: Diffusion-based Channel Augmentation for Radiology Image Retrieval and Classification

Diffusion Models-based Data Augmentation for the Cell Cycle Phase Classification

Automatic data augmentation to improve generalization of deep learning in H&E stained histopathology

Histo-Diffusion: A Diffusion Super-Resolution Method for Digital Pathology with Comprehensive Quality Assessment

Automatic Data Augmentation Learning using Bilevel Optimization for Histopathological Images

A Simple Background Augmentation Method for Object Detection with Diffusion Model

Data Augmentation in Class-Conditional Diffusion Model for Semi-Supervised Medical Image Segmentation

Selective Synthetic Augmentation with HistoGAN for Improved Histopathology Image Classification

Improved Domain Generalization for Cell Detection in Histopathology Images via Test-Time Stain Augmentation

Dataset Distillation for Histopathology Image Classification

Diffusion-based Data Augmentation for Skin Disease Classification: Impact Across Original Medical Datasets to Fully Synthetic Images

Controllable and Efficient Multi-Class Pathology Nuclei Data Augmentation using Text-Conditioned Diffusion Models

Diffusion-based Data Augmentation for Nuclei Image Segmentation

CellMix: A General Instance Relationship based Method for Data Augmentation Towards Pathology Image Classification

Effective Data Augmentation With Diffusion Models