MedAugment: Universal Automatic Data Augmentation Plug-in for Medical Image Analysis
Not on the map. Why?
Abstract (arXiv)
Data augmentation (DA) has been widely leveraged in computer vision to alleviate data shortage, while its application in medical imaging faces multiple challenges. The prevalent DA approaches in medical image analysis encompass conventional DA, synthetic DA, and automatic DA. However, these approaches may result in experience-driven design and intensive computation costs. Here, we propose a suitable yet general automatic DA method for medical images termed MedAugment. We propose pixel and spatial augmentation spaces and exclude the operations that can break medical details and features. Besides, we propose a sampling strategy by sampling a limited number of operations from the two spaces. Moreover, we present a hyperparameter mapping relationship to produce a rational augmentation level and make the MedAugment fully controllable using a single hyperparameter. These configurations settle the differences between natural and medical images. Extensive experimental results on four classification and four segmentation datasets demonstrate the superiority of MedAugment. Compared with existing approaches, the proposed MedAugment prevents producing color distortions or structural alterations while involving negligible computational overhead. Our method can serve as a plugin without an extra training stage, offering significant benefits to the community and medical experts lacking a deep learning foundation. The code is available at https://github.com/NUS-Tim/MedAugment.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 53 of its 87 references that are in the map. First posted on arXiv 30 Jun 2023.
Builds on
53 of 87 references.
References · 53
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 6,013
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2022A ConvNet for the 2020scited by 3,579
- 2016Wide Residual Networkscited by 3,070
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,988
38 more
- 2016Improved Techniques for Training GANscited by 2,970
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
- 2016Aggregated Residual Transformations for Deep Neural Networkscited by 2,546
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,269
- 2017Rethinking Atrous Convolution for Semantic Image Segmentationcited by 2,161
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 2020Training Generative Adversarial Networks with Limited Datacited by 1,186
- 2019AugMix: A Simple Data Processing Method to Improve Robustness and Uncertaintycited by 901
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
- 2015Dueling Network Architectures for Deep Reinforcement Learningcited by 785
- 2018Albumentations: fast and flexible image augmentationscited by 560
- 2022Brain Imaging Generation with Latent Diffusion Modelscited by 268
- 2019Fast AutoAugmentcited by 243
- 2018GAN-based Synthetic Medical Image Augmentation for increased CNN Performance in Liver Lesion Classificationcited by 242
- 2021TrivialAugment: Tuning-free Yet State-of-the-Art Data Augmentationcited by 227
- 2015Deep Generative Image Models using a Laplacian Pyramid of Adversarial Networkscited by 214
- 2020TorchIO: A Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learningcited by 177
- 2016Deep Pyramidal Residual Networkscited by 163
- 2019Population Based Augmentation: Efficient Learning of Augmentation Policy Schedulescited by 154
- 2021Semantic Segmentation with Generative Models: Semi-Supervised Learning and Strong Out-of-Domain Generalizationcited by 141
- 2021Causality-inspired Single-source Domain Generalization for Medical Image Segmentationcited by 119
- 2022A Morphology Focused Diffusion Probabilistic Model for Synthesis of Histopathology Imagescited by 74
- 2017A Bayesian Data Augmentation Approach for Learning Deep Modelscited by 50
- 2020UniformAugment: A Search-free Probabilistic Data Augmentation Approachcited by 38
- 2018Fully Convolutional Multi-scale Residual DenseNets for Cardiac Segmentation and Automated Cardiac Diagnosis using Ensemble of Classifierscited by 35
- 2022When do GANs replicate? On the choice of dataset sizecited by 34
- 2022Transforming medical imaging with Transformers? A comparative review of key properties, current progresses, and future perspectivescited by 33
- 2020Semi-supervised Task-driven Data Augmentation for Medical Image Segmentationcited by 24
- 2015APAC: Augmented PAttern Classification with Neural Networkscited by 23
- 2020Automatic Data Augmentation for 3D Medical Image Segmentationcited by 22
- 2023Why is the winner the best?cited by 21
- 2021Enhancing MR Image Segmentation with Realistic Adversarial Data Augmentationcited by 20
- 2022AADG: Automatic Augmentation for Domain Generalization on Retinal Image Segmentationcited by 18
- 2018High-resolution medical image synthesis using progressively grown generative adversarial networkscited by 17
- 2020Searching Learning Strategy with Reinforcement Learning for 3D Medical Image Segmentationcited by 17
- 2022Recent Progress in Transformer-based Medical Image Analysiscited by 9
- 2020Automatic Data Augmentation via Deep Reinforcement Learning for Effective Kidney Tumor Segmentationcited by 8
- 2023Multi-Level Global Context Cross Consistency Model for Semi-Supervised Ultrasound Image Segmentation with Diffusion Modelcited by 8
50 most cited of 53 All 53 (JSON)
Built on by
Citing papers per year: 2024: 3; 2025: 1.
4 papers.
- 2025Revisiting Data Augmentation for Ultrasound Imagescited by 7
- 2024BSDA: Bayesian Random Semantic Data Augmentation for Medical Image Classificationcited by 5
- 2024Analyzing Data Augmentation for Medical Images: A Case Study in Ultrasound Imagescited by 4
- 2024Segmenting Medical Images with Limited Datacited by 3
- 2025Revisiting Data Augmentation for Ultrasound Images22 Jan 2025
- 2024Segmenting Medical Images with Limited Data12 Jul 2024
- 2024Analyzing Data Augmentation for Medical Images: A Case Study in Ultrasound Images14 Mar 2024
- 2024BSDA: Bayesian Random Semantic Data Augmentation for Medical Image Classification10 Mar 2024