中文

Few-Shot Synthetic Data Generation with Diffusion Models for Downstream Vision Tasks

计算机视觉与模式识别 2026-05-13 v1

摘要

Class imbalance is a persistent challenge in visual recognition, particularly in safety-critical domains where collecting positive examples is expensive and rare events are inherently underrepresented. We propose a lightweight synthetic data augmentation pipeline that fine-tunes a LoRA adapter on as few as 20-50 real images of a rare class and uses a pretrained diffusion model to generate synthetic samples for training. We systematically vary the synthetic-to-real ratio and evaluate the approach across two structurally different domains: chest X-ray pathology classification (NIH ChestX-ray14) and industrial surface crack detection (Magnetic Tile Defect dataset). All evaluations are performed on held-out sets of real images only. Across both domains, synthetic augmentation consistently improves rare-class recall and F1 compared to training with real data alone. Performance improves with moderate synthetic augmentation and shows diminishing returns as the synthetic ratio increases. These results suggest that LoRA-adapted diffusion models provide a simple and scalable mechanism for augmenting rare classes, enabling effective learning in data-scarce scenarios across heterogeneous visual domains.

关键词

引用

@article{arxiv.2605.11898,
  title  = {Few-Shot Synthetic Data Generation with Diffusion Models for Downstream Vision Tasks},
  author = {Daniil Dushenev and Nazariy Karpov and Daniil Zinovjev and Alexander Gorin and Konstantin Kulikov},
  journal= {arXiv preprint arXiv:2605.11898},
  year   = {2026}
}

备注

5 pages, 3 figures, 1 table. Accepted at SynData4CV Workshop @ CVPR 2026