Abstract:Data augmentation has been an essential technique for improving the generalization ability of deep neural networks in image classification tasks. However, intensive changes in appearance and different degrees of occlusion in images are the key factors that severely affect the generalization ability of image classification models. Therefore, in order to enhance the generalization performance and robustness of deep models, data augmentation approaches by providing models with more diverse training data in various scenarios are widely applied. Although many existing data augmentation methods simulate occlusion in the augmented images to enhance the generalization of models, these methods randomly delete some areas in images without considering the semantic information of images. In this work, we propose a novel data augmentation method named AdvMask for image classification based on sparse adversarial attack techniques. AdvMask first identifies the key points that have the greatest influence on the classification results via a proposed end-to-end sparse adversarial attack module. During the data augmentation process, AdvMask efficiently generates diverse augmented data with structured occlusions based on the key points. By doing so, AdvMask can force deep models to seek other relevant content while the most discriminative content is hidden. Extensive experimental results on various benchmark datasets and deep models demonstrate that our proposed method can effectively improve the generalization performance of deep models and significantly outperforms previous data augmentation methods. Code for reproducing our results is available at https://github.com/Jackbrocp/AdvMask.

The Performance Research of the Data Augmentation Method for Image Classification

Boosting Unsupervised Contrastive Learning Using Diffusion-Based Data Augmentation from Scratch

A Preliminary Study on Data Augmentation of Deep Learning for Image Classification

The Effectiveness of Data Augmentation in Image Classification using Deep Learning

On the Generalization Effects of Linear Transformations in Data Augmentation

Exploring Data Augmentations on Self-/Semi-/Fully- Supervised Pre-trained Models

Decoupled Data Augmentation for Improving Image Classification

Data Augmentation Revisited: Rethinking the Distribution Gap between Clean and Augmented Data

Image Data Augmentation for Deep Learning: A Survey

DualAug: Exploiting Additional Heavy Augmentation with OOD Data Rejection

Understanding the Detrimental Class-level Effects of Data Augmentation

Learning Optimal Data Augmentation Policies via Bayesian Optimization for Image Classification Tasks

Self-transformation of Encoded Subregion Image Data Augmentation for Better Classification

Enhancing Performance of Deep Learning Models with a Novel Data Augmentation Approach

A Survey of Automated Data Augmentation Algorithms for Deep Learning-based Image Classification Tasks

Optimizing Few-Shot Remote Sensing Scene Classification Based on an Improved Data Augmentation Approach

AdvMask: A Sparse Adversarial Attack Based Data Augmentation Method for Image Classification

Interpretability-Mask: a Label-Preserving Data Augmentation Scheme for Better Classification

Data Augmentation for Face Recognition

Enhanced Transfer Learning with Data Augmentation

See Better Before Looking Closer: Weakly Supervised Data Augmentation Network for Fine-Grained Visual Classification