Enabling Data Diversity: Efficient Automatic Augmentation via Regularized Adversarial Training
Data augmentation has proved extremely useful by increasing training data variance to alleviate overfitting and improve deep neural networks' generalization performance. In medical image analysis, a well-designed augmentation policy usually requires much expert knowledge and is difficult to generalize to multiple tasks due to the vast discrepancies among pixel intensities, image appearances, and object shapes in different medical tasks. To automate medical data augmentation, we propose a regularized adversarial training framework via two min-max objectives and three differentiable augmentation models covering affine transformation, deformation, and appearance changes. Our method is more automatic and efficient than previous automatic augmentation methods, which still rely on pre-defined operations with human-specified ranges and costly bi-level optimization. Extensive experiments demonstrated that our approach, with less training overhead, achieves superior performance over state-of-the-art auto-augmentation methods on both tasks of 2D skin cancer classification and 3D organs-at-risk segmentation.
Code (1)
Tasks
Cancer ClassificationData AugmentationDiversityMedical Image AnalysisSkin Cancer ClassificationSimilar Papers 제목 키워드 기반
Virtual Data Augmentation: A Robust and General Framework for Fine-tuning Pre-trained Models
Recent works have shown that powerful pre-trained language models (PLM) can be fooled by small perturbations or intentional attacks. To solve this issue, various data augmentation techniques are proposed to improve the r…
Data AugmentationDiversityLanguage ModelingLanguage ModellingAdaptive Divergence Regularized Policy Optimization for Fine-tuning Generative Models
Balancing exploration and exploitation during reinforcement learning fine-tuning of generative models presents a critical challenge, as existing approaches rely on fixed divergence regularization that creates an inherent…
Text-to-Image GenerationReinforcement LearningStyle TransferDivAug: Plug-in Automated Data Augmentation with Explicit Diversity Maximization
Human-designed data augmentation strategies have been replaced by automatically learned augmentation policy in the past two years. Specifically, recent work has empirically shown that the superior performance of the auto…
Data AugmentationDiversityFully Automatic Video Colorization with Self-Regularization and Diversity
We present a fully automatic approach to video colorization with self-regularization and diversity. Our model contains a colorization network for video frame colorization and a refinement network for spatiotemporal color…
ColorizationDiversityImage Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from Pixels
We propose a simple data augmentation technique that can be applied to standard model-free reinforcement learning algorithms, enabling robust learning directly from pixels without the need for auxiliary losses or pre-tra…
AllAtari Games 100kContinuous ControlContrastive Learning+6