swMATH ID: 39069
Software Authors: Ekin D. Cubuk, Barret Zoph, Dandelion Mane, Vijay Vasudevan, Quoc V. Le
Description: AutoAugment: Learning Augmentation Policies from Data. Data augmentation is an effective technique for improving the accuracy of modern image classifiers. However, current data augmentation implementations are manually designed. In this paper, we describe a simple procedure called AutoAugment to automatically search for improved data augmentation policies. In our implementation, we have designed a search space where a policy consists of many sub-policies, one of which is randomly chosen for each image in each mini-batch. A sub-policy consists of two operations, each operation being an image processing function such as translation, rotation, or shearing, and the probabilities and magnitudes with which the functions are applied. We use a search algorithm to find the best policy such that the neural network yields the highest validation accuracy on a target dataset. Our method achieves state-of-the-art accuracy on CIFAR-10, CIFAR-100, SVHN, and ImageNet (without additional data). On ImageNet, we attain a Top-1 accuracy of 83.5
Homepage: https://paperswithcode.com/paper/autoaugment-learning-augmentation-policies
Related Software: ImageNet; Adam; mixup; CIFAR; AlexNet; TensorFlow; Python; MS-COCO; PyTorch; Faster R-CNN; MNIST; DeepLoc; ProSelfLC; MixMatch; AugMix; SGDR; TADAM; SpecAugment; S4L; RandAugment
Cited in: 7 Publications

Citations by Year