← all repositories

mit-han-lab/data-efficient-gans

MIT research project implementing Differentiable Augmentation to enable GAN training with as few as 100 images through learned data transformations.

data-efficient-gans
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides a research implementation of Differentiable Augmentation (DiffAugment) for training Generative Adversarial Networks more efficiently with limited data. It includes PyTorch and TensorFlow implementations supporting StyleGAN2 and BigGAN architectures. The technique applies differentiable augmentation during training to stabilize GAN learning when only small datasets are available, enabling high-quality image generation from as few as 100 training examples without pre-training.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.