When Generative Augmentation Hurts: A Benchmark Study of GAN and Diffusion Models for Bias Correction in AI Classification Systems
On the map: Diffusion models → Image generation, text-to-image and editing
Abstract (arXiv)
Generative models are widely used to compensate for class imbalance in AI training pipelines, yet their failure modes under low-data conditions are poorly understood. This paper reports a controlled benchmark comparing three augmentation strategies applied to a fine-grained animal classification task: traditional transforms, FastGAN, and Stable Diffusion 1.5 fine-tuned with Low-Rank Adaptation (LoRA). Using the Oxford-IIIT Pet Dataset with eight artificially underrepresented breeds, we find that FastGAN augmentation does not merely underperform at very low training set sizes but actively increases classifier bias, with a statistically significant large effect across three random seeds (bias gap increase: +20.7%, Cohen's d = +5.03, p = 0.013). The effect size here is large enough to give confidence in the direction of the finding despite the small number of seeds. Feature embedding analysis using t-distributed Stochastic Neighbor Embedding reveals that FastGAN images for severe-minority breeds form tight isolated clusters outside the real image distribution, a pattern consistent with mode collapse. Stable Diffusion with Low-Rank Adaptation produced the best results overall, achieving the highest macro F1 (0.9125 plus or minus 0.0047) and a 13.1% reduction in the bias gap relative to the unaugmented baseline. The data suggest a sample-size boundary somewhere between 20 and 50 training images per class below which GAN augmentation becomes harmful in this setting, though further work across additional domains is needed to establish where that boundary sits more precisely. All experiments run on a consumer-grade GPU with 6 to 8 GB of memory, with no cloud compute required.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 12 of its 23 references that are in the map. First posted on arXiv 17 Mar 2026.
Builds on
12 of 23 references.
Closest ancestors · 1
Same direction, newest first
Other references · 11
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2017A systematic study of the class imbalance problem in convolutional neural networkscited by 510
- 2022Is synthetic data from generative models ready for image recognition?cited by 320
- 2021Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesiscited by 140
- 2018Synthetic Data Augmentation using GAN for Improved Liver Lesion Classificationcited by 136
- 2018MelanoGANs: High Resolution Skin Lesion Synthesis with GANscited by 14
- 2019Effective Data Augmentation with Multi-Domain Learning GANscited by 11
- 2022Adjusting for Bias with Procedural Datacited by 6
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Generative AI in Depth: A Survey of Recent Advances, Model Variants, and Real-World Applications
- 2023Diffusion Models Beat GANs on Image Classification
- 2025Do We Need All the Synthetic Data? Targeted Image Augmentation via Diffusion Models
- 2023Synthetic Data from Diffusion Models Improves ImageNet Classification
- 2026Expanding the Role of Diffusion Models for Robust Classifier Training
- 2025Generative Adversarial Networks Bridging Art and Machine Intelligence
- 2023Improving the Effectiveness of Deep Generative Data
- 2024Data Augmentation for Image Classification using Generative AI