vix.ing · top · new · best · stats · spec

RandAugment: Practical automated data augmentation with a reduced search\n space

2019/09/30 by Ekin D. Cubuk, Cubuk, Ekin D., Barret Zoph +5 · 114 citations
Computer Science · #Advanced Neural Network Applications #Computer Vision and Pattern Recognition (cs.CV) #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #Multimodal Machine Learning Applications

paper · pdf · doi:10.48550/arxiv.1909.13719

openalex publication_date 2019/09/30 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Recent work has shown that data augmentation has the potential to\nsignificantly improve the generalization of deep learning models. Recently,\nautomated augmentation strategies have led to state-of-the-art results in image\nclassification and object detection. While these strategies were optimized for\nimproving validation accuracy, they also led to state-of-the-art results in\nsemi-supervised learning and improved robustness to common corruptions of\nimages. An obstacle to a large-scale adoption of these methods is a separate\nsearch phase which increases the training complexity and may substantially\nincrease the computational cost. Additionally, due to the separate search\nphase, these approaches are unable to adjust the regularization strength based\non model or dataset size. Automated augmentation policies are often found by\ntraining small models on small datasets and subsequently applied to train\nlarger models. In this work, we remove both of these obstacles. RandAugment has\na significantly reduced search space which allows it to be trained on the\ntarget task with no need for a separate proxy task. Furthermore, due to the\nparameterization, the regularization strength may be tailored to different\nmodel and dataset sizes. RandAugment can be used uniformly across different\ntasks and datasets and works out of the box, matching or surpassing all\nprevious automated augmentation approaches on CIFAR-10/100, SVHN, and ImageNet.\nOn the ImageNet dataset we achieve 85.0% accuracy, a 0.6% increase over the\nprevious state-of-the-art and 1.0% increase over baseline augmentation. On\nobject detection, RandAugment leads to 1.0-1.3% improvement over baseline\naugmentation, and is within 0.3% mAP of AutoAugment on COCO. Finally, due to\nits interpretable hyperparameter, RandAugment may be used to investigate the\nrole of data augmentation with varying model and dataset size. Code is\navailable online.\n

Cited by

Related