Search papers, labs, and topics across Lattice.
This paper introduces learning-state-aware dynamic generative data augmentation (LSADA), which enhances small-scale image classification by tailoring data augmentation strategies to individual sample learning states. By mapping a sample's current loss and loss-decrease rate to specific augmentation strengths, LSADA effectively balances the need for image diversity with the preservation of class semantics. Experimental results demonstrate that LSADA significantly outperforms the state-of-the-art dynamic generative data augmentation method, achieving an average improvement of 4.5% on natural images and 2.5% on medical images across nine datasets.
Tailoring data augmentation to individual learning states boosts performance by an average of 4.5% on natural images, revealing a new frontier in generative data strategies.
Small-scale image classification is often limited by the scarcity of training data. Generative data augmentation (GDA) based on pretrained generative models has emerged as an effective solution. However, existing methods rely on task-agnostic augmentation strategies that overlook downstream model needs. Although recent dynamic GDA methods incorporate model feedback to guide augmentation, they still struggle to reliably determine sample-specific augmentation strengths and adapt augmentation strategies to different image regions while balancing image diversity and class semantics. To address these issues, we propose learning-state-aware dynamic generative data augmentation (LSADA). Specifically, LSADA constructs a learning state for each sample based on its current loss and loss-decrease rate, which is then mapped to a sample-specific augmentation strength. Furthermore, LSADA introduces a decoupled data augmentation and diffusion fusion strategy that applies strength-controlled transformations to class-relevant regions and generates diverse class-irrelevant regions, progressively fusing them to improve image diversity while preserving class semantics. Experiments on nine public datasets show that LSADA outperforms the existing SOTA dynamic GDA method by an average of 4.5% on six natural image datasets and 2.5% on three medical image datasets.
