Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Negative Data Augmentation

About

Data augmentation is often used to enlarge datasets with synthetic samples generated in accordance with the underlying data distribution. To enable a wider range of augmentations, we explore negative data augmentation strategies (NDA)that intentionally create out-of-distribution samples. We show that such negative out-of-distribution samples provide information on the support of the data distribution, and can be leveraged for generative modeling and representation learning. We introduce a new GAN training objective where we use NDA as an additional source of synthetic data for the discriminator. We prove that under suitable conditions, optimizing the resulting objective still recovers the true data distribution but can directly bias the generator towards avoiding samples that lack the desired structure. Empirically, models trained with our method achieve improved conditional/unconditional image generation along with improved anomaly detection capabilities. Further, we incorporate the same negative data augmentation strategy in a contrastive learning framework for self-supervised representation learning on images and videos, achieving improved performance on downstream image classification, object detection, and action recognition tasks. These results suggest that prior knowledge on what does not constitute valid data is an effective form of weak supervision across a range of unsupervised learning tasks.

Abhishek Sinha, Kumar Ayush, Jiaming Song, Burak Uzkent, Hongxia Jin, Stefano Ermon• 2021

Related benchmarks

TaskDatasetResultRank
Unconditional Image GenerationCelebA unconditional 64 x 64
FID37.17
95
Conditional Image GenerationCIFAR10 (test)
Fréchet Inception Distance9.42
66
Class-conditional Image GenerationCIFAR-100 (test)
FID13.9
17
Unconditional Image GenerationCIFAR-10 32x32 (train test)
FID12.61
7
Unconditional Image GenerationCIFAR-100 32x32 (train test)
FID19.72
6
Unconditional Image GenerationSTL10 32x32 (train test)
FID23.94
6
Anomaly DetectionCIFAR-10 (In-Distribution) vs CIFAR-10-C (OOD-2) (test)
Avg AUROC0.63
3
Anomaly DetectionCIFAR-10 (In-Distribution) vs OOD-1 (Out-of-Distribution suite) (test)
AUROC (DTD)0.69
3
Showing 8 of 8 rows

Other info

Follow for update