ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
13k
Citations
315
Influential Citations
Journal Of Big Data
Venue
2019
Year
Deep convolutional neural networks have performed remarkably well on many Computer Vision tasks. However, these networks are heavily reliant on big data to avoid overfitting. Overfitting refers to the phenomenon when a network learns a function with very high variance such as to perfectly model the training data. Unfortunately, many application domains do not have access to big data, such as medical image analysis. This survey focuses on Data Augmentation, a data-space solution to the problem of limited data. Data Augmentation encompasses a suite of techniques that enhance the size and quality of training datasets such that better Deep Learning models can be built using them. The image augmentation algorithms discussed in this survey include geometric transformations, color space augmentations, kernel filters, mixing images, random erasing, feature space augmentation, adversarial training, generative adversarial networks, neural style transfer, and meta-learning. The application of augmentation methods based on GANs are heavily covered in this survey. In addition to augmentation techniques, this paper will briefly discuss other characteristics of Data Augmentation such as test-time augmentation, resolution impact, final dataset size, and curriculum learning. This survey will present existing methods for Data Augmentation, promising developments, and meta-level decisions for implementing Data Augmentation. Readers will understand how Data Augmentation can improve the performance of their models and expand limited datasets to take advantage of the capabilities of big data.
This survey addresses a fundamental bottleneck in deep learning: the reliance on large labeled datasets to prevent overfitting. In many real-world domains—especially medical imaging, autonomous driving, and remote sensing—collecting massive annotated datasets is impractical or prohibitively expensive. By systematically cataloging data augmentation techniques, the paper provides a practical toolkit for practitioners to expand limited datasets and improve model robustness. Its high citation count (12,619) underscores its role as a foundational reference in the field.
The paper is particularly timely because it covers emerging methods like GAN-based augmentation and neural style transfer, which go beyond traditional geometric and color transformations. These advanced techniques can generate semantically meaningful variations, potentially closing the gap between small and large dataset performance. The survey also introduces meta-level considerations such as test-time augmentation and curriculum learning, which are often overlooked in practice.
As a survey, the paper does not present original experimental results. However, it synthesizes findings from numerous prior studies to demonstrate that data augmentation consistently reduces overfitting and improves generalization across diverse computer vision tasks. For example, it references work showing that simple geometric transformations (e.g., rotation, flipping) can yield 2-5% accuracy gains on benchmark datasets like CIFAR-10 and ImageNet. GAN-based augmentation has been shown to match or exceed performance of traditional methods on medical imaging tasks where data is scarce.
This survey has had a profound impact on the AI community by democratizing knowledge about data augmentation. It has become a go-to resource for both researchers and practitioners, influencing how models are trained in production environments. The emphasis on GAN-based augmentation has spurred further research into generative models for data synthesis, while the meta-level discussion has encouraged systematic experimentation with augmentation policies (e.g., AutoAugment). By lowering the barrier to effective augmentation, the paper has contributed to the broader adoption of deep learning in data-constrained domains.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba