Dataset Augmentation by Mixing Visual Concepts
On the map: Diffusion models → Image generation, text-to-image and editing
Abstract (arXiv)
This paper proposes a dataset augmentation method by fine-tuning pre-trained diffusion models. Generating images using a pre-trained diffusion model with textual conditioning often results in domain discrepancy between real data and generated images. We propose a fine-tuning approach where we adapt the diffusion model by conditioning it with real images and novel text embeddings. We introduce a unique procedure called Mixing Visual Concepts (MVC) where we create novel text embeddings from image captions. The MVC enables us to generate multiple images which are diverse and yet similar to the real data enabling us to perform effective dataset augmentation. We perform comprehensive qualitative and quantitative evaluations with the proposed dataset augmentation approach showcasing both coarse-grained and finegrained changes in generated images. Our approach outperforms state-of-the-art augmentation techniques on benchmark classification tasks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 37 of its 49 references that are in the map. First posted on arXiv 19 Dec 2024.
Builds on
37 of 49 references.
Closest ancestors · 6
Same direction, newest first
- 2023Synthetic Data from Diffusion Models Improves ImageNet Classification17 Apr
- 2023Adding Conditional Control to Text-to-Image Diffusion Models10 Feb
- 2023Effective Data Augmentation With Diffusion Models7 Feb
- 2022InstructPix2Pix: Learning to Follow Image Editing Instructions17 Nov
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understanding23 May
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latents13 Apr
Other references · 31
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,137
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 6,013
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2016Wide Residual Networkscited by 3,070
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,269
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2015FlowNet: Learning Optical Flow with Convolutional Networkscited by 1,059
19 more
- 2022Diffusion Models in Vision: A Surveycited by 1,015
- 2019An Introduction to Variational Autoencoderscited by 1,000
- 2016Conditional Image Generation with PixelCNN Decoderscited by 862
- 2016Playing for Data: Ground Truth from Computer Gamescited by 818
- 2018AutoAugment: Learning Augmentation Policies from Datacited by 730
- 2020Jukebox: A Generative Model for Musiccited by 542
- 2017VisDA: The Visual Domain Adaptation Challengecited by 435
- 2022Is synthetic data from generative models ready for image recognition?cited by 320
- 2021DatasetGAN: Efficient Labeled Data Factory with Minimal Human Effortcited by 264
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 252
- 2019Fast AutoAugmentcited by 243
- 2020A Review on Generative Adversarial Networks: Algorithms, Theory, and Applicationscited by 209
- 2017Dataset Augmentation in Feature Spacecited by 124
- 2021Generative Models as a Data Source for Multiview Representation Learningcited by 104
- 2022Distilling Model Failures as Directions in Latent Spacecited by 82
- 2019This dataset does not exist: training models from generated imagescited by 65
- 2017Smart Augmentation - Learning an Optimal Data Augmentation Strategycited by 54
- 2017A Bayesian Data Augmentation Approach for Learning Deep Modelscited by 50
- 2016Deep Directed Generative Models with Energy-Based Probability Estimationcited by 34
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Semantic Augmentation in Images using Language
- 2025Context-guided Responsible Data Augmentation with Diffusion Models
- 2022Cap2Aug: Caption guided Image to Image data Augmentation
- 2025SGD-Mix: Enhancing Domain-Specific Image Classification with Label-Preserving Data Augmentation
- 2024Advances in Diffusion Models for Image Data Augmentation: A Review of Methods, Models, Evaluation Metrics and Future Research Directions
- 2024Enhance Image Classification via Inter-Class Image Mixup with Diffusion Model
- 2023Detail Reinforcement Diffusion Model: Augmentation Fine-Grained Visual Categorization in Few-Shot Conditions
- 2023Multimodal Data Augmentation for Image Captioning using Diffusion Models