Abstract:Medical image segmentation often faces challenges related to overfitting, primarily due to the limited and complex training samples. This challenge often prompts the use of self-supervised learning and data augmentation. However, self-supervised learning requires well-defined hand-crafted tasks and multiple training stages. On the other hand, basic image augmentation techniques like cropping, rotation, and flipping, effective for natural scene images, have limited efficacy for medical images due to their isotropic nature. While regional dropout regularization data augmentation methods have proven effective in image recognition tasks, their application in image segmentation is not as extensively studied. Additionally, existing augmentation methods often operate on square regions, leading to the loss of crucial contour information. This is particularly problematic for medical image segmentation tasks dealing with regions of interest characterized by intricate shapes. In this work, we introduce LCAMix, a novel data augmentation approach designed for medical image segmentation. LCAMix operates by blending two images and their segmentation masks based on their superpixels, incorporating a local-and-contour-aware strategy. The training process on augmented images adopts two auxiliary pretext tasks: firstly, classifying local superpixels in augmented images using an adaptive focal margin, leveraging segmentation ground truth masks as prior knowledge; secondly, reconstructing the two source images using mixed superpixels as mutual masks, emphasizing spatial sensitivity. Our method stands out as a simple, one-stage, model-agnostic, and plug-and-play data augmentation solution applicable to various segmentation tasks. Notably, it requires no external data or additional models. Extensive experiments validate its superior performance across diverse medical segmentation datasets and tasks. The source codes are available at https://github.com/DanielaPlusPlus/DataAug4Medical .

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

GenMix: Combining Generative and Mixture Data Augmentation for Medical Image Classification

LesionMix: A Lesion-Level Data Augmentation Method for Medical Image Segmentation

GradMix for nuclei segmentation and classification in imbalanced pathology image datasets

ContriMix: Scalable stain color augmentation for domain generalization without domain labels in digital pathology

InsMix: Towards Realistic Generative Data Augmentation for Nuclei Instance Segmentation

PathUp: Patch-wise Timestep Tracking for Multi-class Large Pathology Image Synthesising Diffusion Model

Mixing Histopathology Prototypes into Robust Slide-Level Representations for Cancer Subtyping

Data Augmentation Based on DiscrimDiff for Histopathology Image Classification

Improve Global Glomerulosclerosis Classification with Imbalanced Data using CircleMix Augmentation

PreMix: Boosting Multiple Instance Learning in Digital Histopathology through Pre-training with Intra-Batch Slide Mixing

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

NucleiMix: Realistic Data Augmentation for Nuclei Instance Segmentation

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

CalAttnMix: an Instance-Attention Based Data Augmentation Method for Calcification Detection in Mammograms

Synthetic Augmentation with Large-scale Unconditional Pre-training

AnatoMix: Anatomy-aware Data Augmentation for Multi-organ Segmentation

CellMixer: Annotation-free Semantic Cell Segmentation of Heterogeneous Cell Populations

TransformMix: Learning Transformation and Mixing Strategies from Data

LCAMix: Local-and-contour aware grid mixing based data augmentation for medical image segmentation

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