AdaAugment: A Tuning-Free and Adaptive Approach to Enhance Data Augmentation

TL;DR

AdaAugment introduces a tuning-free, adaptive data augmentation method that dynamically adjusts augmentation magnitudes during training using reinforcement learning, leading to improved model performance and efficiency.

Contribution

It proposes a novel reinforcement learning-based adaptive augmentation framework that automatically tunes augmentation magnitudes without manual intervention.

Findings

Outperforms state-of-the-art DA methods on benchmark datasets

Maintains high efficiency during training

Effectively adapts augmentation to training progress

Abstract

Data augmentation (DA) is widely employed to improve the generalization performance of deep models. However, most existing DA methods employ augmentation operations with fixed or random magnitudes throughout the training process. While this fosters data diversity, it can also inevitably introduce uncontrolled variability in augmented data, which could potentially cause misalignment with the evolving training status of the target models. Both theoretical and empirical findings suggest that this misalignment increases the risks of both underfitting and overfitting. To address these limitations, we propose AdaAugment, an innovative and tuning-free adaptive augmentation method that leverages reinforcement learning to dynamically and adaptively adjust augmentation magnitudes for individual training samples based on real-time feedback from the target network. Specifically, AdaAugment features a dual-model architecture consisting of a policy network and a target network, which are jointly optimized to adapt augmentation magnitudes in accordance with the model's training progress effectively. The policy network optimizes the variability within the augmented data, while the target network utilizes the adaptively augmented samples for training. These two networks are jointly optimized and mutually reinforce each other. Extensive experiments across benchmark datasets and deep architectures demonstrate that AdaAugment consistently outperforms other state-of-the-art DA methods in effectiveness while maintaining remarkable efficiency. Code is available at https://github.com/Jackbrocp/AdaAugment.