Abstract
In this paper, we take a closer look at data augmentation for images, anddescribe a simple procedure called AutoAugment to search for improved dataaugmentation policies. Our key insight is to create a search space of dataaugmentation policies, evaluating the quality of a particular policy directlyon the dataset of interest. In our implementation, we have designed a searchspace where a policy consists of many sub-policies, one of which is randomlychosen for each image in each mini-batch. A sub-policy consists of twooperations, each operation being an image processing function such astranslation, rotation, or shearing, and the probabilities and magnitudes withwhich the functions are applied. We use a search algorithm to find the bestpolicy such that the neural network yields the highest validation accuracy on atarget dataset. Our method achieves state-of-the-art accuracy on CIFAR-10,CIFAR-100, SVHN, and ImageNet (without additional data). On ImageNet, we attaina Top-1 accuracy of 83.54%. On CIFAR-10, we achieve an error rate of 1.48%,which is 0.65% better than the previous state-of-the-art. On reduced datasettings, AutoAugment performs comparably to semi-supervised methods withoutusing any unlabeled examples. Finally, policies learned from one dataset can betransferred to work well on other similar datasets. For example, the policylearned on ImageNet allows us to achieve state-of-the-art accuracy on the finegrained visual classification dataset Stanford Cars, without fine-tuningweights pre-trained on additional data.