AutoAugment: Learning Augmentation Strategies from Data

AutoAugment: Learning Augmentation Strategies from Data

11 Apr 2019 | Ekin D. Cubuk *, Barret Zoph; Dandelion Mané, Vijay Vasudevan, Quoc V. Le
AutoAugment is a method for automatically learning data augmentation policies to improve the accuracy of modern image classifiers. The authors describe a search space where a policy consists of multiple sub-policies, each containing two image processing operations (e.g., translation, rotation, or shearing) with associated probabilities and magnitudes. A reinforcement learning algorithm is used to find the best policy that maximizes validation accuracy on a target dataset. The method achieves state-of-the-art accuracy on several datasets, including CIFAR-10, CIFAR-100, SVHN, and ImageNet, without additional data. The learned augmentation policies are also transferable to other datasets, showing significant improvements on various fine-grained classification tasks. The paper discusses the effectiveness of AutoAugment, compares it to other automated data augmentation methods, and provides ablation experiments to validate its key components.AutoAugment is a method for automatically learning data augmentation policies to improve the accuracy of modern image classifiers. The authors describe a search space where a policy consists of multiple sub-policies, each containing two image processing operations (e.g., translation, rotation, or shearing) with associated probabilities and magnitudes. A reinforcement learning algorithm is used to find the best policy that maximizes validation accuracy on a target dataset. The method achieves state-of-the-art accuracy on several datasets, including CIFAR-10, CIFAR-100, SVHN, and ImageNet, without additional data. The learned augmentation policies are also transferable to other datasets, showing significant improvements on various fine-grained classification tasks. The paper discusses the effectiveness of AutoAugment, compares it to other automated data augmentation methods, and provides ablation experiments to validate its key components.
Reach us at info@study.space