Abstract:Data augmentation is essential to achieve state-of-the-art performance in many deep learning applications. However, the most effective augmentation techniques become computationally prohibitive for even medium-sized datasets. To address this, we propose a rigorous technique to select subsets of data points that when augmented, closely capture the training dynamics of full data augmentation. We first show that data augmentation, modeled as additive perturbations, improves learning and generalization by relatively enlarging and perturbing the smaller singular values of the network Jacobian, while preserving its prominent directions. This prevents overfitting and enhances learning the harder to learn information. Then, we propose a framework to iteratively extract small subsets of training data that when augmented, closely capture the alignment of the fully augmented Jacobian with labels/residuals. We prove that stochastic gradient descent applied to the augmented subsets found by our approach has similar training dynamics to that of fully augmented data. Our experiments demonstrate that our method achieves 6.3x speedup on CIFAR10 and 2.2x speedup on SVHN, and outperforms the baselines by up to 10% across various subset sizes. Similarly, on TinyImageNet and ImageNet, our method beats the baselines by up to 8%, while achieving up to 3.3x speedup across various subset sizes. Finally, training on and augmenting 50% subsets using our method on a version of CIFAR10 corrupted with label noise even outperforms using the full dataset. Our code is available at: <a class="link-external link-https" href="https://github.com/tianyu139/data-efficient-augmentation" rel="external noopener nofollow">this https URL</a>

Automatic Data Augmentation by Learning the Deterministic Policy

Deep AutoAugment

AutoAugment: Learning Augmentation Policies from Data

Learning Data Augmentation Policies Using Augmented Random Search.

AdaAugment: A Tuning-Free and Adaptive Approach to Enhance Data Augmentation

Data-Efficient Augmentation for Training Neural Networks

MetaAugment: Sample-Aware Data Augmentation Policy Learning

EventAugment: Learning Augmentation Policies from Asynchronous Event-based Data

DADA: Differentiable Automatic Data Augmentation

SelectAugment: Hierarchical Deterministic Sample Selection for Data Augmentation

A Good Data Augmentation Policy Is Not All You Need: A Multi-Task Learning Perspective

A Deep Data Augmentation Framework Based on Generative Adversarial Networks

Universal Adaptive Data Augmentation

Population Based Augmentation: Efficient Learning of Augmentation Policy Schedules

DynamicAug: Enhancing Transfer Learning Through Dynamic Data Augmentation Strategies Based on Model State

Automatic Data Augmentation for Generalization in Deep Reinforcement Learning

LA3: Efficient Label-Aware AutoAugment

Automatic Data Augmentation for Generalization in Reinforcement Learning

DualAug: Exploiting Additional Heavy Augmentation with OOD Data Rejection

Local Patch AutoAugment with Multi-Agent Collaboration

BO-Aug: Learning Data Augmentation Policies Via Bayesian Optimization