arXiv:2608.18907cs.CVcs.AI2026-08

根据模型学习状态动态调整数据增强强度,提升小样本图像分类效果。

Learning-State-Aware Dynamic Generative Data Augmentation on Small-Scale Datasets

论文配图:Learning-State-Aware Dynamic Generative Data Augmentation on Small-Scale Datasets
图 1 · 摘自论文原文
  • 基于损失与损失下降率构建样本学习状态,决定增强强度。
  • 在自然图像上平均提升4.5%,医学图像上提升2.5%。
  • 适合小样本图像分类任务,尤其对医疗图像有效。

小规模图像分类常因训练数据不足受限。基于预训练生成模型的生成式数据增强(GDA)已成为有效解决方案。然而,现有方法依赖任务无关的增强策略,忽视下游模型需求。尽管近期动态GDA方法引入模型反馈指导增强,仍难以可靠确定样本级增强强度,且无法自适应不同图像区域,难以平衡图像多样性与类别语义。为此,我们提出学习状态感知的动态生成数据增强(LSADA)。具体而言,LSADA基于每个样本当前损失及其下降速率构建学习状态,并映射为样本级增强强度。此外,LSADA提出解耦的数据增强与扩散融合策略,对类别相关区域施加可控变换,生成多样化的类别无关区域,逐步融合以提升图像多样性并保留类别语义。在九个公开数据集上的实验表明,LSADA在六个自然图像数据集上平均优于现有最优动态GDA方法4.5%,在三个医学图像数据集上提升2.5%。

原文摘要 · Abstract (English)

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.

小样本学习数据增强生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。