In plain words: A text-to-image generator is taught to make pictures for each known class, so its fake images can be added to the real training set. Classifiers trained on this mix beat strong versions trained only on real photos.
Abstract
Deep generative models are becoming increasingly powerful, now generating diverse high fidelity photo-realistic samples given text prompts. Have they reached the point where models of natural images can be used for generative data augmentation, helping to improve challenging discriminative tasks? We show that large-scale text-to image diffusion models can be fine-tuned to produce class conditional models with SOTA FID (1.76 at 256x256 resolution) and Inception Score (239 at 256x256). The model also yields a new SOTA in Classification Accuracy Scores (64.96 for 256x256 generative samples, improving to 69.24 for 1024x1024 samples). Augmenting the ImageNet training set with samples from the resulting models yields significant improvements in ImageNet classification accuracy over strong ResNet and Vision Transformer baselines.
Shekoofeh Azizi, Simon Kornblith, Chitwan Saharia, Mohammad Norouzi, David J. Fleet
arXiv:2304.08466 · cs.CV, cs.AI, cs.CL, cs.LG · submitted Apr 17, 2023
abstract · pdf · html
Does it give more lift than just using the training data from the diffusion model? You can search through LAION 5B in Clip embedding space and pull out lots of new training data without ever using a generative model.
Are they able to use prompts to generate augmentations around whatever invariant features - different colors or positions or scenes- in order to train better?